phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-03 18:00:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

reverted some formatting changes

Browse source
This commit is contained in:
Omar Abdelwahab 2025-11-04 12:39:48 -08:00
parent fec6f20792
commit abc717ed1d
1 changed files with 7 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

30
src/llama_stack/providers/inline/agents/meta_reference/responses/tool_executor.py
View file

@ -26,7 +26,10 @@ from llama_stack.apis.agents.openai_responses import (
OpenAIResponseOutputMessageMCPCall, OpenAIResponseOutputMessageMCPCall,
OpenAIResponseOutputMessageWebSearchToolCall, OpenAIResponseOutputMessageWebSearchToolCall,
) )
from llama_stack.apis.common.content_types import ImageContentItem, TextContentItem from llama_stack.apis.common.content_types import (
ImageContentItem,
TextContentItem,
)
from llama_stack.apis.inference import ( from llama_stack.apis.inference import (
OpenAIChatCompletionContentPartImageParam, OpenAIChatCompletionContentPartImageParam,
OpenAIChatCompletionContentPartTextParam, OpenAIChatCompletionContentPartTextParam,
@ -86,12 +89,7 @@ class ToolExecutor:
# Emit progress events for tool execution start # Emit progress events for tool execution start
async for event_result in self._emit_progress_events( async for event_result in self._emit_progress_events(
function.name, function.name, ctx, sequence_number, output_index, item_id, mcp_tool_to_server
ctx,
sequence_number,
output_index,
item_id,
mcp_tool_to_server,
): ):
sequence_number = event_result.sequence_number sequence_number = event_result.sequence_number
yield event_result yield event_result
@ -111,28 +109,14 @@ class ToolExecutor:
) )
) )
async for event_result in self._emit_completion_events( async for event_result in self._emit_completion_events(
function.name, function.name, ctx, sequence_number, output_index, item_id, has_error, mcp_tool_to_server,
ctx,
sequence_number,
output_index,
item_id,
has_error,
mcp_tool_to_server,
): ):
sequence_number = event_result.sequence_number sequence_number = event_result.sequence_number
yield event_result yield event_result
# Build result messages from tool execution # Build result messages from tool execution
output_message, input_message = await self._build_result_messages( output_message, input_message = await self._build_result_messages(
function, function, tool_call_id, item_id, tool_kwargs, ctx, error_exc, result, has_error, mcp_tool_to_server,
tool_call_id,
item_id,
tool_kwargs,
ctx,
error_exc,
result,
has_error,
mcp_tool_to_server,
) )
# Yield the final result # Yield the final result