| | |
|---|
| | | |
|---|
| | | # 监听毕昇发来的消息并转发给客户端 |
|---|
| | | 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", "") |
|---|
| | | msg = data.get("message", "") |
|---|
| | | |
|---|
| | | if len(files) != 0 or (steps and last_message == "step") or msg or data["type"] == "close": |
|---|
| | | if len(files) != 0 or steps or data["type"] == "close": |
|---|
| | | if data["type"] == "close": |
|---|
| | | t = "close" |
|---|
| | | else: |
|---|
| | | t = "stream" |
|---|
| | | process_files(files, agent_id) |
|---|
| | | result = {"step_message": steps, "message": msg, "type": t, "files": files} |
|---|
| | | result = {"step_message": steps, "type": t, "files": files} |
|---|
| | | await websocket.send_json(result) |
|---|
| | | print(f"Forwarded to client, {chat_id}: {result}") |
|---|
| | | |
|---|
| | | last_message = "message" if msg else "step" |
|---|
| | | |
|---|
| | | # 启动两个任务，分别处理客户端和服务端的消息 |
|---|
| | | tasks = [ |
|---|
