Fix websocket connection: Update headers and state checks

[mubashirsidiki]

The code wasn’t working because of two issues. Below are the before/after examples for each fix:

1. Websocket Header Parameter

• Before:

  async with websockets.connect(
      'wss://api.openai.com/v1/realtime?model=gpt-4o-realtime-preview-2024-10-01',
      extra_headers={
          "Authorization": f"Bearer {OPENAI_API_KEY}",
          "OpenAI-Beta": "realtime=v1"
      }
  ) as openai_ws:

• After:

  async with websockets.connect(
      'wss://api.openai.com/v1/realtime?model=gpt-4o-realtime-preview-2024-10-01',
      additional_headers={
          "Authorization": f"Bearer {OPENAI_API_KEY}",
          "OpenAI-Beta": "realtime=v1"
      }
  ) as openai_ws:

Source: Official websockets documentation – Websockets HTTP headers

2. Connection State Check

• Before:

  if data['event'] == 'media' and openai_ws.open:
      # ... send audio data ...
  
  except WebSocketDisconnect:
      print("Client disconnected.")
      if openai_ws.open:
          await openai_ws.close()

• After:

  if data['event'] == 'media' and openai_ws.state.name == 'OPEN':
      # ... send audio data ...
  
  except WebSocketDisconnect:
      print("Client disconnected.")
      if openai_ws.state.name == 'OPEN':
          await openai_ws.close()

Source: last line of 'Backwards-incompatible changes' heading – WebSocketCommonProtocol.state