o1-preview and o1-mini can stream now

Refs #676 (comment)
simonw · Dec 17, 2024 · 6cf4127 · 6cf4127
1 parent 8378da6
commit 6cf4127
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/llm/default_plugins/openai_models.py b/llm/default_plugins/openai_models.py
@@ -63,12 +63,12 @@ def register_models(register):
     )
     # o1
     register(
-        Chat("o1-preview", can_stream=False, allows_system_prompt=False),
-        AsyncChat("o1-preview", can_stream=False, allows_system_prompt=False),
+        Chat("o1-preview", allows_system_prompt=False),
+        AsyncChat("o1-preview", allows_system_prompt=False),
     )
     register(
-        Chat("o1-mini", can_stream=False, allows_system_prompt=False),
-        AsyncChat("o1-mini", can_stream=False, allows_system_prompt=False),
+        Chat("o1-mini", allows_system_prompt=False),
+        AsyncChat("o1-mini", allows_system_prompt=False),
     )
     # The -instruct completion model
     register(