feat: main realtime added

Author: csgulati09

portkey_ai/__init__.py

@@ -135,6 +135,8 @@
     AsyncProviders,
     Webhooks,
     AsyncWebhooks,
+    MainRealtime,
+    AsyncMainRealtime,
 )
 
 from portkey_ai.version import VERSION
@@ -289,4 +291,6 @@
     "AsyncProviders",
     "Webhooks",
     "AsyncWebhooks",
+    "MainRealtime",
+    "AsyncMainRealtime",
 ]

portkey_ai/api_resources/__init__.py

@@ -123,6 +123,8 @@
     AsyncProviders,
     Webhooks,
     AsyncWebhooks,
+    MainRealtime,
+    AsyncMainRealtime,
 )
 from .utils import (
     Modes,
@@ -281,4 +283,6 @@
     "AsyncProviders",
     "Webhooks",
     "AsyncWebhooks",
+    "MainRealtime",
+    "AsyncMainRealtime",
 ]

portkey_ai/api_resources/apis/__init__.py

@@ -157,6 +157,11 @@
     AsyncProviders,
 )
 
+from .main_realtime import (
+    MainRealtime,
+    AsyncMainRealtime,
+)
+
 sys.modules["openai"] = vendored_openai  # For pydantic v1 and v2 compatibility
 
 __all__ = [
@@ -293,4 +298,6 @@
     "AsyncProviders",
     "Webhooks",
     "AsyncWebhooks",
+    "MainRealtime",
+    "AsyncMainRealtime",
 ]

portkey_ai/api_resources/apis/main_realtime.py

@@ -0,0 +1,52 @@
+from portkey_ai._vendor.openai.resources.realtime.realtime import (
+    AsyncRealtimeConnectionManager,
+    RealtimeConnectionManager,
+)
+from portkey_ai._vendor.openai.types.websocket_connection_options import (
+    WebsocketConnectionOptions,
+)
+from portkey_ai.api_resources.apis.api_resource import APIResource, AsyncAPIResource
+from portkey_ai.api_resources.client import AsyncPortkey, Portkey
+from portkey_ai.api_resources.types.shared_types import Headers, Query
+
+
+class MainRealtime(APIResource):
+    def __init__(self, client: Portkey) -> None:
+        super().__init__(client)
+        self.openai_client = client.openai_client
+
+    def connect(
+        self,
+        *,
+        model: str,
+        extra_query: Query = {},
+        extra_headers: Headers = {},
+        websocket_connection_options: WebsocketConnectionOptions = {},
+    ) -> RealtimeConnectionManager:
+        return self.openai_client.realtime.connect(
+            model=model,
+            extra_query=extra_query,
+            extra_headers=extra_headers,
+            websocket_connection_options=websocket_connection_options,
+        )
+
+
+class AsyncMainRealtime(AsyncAPIResource):
+    def __init__(self, client: AsyncPortkey) -> None:
+        super().__init__(client)
+        self.openai_client = client.openai_client
+
+    def connect(
+        self,
+        *,
+        model: str,
+        extra_query: Query = {},
+        extra_headers: Headers = {},
+        websocket_connection_options: WebsocketConnectionOptions = {},
+    ) -> AsyncRealtimeConnectionManager:
+        return self.openai_client.realtime.connect(
+            model=model,
+            extra_query=extra_query,
+            extra_headers=extra_headers,
+            websocket_connection_options=websocket_connection_options,
+        )

portkey_ai/api_resources/client.py

@@ -38,6 +38,7 @@ class Portkey(APIClient):
     collections: apis.Collections
     integrations: apis.Integrations
     providers: apis.Providers
+    realtime: apis.MainRealtime
 
     class beta:
         assistants: apis.Assistants
@@ -184,6 +185,7 @@ def __init__(
         self.collections = apis.Collections(self)
         self.integrations = apis.Integrations(self)
         self.providers = apis.Providers(self)
+        self.realtime = apis.MainRealtime(self)
         self.beta = self.beta(self)  # type: ignore
 
         if self.instrumentation:
@@ -366,6 +368,7 @@ class AsyncPortkey(AsyncAPIClient):
     collections: apis.AsyncCollections
     integrations: apis.AsyncIntegrations
     providers: apis.AsyncProviders
+    realtime: apis.AsyncMainRealtime
 
     class beta:
         assistants: apis.AsyncAssistants
@@ -512,6 +515,7 @@ def __init__(
         self.collections = apis.AsyncCollections(self)
         self.integrations = apis.AsyncIntegrations(self)
         self.providers = apis.AsyncProviders(self)
+        self.realtime = apis.AsyncMainRealtime(self)
         self.beta = self.beta(self)  # type: ignore
 
         if self.instrumentation: