feat: beta realtime transcription_sessions added

Author: csgulati09

portkey_ai/api_resources/apis/beta_realtime.py

@@ -3,6 +3,9 @@
     AsyncRealtimeConnectionManager,
     RealtimeConnectionManager,
 )
+from portkey_ai._vendor.openai.types.beta.realtime.transcription_session import (
+    TranscriptionSession,
+)
 from portkey_ai._vendor.openai.types.websocket_connection_options import (
     WebsocketConnectionOptions,
 )
@@ -17,6 +20,7 @@ def __init__(self, client: Portkey) -> None:
         super().__init__(client)
         self.openai_client = client.openai_client
         self.sessions = BetaSessions(client)
+        self.transcription_sessions = BetaTranscriptionSessions(client)
 
     def connect(
         self,
@@ -33,11 +37,49 @@ def connect(
         )
 
 
+class BetaTranscriptionSessions(APIResource):
+    def __init__(self, client: Portkey) -> None:
+        super().__init__(client)
+        self.openai_client = client.openai_client
+
+    def create(
+        self,
+        *,
+        client_secret: Union[Any, NotGiven] = NOT_GIVEN,
+        include: Union[List[Any], NotGiven] = NOT_GIVEN,
+        input_audio_format: Union[Any, NotGiven] = NOT_GIVEN,
+        input_audio_noise_reduction: Union[Any, NotGiven] = NOT_GIVEN,
+        input_audio_transcription: Union[Any, NotGiven] = NOT_GIVEN,
+        modalities: Union[List[Any], NotGiven] = NOT_GIVEN,
+        turn_detection: Union[Any, NotGiven] = NOT_GIVEN,
+        **kwargs,
+    ) -> TranscriptionSession:
+        extra_headers = kwargs.pop("extra_headers", None)
+        extra_query = kwargs.pop("extra_query", None)
+        extra_body = kwargs.pop("extra_body", None)
+        timeout = kwargs.pop("timeout", None)
+        response = self.openai_client.beta.realtime.transcription_sessions.create(
+            client_secret=client_secret,
+            include=include,
+            input_audio_format=input_audio_format,
+            input_audio_noise_reduction=input_audio_noise_reduction,
+            input_audio_transcription=input_audio_transcription,
+            modalities=modalities,
+            turn_detection=turn_detection,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body={**(extra_body or {}), **kwargs},
+            timeout=timeout,
+        )
+        return response
+
+
 class AsyncBetaRealtime(AsyncAPIResource):
     def __init__(self, client: AsyncPortkey) -> None:
         super().__init__(client)
         self.openai_client = client.openai_client
         self.sessions = AsyncBetaSessions(client)
+        self.transcription_sessions = AsyncBetaTranscriptionSessions(client)
 
     def connect(
         self,
@@ -54,6 +96,43 @@ def connect(
         )
 
 
+class AsyncBetaTranscriptionSessions(AsyncAPIResource):
+    def __init__(self, client: AsyncPortkey) -> None:
+        super().__init__(client)
+        self.openai_client = client.openai_client
+
+    async def create(
+        self,
+        *,
+        client_secret: Union[Any, NotGiven] = NOT_GIVEN,
+        include: Union[List[Any], NotGiven] = NOT_GIVEN,
+        input_audio_format: Union[Any, NotGiven] = NOT_GIVEN,
+        input_audio_noise_reduction: Union[Any, NotGiven] = NOT_GIVEN,
+        input_audio_transcription: Union[Any, NotGiven] = NOT_GIVEN,
+        modalities: Union[List[Any], NotGiven] = NOT_GIVEN,
+        turn_detection: Union[Any, NotGiven] = NOT_GIVEN,
+        **kwargs,
+    ) -> TranscriptionSession:
+        extra_headers = kwargs.pop("extra_headers", None)
+        extra_query = kwargs.pop("extra_query", None)
+        extra_body = kwargs.pop("extra_body", None)
+        timeout = kwargs.pop("timeout", None)
+        response = await self.openai_client.beta.realtime.transcription_sessions.create(
+            client_secret=client_secret,
+            include=include,
+            input_audio_format=input_audio_format,
+            input_audio_noise_reduction=input_audio_noise_reduction,
+            input_audio_transcription=input_audio_transcription,
+            modalities=modalities,
+            turn_detection=turn_detection,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body={**(extra_body or {}), **kwargs},
+            timeout=timeout,
+        )
+        return response
+
+
 class BetaSessions(APIResource):
     def __init__(self, client: Portkey) -> None:
         super().__init__(client)