fix bug with using enum as string

Hashim-K · Mar 19, 2024 · 212aa22 · 212aa22
1 parent 81c4fbe
commit 212aa22
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/backend/llm.py b/backend/llm.py
@@ -35,7 +35,12 @@ async def stream_openai_response(
     model = Llm.GPT_4_VISION
 
     # Base parameters
-    params = {"model": model, "messages": messages, "stream": True, "timeout": 600}
+    params = {
+        "model": model.value,
+        "messages": messages,
+        "stream": True,
+        "timeout": 600,
+    }
 
     # Add 'max_tokens' only if the model is a GPT4 vision model
     if model == Llm.GPT_4_VISION: