openai_realtime_api_websocket_text_message_quickstart.py

1import asyncio
2import os
3import json
4import websockets
5
6async def quickstart():
7    url = "wss://api.openai.com/v1/realtime?model=gpt-4o-realtime-preview-2024-10-01"
8    headers = {
9        "Authorization": "Bearer " + os.getenv("OPENAI_API_KEY"),
10        "OpenAI-Beta": "realtime=v1",
11    }
12
13    async with websockets.connect(url, extra_headers=headers) as ws:
14        # Initialize the session or send an initial event
15        event = {
16            "type": "response.create",
17            "response": {
18                "modalities": ["text"],
19                "instructions": "Say hello!",
20            }
21        }
22        await ws.send(json.dumps(event))
23
24        async for message in ws:
25            response = json.loads(message)
26            # Print the events received from the server
27            print(json.dumps(response, indent=2))
28
29            # Stop after the response is completed
30            if response.get("type") == "response.done":
31                break
32
33if __name__ == "__main__":
34    asyncio.run(quickstart())