Skip to content

Commit 62375b8

Browse files
committed
address comments
1 parent 98a3c0e commit 62375b8

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

  • src/pydantic_ai_lightspeed/llamastack

src/pydantic_ai_lightspeed/llamastack/_model.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,10 @@ async def request_stream(
225225
"Streamed response ended without content or tool calls"
226226
)
227227

228-
assert isinstance(first_chunk, responses.ResponseCreatedEvent)
228+
if not isinstance(first_chunk, responses.ResponseCreatedEvent):
229+
raise UnexpectedModelBehavior(
230+
f"Expected ResponseCreatedEvent, got {type(first_chunk).__name__}"
231+
)
229232

230233
yield OpenAIResponsesStreamedResponse(
231234
model_request_parameters=model_request_parameters,

0 commit comments

Comments
 (0)