feat(audio): add tracking for audio transcriptions in OpenAI client

posthog/ai/openai/openai.py

@@ -51,6 +51,7 @@ def __init__(self, posthog_client: Optional[PostHogClient] = None, **kwargs):
         self._original_embeddings = getattr(self, "embeddings", None)
         self._original_beta = getattr(self, "beta", None)
         self._original_responses = getattr(self, "responses", None)
+        self._original_audio = getattr(self, "audio", None)
 
         # Replace with wrapped versions (only if originals exist)
         if self._original_chat is not None:
@@ -65,6 +66,9 @@ def __init__(self, posthog_client: Optional[PostHogClient] = None, **kwargs):
         if self._original_responses is not None:
             self.responses = WrappedResponses(self, self._original_responses)
 
+        if self._original_audio is not None:
+            self.audio = WrappedAudio(self, self._original_audio)
+
 
 class WrappedResponses:
     """Wrapper for OpenAI responses that tracks usage in PostHog."""
@@ -534,6 +538,136 @@ def create(
         return response
 
 
+class WrappedAudio:
+    """Wrapper for OpenAI audio that tracks usage in PostHog."""
+
+    def __init__(self, client: OpenAI, original_audio):
+        self._client = client
+        self._original = original_audio
+
+    def __getattr__(self, name):
+        """Fallback to original audio object for any methods we don't explicitly handle."""
+        return getattr(self._original, name)
+
+    @property
+    def transcriptions(self):
+        return WrappedTranscriptions(self._client, self._original.transcriptions)
+
+
+class WrappedTranscriptions:
+    """Wrapper for OpenAI audio transcriptions that tracks usage in PostHog."""
+
+    def __init__(self, client: OpenAI, original_transcriptions):
+        self._client = client
+        self._original = original_transcriptions
+
+    def __getattr__(self, name):
+        """Fallback to original transcriptions object for any methods we don't explicitly handle."""
+        return getattr(self._original, name)
+
+    def create(
+        self,
+        posthog_distinct_id: Optional[str] = None,
+        posthog_trace_id: Optional[str] = None,
+        posthog_properties: Optional[Dict[str, Any]] = None,
+        posthog_privacy_mode: bool = False,
+        posthog_groups: Optional[Dict[str, Any]] = None,
+        **kwargs: Any,
+    ):
+        """
+        Create a transcription using OpenAI's 'audio.transcriptions.create' method,
+        and track usage in PostHog.
+
+        Args:
+            posthog_distinct_id: Optional ID to associate with the usage event.
+            posthog_trace_id: Optional trace UUID for linking events.
+            posthog_properties: Optional dictionary of extra properties to include in the event.
+            posthog_privacy_mode: Whether to anonymize the input and output.
+            posthog_groups: Optional dictionary of groups to associate with the event.
+            **kwargs: Any additional parameters for the OpenAI Transcriptions API.
+
+        Returns:
+            The response from OpenAI's audio.transcriptions.create call.
+        """
+        if posthog_trace_id is None:
+            posthog_trace_id = str(uuid.uuid4())
+
+        # Extract file info before API call
+        file_obj = kwargs.get("file")
+        file_name = getattr(file_obj, "name", None) if file_obj else None
+
+        start_time = time.time()
+        error = None
+        http_status = 200
+        response = None
+
+        try:
+            response = self._original.create(**kwargs)
+        except Exception as e:
+            error = e
+            http_status = getattr(e, "status_code", 0)
+
+        end_time = time.time()
+        latency = end_time - start_time
+
+        # Extract transcription output
+        output_text = getattr(response, "text", None) if response else None
+
+        # Extract duration if available (verbose_json response format)
+        duration = getattr(response, "duration", None) if response else None
+
+        # Build event properties
+        event_properties = {
+            "$ai_provider": "openai",
+            "$ai_model": kwargs.get("model"),
+            "$ai_input": with_privacy_mode(
+                self._client._ph_client,
+                posthog_privacy_mode,
+                file_name,
+            ),
+            "$ai_output_text": with_privacy_mode(
+                self._client._ph_client,
+                posthog_privacy_mode,
+                output_text,
+            ),
+            "$ai_http_status": http_status,
+            "$ai_latency": latency,
+            "$ai_trace_id": posthog_trace_id,
+            "$ai_base_url": str(self._client.base_url),
+            **(posthog_properties or {}),
+        }
+
+        # Add optional properties
+        if kwargs.get("language"):
+            event_properties["$ai_language"] = kwargs.get("language")
+
+        if duration is not None:
+            event_properties["$ai_audio_duration"] = duration
+
+        # Add error properties if an error occurred
+        if error is not None:
+            event_properties["$ai_is_error"] = True
+            event_properties["$ai_error"] = str(error)
+
+        if posthog_distinct_id is None:
+            event_properties["$process_person_profile"] = False
+
+        # Capture event
+        if hasattr(self._client._ph_client, "capture"):
+            self._client._ph_client.capture(
+                distinct_id=posthog_distinct_id or posthog_trace_id,
+                event="$ai_transcription",
+                properties=event_properties,
+                groups=posthog_groups,
+            )
+
+        # Re-raise the error after capturing the event
+        if error is not None:
+            raise error
+
+        return response
+
+
 class WrappedBeta:
     """Wrapper for OpenAI beta features that tracks usage in PostHog."""
 

posthog/ai/openai/openai_async.py

@@ -54,6 +54,7 @@ def __init__(self, posthog_client: Optional[PostHogClient] = None, **kwargs):
         self._original_embeddings = getattr(self, "embeddings", None)
         self._original_beta = getattr(self, "beta", None)
         self._original_responses = getattr(self, "responses", None)
+        self._original_audio = getattr(self, "audio", None)
 
         # Replace with wrapped versions (only if originals exist)
         if self._original_chat is not None:
@@ -68,6 +69,9 @@ def __init__(self, posthog_client: Optional[PostHogClient] = None, **kwargs):
         if self._original_responses is not None:
             self.responses = WrappedResponses(self, self._original_responses)
 
+        if self._original_audio is not None:
+            self.audio = WrappedAudio(self, self._original_audio)
+
 
 class WrappedResponses:
     """Async wrapper for OpenAI responses that tracks usage in PostHog."""
@@ -589,6 +593,136 @@ async def create(
         return response
 
 
+class WrappedAudio:
+    """Async wrapper for OpenAI audio that tracks usage in PostHog."""
+
+    def __init__(self, client: AsyncOpenAI, original_audio):
+        self._client = client
+        self._original = original_audio
+
+    def __getattr__(self, name):
+        """Fallback to original audio object for any methods we don't explicitly handle."""
+        return getattr(self._original, name)
+
+    @property
+    def transcriptions(self):
+        return WrappedTranscriptions(self._client, self._original.transcriptions)
+
+
+class WrappedTranscriptions:
+    """Async wrapper for OpenAI audio transcriptions that tracks usage in PostHog."""
+
+    def __init__(self, client: AsyncOpenAI, original_transcriptions):
+        self._client = client
+        self._original = original_transcriptions
+
+    def __getattr__(self, name):
+        """Fallback to original transcriptions object for any methods we don't explicitly handle."""
+        return getattr(self._original, name)
+
+    async def create(
+        self,
+        posthog_distinct_id: Optional[str] = None,
+        posthog_trace_id: Optional[str] = None,
+        posthog_properties: Optional[Dict[str, Any]] = None,
+        posthog_privacy_mode: bool = False,
+        posthog_groups: Optional[Dict[str, Any]] = None,
+        **kwargs: Any,
+    ):
+        """
+        Create a transcription using OpenAI's 'audio.transcriptions.create' method,
+        and track usage in PostHog.
+
+        Args:
+            posthog_distinct_id: Optional ID to associate with the usage event.
+            posthog_trace_id: Optional trace UUID for linking events.
+            posthog_properties: Optional dictionary of extra properties to include in the event.
+            posthog_privacy_mode: Whether to anonymize the input and output.
+            posthog_groups: Optional dictionary of groups to associate with the event.
+            **kwargs: Any additional parameters for the OpenAI Transcriptions API.
+
+        Returns:
+            The response from OpenAI's audio.transcriptions.create call.
+        """
+        if posthog_trace_id is None:
+            posthog_trace_id = str(uuid.uuid4())
+
+        # Extract file info before API call
+        file_obj = kwargs.get("file")
+        file_name = getattr(file_obj, "name", None) if file_obj else None
+
+        start_time = time.time()
+        error = None
+        http_status = 200
+        response = None
+
+        try:
+            response = await self._original.create(**kwargs)
+        except Exception as e:
+            error = e
+            http_status = getattr(e, "status_code", 0)
+
+        end_time = time.time()
+        latency = end_time - start_time
+
+        # Extract transcription output
+        output_text = getattr(response, "text", None) if response else None
+
+        # Extract duration if available (verbose_json response format)
+        duration = getattr(response, "duration", None) if response else None
+
+        # Build event properties
+        event_properties = {
+            "$ai_provider": "openai",
+            "$ai_model": kwargs.get("model"),
+            "$ai_input": with_privacy_mode(
+                self._client._ph_client,
+                posthog_privacy_mode,
+                file_name,
+            ),
+            "$ai_output_text": with_privacy_mode(
+                self._client._ph_client,
+                posthog_privacy_mode,
+                output_text,
+            ),
+            "$ai_http_status": http_status,
+            "$ai_latency": latency,
+            "$ai_trace_id": posthog_trace_id,
+            "$ai_base_url": str(self._client.base_url),
+            **(posthog_properties or {}),
+        }
+
+        # Add optional properties
+        if kwargs.get("language"):
+            event_properties["$ai_language"] = kwargs.get("language")
+
+        if duration is not None:
+            event_properties["$ai_audio_duration"] = duration
+
+        # Add error properties if an error occurred
+        if error is not None:
+            event_properties["$ai_is_error"] = True
+            event_properties["$ai_error"] = str(error)
+
+        if posthog_distinct_id is None:
+            event_properties["$process_person_profile"] = False
+
+        # Capture event
+        if hasattr(self._client._ph_client, "capture"):
+            self._client._ph_client.capture(
+                distinct_id=posthog_distinct_id or posthog_trace_id,
+                event="$ai_transcription",
+                properties=event_properties,
+                groups=posthog_groups,
+            )
+
+        # Re-raise the error after capturing the event
+        if error is not None:
+            raise error
+
+        return response
+
+
 class WrappedBeta:
     """Async wrapper for OpenAI beta features that tracks usage in PostHog."""