🐛 minor fix for o1 models (#42)

WecoAI · Jan 24, 2025 · 51d09b7 · 51d09b7
1 parent 9e48792
commit 51d09b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aide/backend/__init__.py b/aide/backend/__init__.py
@@ -35,7 +35,7 @@ def query(
 
     # Handle models with beta limitations
     # ref: https://platform.openai.com/docs/guides/reasoning/beta-limitations
-    if model.startswith("o1-"):
+    if model.startswith("o1"):
         if system_message:
             user_message = system_message
         system_message = None