Implement WS v1 protocol and runtime-config powered debug drawer
This commit is contained in:
@@ -36,7 +36,7 @@ def generate_sine_wave(duration_ms=1000):
|
||||
return audio_data
|
||||
|
||||
|
||||
async def receive_loop(ws):
|
||||
async def receive_loop(ws, ready_event: asyncio.Event):
|
||||
"""Listen for incoming messages from the server."""
|
||||
print("👂 Listening for server responses...")
|
||||
async for msg in ws:
|
||||
@@ -45,8 +45,10 @@ async def receive_loop(ws):
|
||||
if msg.type == aiohttp.WSMsgType.TEXT:
|
||||
try:
|
||||
data = json.loads(msg.data)
|
||||
event_type = data.get('event', 'Unknown')
|
||||
event_type = data.get('type', 'Unknown')
|
||||
print(f"[{timestamp}] 📨 Event: {event_type} | {msg.data[:150]}...")
|
||||
if event_type == "session.started":
|
||||
ready_event.set()
|
||||
except json.JSONDecodeError:
|
||||
print(f"[{timestamp}] 📨 Text: {msg.data[:100]}...")
|
||||
|
||||
@@ -118,35 +120,43 @@ async def run_client(url, file_path=None, use_sine=False):
|
||||
print(f"🔌 Connecting to {url}...")
|
||||
async with session.ws_connect(url) as ws:
|
||||
print("✅ Connected!")
|
||||
session_ready = asyncio.Event()
|
||||
recv_task = asyncio.create_task(receive_loop(ws, session_ready))
|
||||
|
||||
# Send initial invite command
|
||||
init_cmd = {
|
||||
"command": "invite",
|
||||
"option": {
|
||||
"codec": "pcm",
|
||||
"samplerate": SAMPLE_RATE
|
||||
# Send v1 hello + session.start handshake
|
||||
await ws.send_json({"type": "hello", "version": "v1"})
|
||||
await ws.send_json({
|
||||
"type": "session.start",
|
||||
"audio": {
|
||||
"encoding": "pcm_s16le",
|
||||
"sample_rate_hz": SAMPLE_RATE,
|
||||
"channels": 1
|
||||
}
|
||||
}
|
||||
await ws.send_json(init_cmd)
|
||||
print("📤 Sent Invite Command")
|
||||
})
|
||||
print("📤 Sent v1 hello/session.start")
|
||||
await asyncio.wait_for(session_ready.wait(), timeout=8)
|
||||
|
||||
# Select sender based on args
|
||||
if use_sine:
|
||||
sender_task = send_sine_loop(ws)
|
||||
await send_sine_loop(ws)
|
||||
elif file_path:
|
||||
sender_task = send_file_loop(ws, file_path)
|
||||
await send_file_loop(ws, file_path)
|
||||
else:
|
||||
# Default to sine wave
|
||||
sender_task = send_sine_loop(ws)
|
||||
await send_sine_loop(ws)
|
||||
|
||||
# Run send and receive loops in parallel
|
||||
await asyncio.gather(
|
||||
receive_loop(ws),
|
||||
sender_task
|
||||
)
|
||||
await ws.send_json({"type": "session.stop", "reason": "test_complete"})
|
||||
await asyncio.sleep(1)
|
||||
recv_task.cancel()
|
||||
try:
|
||||
await recv_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
except aiohttp.ClientConnectorError:
|
||||
print(f"❌ Connection Failed. Is the server running at {url}?")
|
||||
except asyncio.TimeoutError:
|
||||
print("❌ Timeout waiting for session.started")
|
||||
except Exception as e:
|
||||
print(f"❌ Error: {e}")
|
||||
finally:
|
||||
|
||||
Reference in New Issue
Block a user