| | |
|---|
| | | |
|---|
| | | # 监听毕昇发来的消息并转发给客户端 |
|---|
| | | async def forward_to_client(): |
|---|
| | | last_message = "step" |
|---|
| | | while True: |
|---|
| | | message = await service_websocket.recv() |
|---|
| | | print(f"Received from bisheng: {message}") |
|---|
| | | data = json.loads(message) |
|---|
| | | files = data.get("files", []) |
|---|
| | | steps = data.get("intermediate_steps", "") |
|---|
| | | if len(files) != 0 or steps != "" or data["type"] == "close": |
|---|
| | | msg = data.get("message", "") |
|---|
| | | if len(files) != 0 or (steps and last_message == "step") or msg or data["type"] == "close": |
|---|
| | | last_message = "step" if steps else "message" |
|---|
| | | if data["type"] == "close": |
|---|
| | | t = "close" |
|---|
| | | else: |
|---|
| | | t = "stream" |
|---|
| | | process_files(files, agent_id) |
|---|
| | | result = {"step_message": steps, "type": t, "files": files} |
|---|
| | | result = {"step_message": steps, "message": msg, "type": t, "files": files} |
|---|
| | | await websocket.send_json(result) |
|---|
| | | print(f"Forwarded to client, {chat_id}: {result}") |
|---|
| | | |
|---|
