AashitaK commited on
Commit
127a9af
·
verified ·
1 Parent(s): af770af

Update mistral_api.py

Browse files
Files changed (1) hide show
  1. mistral_api.py +1 -1
mistral_api.py CHANGED
@@ -31,7 +31,7 @@ def get_response(messages: list[dict], temperature=0, max_tokens=800) -> str:
31
  # stream=True # Enables streaming
32
  )
33
  print(response)
34
- return response#["choices"][0]["message"]["content"]
35
 
36
  # def get_response(message, system_message=META_PROMPT, max_tokens=800, temperature=0) -> str:
37
  # """Chat completion using HuggingFace's Inference API"""
 
31
  # stream=True # Enables streaming
32
  )
33
  print(response)
34
+ return "hello"#response["choices"][0]["message"]["content"]
35
 
36
  # def get_response(message, system_message=META_PROMPT, max_tokens=800, temperature=0) -> str:
37
  # """Chat completion using HuggingFace's Inference API"""