Skip to content

Commit 357074c

Browse files
committed
wip litellm thinking
1 parent 83ce49e commit 357074c

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

src/agents/extensions/models/litellm_model.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,13 @@
2626
ChatCompletionMessage,
2727
)
2828
from openai.types.chat.chat_completion_message_tool_call import Function
29-
from openai.types.responses import Response
29+
from openai.types.responses import Response, ResponseReasoningItem
30+
from openai.types.responses.response_reasoning_item import Summary
3031

3132
from ... import _debug
3233
from ...agent_output import AgentOutputSchemaBase
3334
from ...handoffs import Handoff
34-
from ...items import ModelResponse, TResponseInputItem, TResponseStreamEvent
35+
from ...items import ModelResponse, ReasoningItem, TResponseInputItem, TResponseStreamEvent
3536
from ...logger import logger
3637
from ...model_settings import ModelSettings
3738
from ...models.chatcmpl_converter import Converter
@@ -123,10 +124,21 @@ async def get_response(
123124
"output_tokens": usage.output_tokens,
124125
}
125126

127+
message = response.choices[0].message
128+
126129
items = Converter.message_to_output_items(
127-
LitellmConverter.convert_message_to_openai(response.choices[0].message)
130+
LitellmConverter.convert_message_to_openai(message)
128131
)
129132

133+
if hasattr(message, "reasoning_content") and message.reasoning_content:
134+
items.append(
135+
ResponseReasoningItem(
136+
id=FAKE_RESPONSES_ID,
137+
summary=[Summary(text=message.reasoning_content, type="summary_text")],
138+
type="reasoning",
139+
)
140+
)
141+
130142
return ModelResponse(
131143
output=items,
132144
usage=usage,

0 commit comments

Comments
 (0)