66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
小智设备连接测试客户端
|
||
连接到 xiaozhi-server，模拟设备握手，触发 MCP 工具初始化
|
||
"""
|
||
|
||
import asyncio
|
||
import json
|
||
import uuid
|
||
import websockets
|
||
import sys
|
||
|
||
XIAOZHI_WS = "ws://localhost:8000/xiaozhi/v1/"
|
||
DEVICE_ID = str(uuid.uuid4())
|
||
TOPIC = str(uuid.uuid4())
|
||
|
||
|
||
async def main():
|
||
print(f"连接 xiaozhi-server: {XIAOZHI_WS}")
|
||
print(f"Device ID: {DEVICE_ID}")
|
||
|
||
async with websockets.connect(XIAOZHI_WS, ping_interval=None) as ws:
|
||
# 发送设备 hello
|
||
hello = {
|
||
"type": "hello",
|
||
"deviceId": DEVICE_ID,
|
||
"deviceName": "测试设备",
|
||
"clientType": "py-websocket-test",
|
||
"protocolVersion": "1.0.0",
|
||
"featureSet": {"mcp": True},
|
||
}
|
||
await ws.send(json.dumps(hello))
|
||
print(f"发送 hello: {hello}")
|
||
|
||
# 接收消息（最多等 30 秒）
|
||
for i in range(20):
|
||
try:
|
||
msg = await asyncio.wait_for(ws.recv(), timeout=5.0)
|
||
data = json.loads(msg)
|
||
msg_type = data.get("type", "unknown")
|
||
print(f"收到 [{msg_type}]: {json.dumps(data, ensure_ascii=False)[:200]}")
|
||
|
||
if msg_type == "welcome":
|
||
print("\n✅ 连接成功！")
|
||
if "tools" in data:
|
||
print(f"MCP 工具列表: {[t.get('name') for t in data.get('tools', [])]}")
|
||
if "mcpTools" in data:
|
||
print(f"MCP 工具: {data.get('mcpTools')}")
|
||
break
|
||
except asyncio.TimeoutError:
|
||
print(f" ... 等待中 ({i+1}/20)")
|
||
continue
|
||
|
||
print("\n测试完成，5秒后关闭...")
|
||
await asyncio.sleep(5)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
try:
|
||
asyncio.run(main())
|
||
except KeyboardInterrupt:
|
||
print("\n已退出")
|
||
except Exception as e:
|
||
print(f"错误: {e}")
|
||
sys.exit(1)
|