diff --git a/src/blackbox/chat.py b/src/blackbox/chat.py index 0433ce3..a0bc1bd 100644 --- a/src/blackbox/chat.py +++ b/src/blackbox/chat.py @@ -140,13 +140,13 @@ class Chat(Blackbox): "content": user_question } ], - "temperature": user_temperature, - "top_p": user_top_p, - "n": user_n, - "max_tokens": user_max_tokens, - "frequency_penalty": user_frequency_penalty, - "presence_penalty": user_presence_penalty, - "stop": user_stop + "temperature": str(user_temperature), + "top_p": str(user_top_p), + "n": str(user_n), + "max_tokens": str(user_max_tokens), + "frequency_penalty": str(user_frequency_penalty), + "presence_penalty": str(user_presence_penalty), + "stop": str(user_stop) } header = { diff --git a/src/blackbox/chroma_chat.py b/src/blackbox/chroma_chat.py index 65495e6..26c1093 100755 --- a/src/blackbox/chroma_chat.py +++ b/src/blackbox/chroma_chat.py @@ -42,7 +42,7 @@ class ChromaChat(Blackbox): # "context": [], # "template": "", # "temperature": 0, - # "top_p": 0, + # "top_p": 0.1, # "n": 1, # "max_tokens": 1024, # "frequency_penalty": 0,