scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-25 01:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

fix(responses): keep the MCP lifecycle change within the type-discipline budget

Browse source 
Mark the dict literals handed straight to model constructors, clear the pending mcp_call list in place, and bind the merged response before setting it on the event, so LIT002 stays at or below its base count.
This commit is contained in:
Atharva-Kanherkar 2026-09-07 18:49:04 +05:30
parent 13ce30488f
commit 146e5c492b
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
litellm/responses/mcp/mcp_streaming_iterator.py
View file

@ -602,7 +602,7 @@ class MCPEnhancedStreamingIterator(BaseResponsesAPIStreamingIterator):
self._composed_output.extend(_output_items(response))
self._composed_output.extend(self._pending_mcp_call_items)
self._output_index_offset += width + len(self._pending_mcp_call_items)
self._pending_mcp_call_items = []
self._pending_mcp_call_items.clear()
self._round_max_output_index = -1
async def _compose_round_chunk(self, chunk: ResponsesAPIStreamingResponse) -> ResponsesAPIStreamingResponse | None:
@ -648,7 +648,10 @@ class MCPEnhancedStreamingIterator(BaseResponsesAPIStreamingIterator):
*self._composed_output,
*_output_items(response_obj),
]
_set_event_field(chunk, "response", response_obj.model_copy(update={"output": merged_output}))
merged_response: Final = response_obj.model_copy(
update={"output": merged_output} # mutable-ok: pydantic's update argument must be a dict
)
_set_event_field(chunk, "response", merged_response)
return chunk
async def _process_base_iterator_chunk(self) -> ResponsesAPIStreamingResponse:
@ -769,11 +772,11 @@ class MCPEnhancedStreamingIterator(BaseResponsesAPIStreamingIterator):
call_items[tool_call_id] = (item_id, output_index)
self.tool_execution_events.append(
OutputItemAddedEvent.model_validate(
{
{ # mutable-ok: consumed once by model_validate
"type": ResponsesAPIStreamEvents.OUTPUT_ITEM_ADDED,
"sequence_number": len(self.tool_execution_events) + 1,
"output_index": output_index,
"item": {
"item": { # mutable-ok: consumed once by model_validate
"id": item_id,
"type": "mcp_call",
"status": "in_progress",
@ -850,7 +853,7 @@ class MCPEnhancedStreamingIterator(BaseResponsesAPIStreamingIterator):
from litellm.types.llms.openai import OutputItemDoneEvent
mcp_call_item = BaseLiteLLMOpenAIResponseObject(
**{
**{ # mutable-ok: consumed once by the model constructor
"id": item_id,
"type": "mcp_call",
"approval_request_id": f"mcpr_{uuid.uuid4().hex[:8]}",