fix: async + remove log

Author: k11kirky

posthog/ai/openai/openai.py

@@ -280,7 +280,6 @@ def parse(
         posthog_groups: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ):
-        print("kwargs", kwargs)
         return call_llm_and_track_usage(
             posthog_distinct_id,
             self._client._ph_client,

posthog/ai/openai/openai_async.py

@@ -30,6 +30,7 @@ def __init__(self, posthog_client: PostHogClient, **kwargs):
         self._ph_client = posthog_client
         self.chat = WrappedChat(self)
         self.embeddings = WrappedEmbeddings(self)
+        self.beta = WrappedBeta(self)
 
 
 class WrappedChat(openai.resources.chat.AsyncChat):
@@ -250,26 +251,26 @@ async def create(
         return response
 
 
-class WrappedBeta(openai.resources.beta.Beta):
+class WrappedBeta(openai.resources.beta.AsyncBeta):
     _client: AsyncOpenAI
 
     @property
     def chat(self):
         return WrappedBetaChat(self._client)
 
 
-class WrappedBetaChat(openai.resources.beta.chat.Chat):
+class WrappedBetaChat(openai.resources.beta.chat.AsyncChat):
     _client: AsyncOpenAI
 
     @property
     def completions(self):
         return WrappedBetaCompletions(self._client)
 
 
-class WrappedBetaCompletions(openai.resources.beta.chat.completions.Completions):
+class WrappedBetaCompletions(openai.resources.beta.chat.completions.AsyncCompletions):
     _client: AsyncOpenAI
 
-    def parse(
+    async def parse(
         self,
         posthog_distinct_id: Optional[str] = None,
         posthog_trace_id: Optional[str] = None,
@@ -278,7 +279,7 @@ def parse(
         posthog_groups: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ):
-        return call_llm_and_track_usage_async(
+        return await call_llm_and_track_usage_async(
             posthog_distinct_id,
             self._client._ph_client,
             "openai",