Skip to content

Commit cf770d3

Browse files
committed
Fix for broken o3-streaming model, closes #17
1 parent e4303c9 commit cf770d3

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

llm_openai.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,9 @@ def __init__(
141141
self, model_name, vision=False, streaming=True, schemas=True, reasoning=False
142142
):
143143
self.model_id = "openai/" + model_name
144+
streaming_suffix = "-streaming"
145+
if model_name.endswith(streaming_suffix):
146+
model_name = model_name[: -len(streaming_suffix)]
144147
self.model_name = model_name
145148
self.can_stream = streaming
146149
self.supports_schema = schemas

0 commit comments

Comments
 (0)