Run black

Author: Twixes

posthog/ai/providers/openai/openai.py

@@ -7,9 +7,7 @@
 try:
     import openai
 except ImportError:
-    raise ModuleNotFoundError(
-        "Please install the OpenAI SDK to use this feature: 'pip install openai'"
-    )
+    raise ModuleNotFoundError("Please install the OpenAI SDK to use this feature: 'pip install openai'")
 
 from posthog.ai.utils import call_llm_and_track_usage, get_model_params
 from posthog.client import Client as PostHogClient
@@ -84,9 +82,7 @@ def _create_streaming(
         usage_stats: Dict[str, int] = {}
         accumulated_content = []
         stream_options = {"include_usage": True}
-        response = self._client.chat.completions.create(
-            **kwargs, stream_options=stream_options
-        )
+        response = self._client.chat.completions.create(**kwargs, stream_options=stream_options)
 
         def generator():
             nonlocal usage_stats
@@ -147,9 +143,7 @@ def _capture_streaming_event(
                     }
                 ]
             },
-            "$ai_request_url": str(
-                self._client.base_url.join("chat/completions")
-            ),
+            "$ai_request_url": str(self._client.base_url.join("chat/completions")),
             "$ai_http_status": 200,
             "$ai_input_tokens": usage_stats.get("prompt_tokens", 0),
             "$ai_output_tokens": usage_stats.get("completion_tokens", 0),

posthog/ai/providers/openai/openai_async.py

@@ -7,9 +7,7 @@
 try:
     import openai
 except ImportError:
-    raise ModuleNotFoundError(
-        "Please install the OpenAI SDK to use this feature: 'pip install openai'"
-    )
+    raise ModuleNotFoundError("Please install the OpenAI SDK to use this feature: 'pip install openai'")
 
 from posthog.ai.utils import call_llm_and_track_usage_async, get_model_params
 from posthog.client import Client as PostHogClient
@@ -85,9 +83,7 @@ async def _create_streaming(
         usage_stats: Dict[str, int] = {}
         accumulated_content = []
         stream_options = {"include_usage": True}
-        response = await self._client.chat.completions.create(
-            **kwargs, stream_options=stream_options
-        )
+        response = await self._client.chat.completions.create(**kwargs, stream_options=stream_options)
 
         async def async_generator():
             nonlocal usage_stats, accumulated_content
@@ -153,9 +149,7 @@ def _capture_streaming_event(
             "$ai_latency": latency,
             "$ai_trace_id": posthog_trace_id,
             "$ai_posthog_properties": posthog_properties,
-            "$ai_request_url": str(
-                self._client.base_url.join("chat/completions")
-            ),
+            "$ai_request_url": str(self._client.base_url.join("chat/completions")),
         }
 
         if hasattr(self._client._ph_client, "capture"):