This commit is contained in:
Xi Yan 2025-03-04 09:42:50 -08:00
parent 2d0bf027e5
commit fdd70d71cd
5 changed files with 5311 additions and 7 deletions

View file

@ -243,7 +243,15 @@ class ChatAgent(ShieldRunnerMixin):
steps=steps,
)
await self.storage.add_turn_to_session(request.session_id, turn)
if output_message.tool_calls:
chunk = AgentTurnResponseStreamChunk(
event=AgentTurnResponseEvent(
payload=AgentTurnResponseTurnAwaitingInputPayload(
turn=turn,
)
)
)
else:
chunk = AgentTurnResponseStreamChunk(
event=AgentTurnResponseEvent(
payload=AgentTurnResponseTurnCompletePayload(

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long