diff --git a/src/backend/constants.py b/src/backend/constants.py index c6144a6..abae5c3 100644 --- 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",