Corrected liteLLM model providers

When I use default setting and clicked on fast/powerful, the following error message occurs: 500: LLM Provider NOT provided. Pass in the LLM provider you are trying to call. You passed model=OpenAI/gpt-4o-mini Pass model as E.g. For 'Huggingface' inference endpoints pass in completion(model='huggingface/starcoder',..) Learn more: https://docs.litellm.ai/docs/providers
rashadphz · Sep 9, 2024 · c0e50de · c0e50de
1 parent 883003f
commit c0e50de
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",