From 22355e3b1f2e19b4c387c18d6dcda21eec237939 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 20 Feb 2025 17:53:29 -0800 Subject: [PATCH] add back 2/n --- .../providers/inline/agents/meta_reference/agent_instance.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py index 6a581cf11..bd12222a8 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py +++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py @@ -268,7 +268,7 @@ class ChatAgent(ShieldRunnerMixin): for i, turn in enumerate(turns): messages.extend(self.turn_to_messages(turn)) - messages.extend(request.messages) + messages.extend(request.tool_responses) # steps = [] # output_message = None @@ -673,6 +673,7 @@ class ChatAgent(ShieldRunnerMixin): ) ) ) + tool_call = message.tool_calls[0] yield AgentTurnResponseStreamChunk( event=AgentTurnResponseEvent( payload=AgentTurnResponseStepProgressPayload( @@ -688,7 +689,6 @@ 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