mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-12 13:00:39 +00:00
add back 2/n
This commit is contained in:
parent
157cf320d9
commit
22355e3b1f
1 changed files with 2 additions and 2 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue