Commit78f08f8

authored

Handle DeepSeek reasoning_content in streamed responses (#2174)

1 parent6d0e850 commit78f08f8Copy full SHA for 78f08f8

File tree

+307

-1

lines changed

pydantic_ai_slim/pydantic_ai/models
- openai.py
tests/models
- cassettes/test_deepseek
  - test_deepseek_model_thinking_stream.yaml
- test_deepseek.py

+307

-1

lines changed

Lines changed: 6 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -994,6 +994,12 @@ async def _get_event_iterator(self) -> AsyncIterator[ModelResponseStreamEvent]:`
`994`	`994`	`ifcontentisnotNone:`
`995`	`995`	`yieldself._parts_manager.handle_text_delta(vendor_part_id='content',content=content)`
`996`	`996`
	`997`	`+# Handle reasoning part of the response, present in DeepSeek models`
	`998`	`+ifreasoning_content:=getattr(choice.delta,'reasoning_content',None):`
	`999`	`+yieldself._parts_manager.handle_thinking_delta(`
	`1000`	`+vendor_part_id='reasoning_content',content=reasoning_content`
	`1001`	`+ )`
	`1002`	`+`
`997`	`1003`	`fordtcinchoice.delta.tool_callsor []:`
`998`	`1004`	`maybe_event=self._parts_manager.handle_tool_call_delta(`
`999`	`1005`	`vendor_part_id=dtc.index,`

Comments

(0)