fix: formatting fun

Author: k11kirky

llm_observability_examples.py

@@ -143,20 +143,36 @@ def image_openai_call():
     print(response)
     return response
 
+
 # none instrumented
 async def image_async_openai_call():
-    response = await async_openai_client.images.generate(model="dall-e-3", prompt="A cute baby hedgehog", n=1, size="1024x1024")
+    response = await async_openai_client.images.generate(
+        model="dall-e-3", prompt="A cute baby hedgehog", n=1, size="1024x1024"
+    )
     print(response)
     return response
 
 
 def embedding_openai_call(posthog_distinct_id, posthog_trace_id, posthog_properties):
-    response = openai_client.embeddings.create(input="The hedgehog is cute", model="text-embedding-3-small", posthog_distinct_id=posthog_distinct_id, posthog_trace_id=posthog_trace_id, posthog_properties=posthog_properties)
+    response = openai_client.embeddings.create(
+        input="The hedgehog is cute",
+        model="text-embedding-3-small",
+        posthog_distinct_id=posthog_distinct_id,
+        posthog_trace_id=posthog_trace_id,
+        posthog_properties=posthog_properties,
+    )
     print(response)
     return response
 
+
 async def embedding_async_openai_call(posthog_distinct_id, posthog_trace_id, posthog_properties):
-    response = await async_openai_client.embeddings.create(input="The hedgehog is cute", model="text-embedding-3-small", posthog_distinct_id=posthog_distinct_id, posthog_trace_id=posthog_trace_id, posthog_properties=posthog_properties)
+    response = await async_openai_client.embeddings.create(
+        input="The hedgehog is cute",
+        model="text-embedding-3-small",
+        posthog_distinct_id=posthog_distinct_id,
+        posthog_trace_id=posthog_trace_id,
+        posthog_properties=posthog_properties,
+    )
     print(response)
     return response
 

posthog/ai/openai/openai.py

@@ -234,4 +234,4 @@ def create(
                 properties=event_properties,
             )
 
-        return response
+        return response

posthog/ai/openai/openai_async.py

@@ -31,6 +31,7 @@ def __init__(self, posthog_client: PostHogClient, **kwargs):
         self.chat = WrappedChat(self)
         self.embeddings = WrappedEmbeddings(self)
 
+
 class WrappedChat(openai.resources.chat.AsyncChat):
     _client: AsyncOpenAI
 
@@ -168,6 +169,7 @@ def _capture_streaming_event(
                 properties=event_properties,
             )
 
+
 class WrappedEmbeddings(openai.resources.embeddings.AsyncEmbeddings):
     _client: AsyncOpenAI