[Improvement] Allow o-series models with OpenAI Assistant mode

amidabuddha · Feb 1, 2025 · f98f97d · f98f97d
1 parent 3c6acca
commit f98f97d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/console_gpt/menus/combined_menu.py b/console_gpt/menus/combined_menu.py
@@ -36,7 +36,7 @@ def combined_menu() -> ChatObject | AssistantObject:
     model = model_menu()
     if model["api_key"] in api_key_placeholders:
         model = set_api_key(model)
-    assistant = assistant_menu(model) if model["model_title"].lower().startswith(("gpt")) else None
+    assistant = assistant_menu(model) if model["model_title"].lower().startswith(("gpt","o1","o3")) else None
     if assistant:
         return AssistantObject(
             model=model, assistant_name=assistant[0], assistant_id=assistant[1], thread_id=assistant[2]