fix: realtime api headers

Portkey-AI · Dec 26, 2024 · c3b60eb · c3b60eb
1 parent bb5989e
commit c3b60eb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/portkey_ai/api_resources/apis/beta_realtime.py b/portkey_ai/api_resources/apis/beta_realtime.py
@@ -29,6 +29,7 @@ def connect(
         return self.openai_client.beta.realtime.connect(
             model=model,
             websocket_connection_options=websocket_connection_options,
+            extra_headers=self.openai_client.default_headers,
             **kwargs,
         )
 
@@ -49,6 +50,7 @@ def connect(
         return self.openai_client.beta.realtime.connect(
             model=model,
             websocket_connection_options=websocket_connection_options,
+            extra_headers=self.openai_client.default_headers,
             **kwargs,
         )
 

diff --git a/portkey_ai/api_resources/client.py b/portkey_ai/api_resources/client.py
@@ -128,7 +128,7 @@ def __init__(
             default_headers=self.allHeaders,
             http_client=http_client,
             max_retries=0,
-            websocket_base_url=websocket_base_url,
+            websocket_base_url=self.websocket_base_url,
         )
 
         self.completions = apis.Completion(self)
@@ -352,7 +352,7 @@ def __init__(
             default_headers=self.allHeaders,
             http_client=http_client,
             max_retries=0,
-            websocket_base_url=websocket_base_url,
+            websocket_base_url=self.websocket_base_url,
         )
 
         self.completions = apis.AsyncCompletion(self)