This commit is contained in:
2025-02-24 20:02:06 +08:00
parent 89728ada0b
commit eac4969e6c
2 changed files with 5 additions and 3 deletions

View File

@@ -220,8 +220,9 @@ async def chat(websocket: WebSocket):
async for message in stream:
if isinstance(message, TaskResult):
continue
if message.type == 'TextMessage' or message.type == 'HandoffMessage':
print(f"----------------{message.source}----------------\n {message.content}")
print(f"----------------{message.source}----------------\n {message.content}")
if message.type == 'TextMessage' or message.type == 'HandoffMessage' \
or message.type == 'ToolCallRequestEvent' or message.type == 'UserInputRequestedEvent':
await websocket.send_json(message.model_dump())
if not isinstance(message, UserInputRequestedEvent):
history.append(message.model_dump())