Merge pull request #93 from whyXVI/main

Corrected liteLLM model providers
rashadphz · Sep 27, 2024 · a9dd493 · a9dd493
2 parents 883003f + c0e50de
commit a9dd493
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/backend/constants.py b/src/backend/constants.py
@@ -23,8 +23,8 @@ class ChatModel(str, Enum):
 
 
 model_mappings: dict[ChatModel, str] = {
-    ChatModel.GPT_4o: "gpt-4o",
-    ChatModel.GPT_4o_mini: "gpt-4o-mini",
+    ChatModel.GPT_4o: "openai/gpt-4o",
+    ChatModel.GPT_4o_mini: "openai/gpt-4o-mini",
     ChatModel.LLAMA_3_70B: "groq/llama-3.1-70b-versatile",
     ChatModel.LOCAL_LLAMA_3: "ollama_chat/llama3.1",
     ChatModel.LOCAL_GEMMA: "ollama_chat/gemma",