phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-13 05:17:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add back 2/n

Browse source
This commit is contained in:
Xi Yan 2025-02-20 17:52:01 -08:00
parent ee3c174bb3
commit 157cf320d9
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
llama_stack/providers/inline/agents/meta_reference/agent_instance.py
View file

@ -663,11 +663,7 @@ class ChatAgent(ShieldRunnerMixin):
input_messages = input_messages + [message] input_messages = input_messages + [message]
else: else:
log.info(f"{str(message)}") log.info(f"{str(message)}")
tool_call = message.tool_calls[0] # 1. Start the tool execution step and progress
if tool_call.tool_name in client_tools:
yield message
return
step_id = str(uuid.uuid4()) step_id = str(uuid.uuid4())
yield AgentTurnResponseStreamChunk( yield AgentTurnResponseStreamChunk(
event=AgentTurnResponseEvent( event=AgentTurnResponseEvent(
@ -691,6 +687,13 @@ class ChatAgent(ShieldRunnerMixin):
) )
) )
# If tool is a client tool, yield CompletionMessage and return
tool_call = message.tool_calls[0]
if tool_call.tool_name in client_tools:
yield message
return
# If tool is a builtin server tool, execute it
tool_name = tool_call.tool_name tool_name = tool_call.tool_name
if isinstance(tool_name, BuiltinTool): if isinstance(tool_name, BuiltinTool):
tool_name = tool_name.value tool_name = tool_name.value