phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-12 13:00:39 +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:40:39 -08:00
parent cd36a77e20
commit ee3c174bb3
1 changed files with 16 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -31,6 +31,7 @@ from llama_stack.apis.agents import (
AgentTurnResponseStepProgressPayload, AgentTurnResponseStepProgressPayload,
AgentTurnResponseStepStartPayload, AgentTurnResponseStepStartPayload,
AgentTurnResponseStreamChunk, AgentTurnResponseStreamChunk,
AgentTurnResponseTurnAwaitingInputPayload,
AgentTurnResponseTurnCompletePayload, AgentTurnResponseTurnCompletePayload,
AgentTurnResponseTurnStartPayload, AgentTurnResponseTurnStartPayload,
Attachment, Attachment,
@ -227,13 +228,23 @@ class ChatAgent(ShieldRunnerMixin):
) )
await self.storage.add_turn_to_session(request.session_id, turn) await self.storage.add_turn_to_session(request.session_id, turn)
chunk = AgentTurnResponseStreamChunk( if output_message.tool_calls:
event=AgentTurnResponseEvent( chunk = AgentTurnResponseStreamChunk(
payload=AgentTurnResponseTurnCompletePayload( event=AgentTurnResponseEvent(
turn=turn, payload=AgentTurnResponseTurnAwaitingInputPayload(
turn=turn,
)
) )
) )
) else:
chunk = AgentTurnResponseStreamChunk(
event=AgentTurnResponseEvent(
payload=AgentTurnResponseTurnCompletePayload(
turn=turn,
)
)
)
yield chunk yield chunk
async def continue_turn(self, request: AgentTurnContinueRequest) -> AsyncGenerator: async def continue_turn(self, request: AgentTurnContinueRequest) -> AsyncGenerator: