request_demo.py

import asyncio
import websockets
import json

async def send_request():
    url = "ws://0.0.0.0:8000/chat"  # WebSocket 服务地址
    async with websockets.connect(url) as websocket:
        infro = {
            "token": "878141"
        }
        await websocket.send(json.dumps(infro))

        data = {
            "planner_model": "deepseek-v3",
            "planner_provider": "dashscope",
            "planner_api_key": "sk-xxx",
            "planner_base_url": "",
            "dispatcher_model": "qwen-max-2025-01-25",
            "dispatcher_provider": "dashscope",
            "dispatcher_api_key": "sk-xxx",
            "dispatcher_base_url": "",
            "executor_model": "qwen2.5-vl-7b-instruct",
            "executor_provider": "dashscope",
            "executor_api_key": "sk-xxx",
            "executor_base_url": "",
            "user_query": "Help me find some information from GitHub trending and write it to Word, including the project name and description",
        }

        await websocket.send(json.dumps(data))

        while True:
            response = await websocket.recv()

            # 判断是否是二进制数据
            if isinstance(response, bytes):
                # 这里可以保存为图片或处理二进制数据
                with open("received_image.jpg", "wb") as image_file:
                    image_file.write(response)

            elif isinstance(response, str):
                # 如果是字符串，则认为是 JSON 数据
                try:
                    response = json.loads(response)  # 将字符串解析为 JSON 对象
                    print("Received JSON data:", response)
                    if response["message"] == "Process processing":
                        await websocket.send(json.dumps({"message": "Successfully obtained data"}))
                    elif response["message"] == "Processing complete" or response["message"] == "Process interruption":
                        await websocket.close()  # 关闭 WebSocket 连接
                        break

                except json.JSONDecodeError:
                    print("Failed to decode JSON data")


                await asyncio.sleep(0.5)


# 运行客户端
asyncio.get_event_loop().run_until_complete(send_request())