Introduce new OpenAI models for Querybook

pinterest · Oct 22, 2024 · 7461f6e · 7461f6e
1 parent ee25d74
commit 7461f6e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/querybook/server/lib/ai_assistant/assistants/openai_assistant.py b/querybook/server/lib/ai_assistant/assistants/openai_assistant.py
@@ -9,13 +9,16 @@
 
 
 OPENAI_MODEL_CONTEXT_WINDOW_SIZE = {
+    # Legacy models
     "gpt-3.5-turbo": 16385,
     "gpt-4": 8192,
     "gpt-4-32k": 32768,
     "gpt-4-turbo": 128000,
+    # Current models
     "gpt-4o": 128000,
+    "gpt-4o-mini": 128000,
 }
-DEFAULT_MODEL_NAME = "gpt-3.5-turbo"
+DEFAULT_MODEL_NAME = "gpt-4o-mini"
 
 
 class OpenAIAssistant(BaseAIAssistant):