import asyncio
import websockets
import json
async def stream_execution():
uri = f"wss://{sandbox_public_host.replace('https://', '')}/execute/stream"
headers = {"Authorization": f"Bearer {auth_token}"}
async with websockets.connect(uri, extra_headers=headers) as ws:
# Send execution request
await ws.send(json.dumps({
"type": "execute",
"code": "for i in range(5): print(i)",
"language": "python"
}))
# Receive messages
async for message in ws:
data = json.loads(message)
if data["type"] == "stdout":
print(data["data"], end="")
elif data["type"] == "complete":
print(f"\nExit code: {data['exit_code']}")
break
asyncio.run(stream_execution())