fix: test

Author: k11kirky

posthog/ai/openai/openai_async.py

@@ -50,7 +50,7 @@ async def create(
             posthog_trace_id = uuid.uuid4()
 
         if kwargs.get("stream", False):
-            return self._create_streaming(
+            return await self._create_streaming(
                 posthog_distinct_id,
                 posthog_trace_id,
                 posthog_properties,
@@ -84,14 +84,14 @@ async def _create_streaming(
         start_time = time.time()
         usage_stats: Dict[str, int] = {}
         final_content = []
-        response = super().create(**kwargs)
+        response = await super().create(**kwargs)
 
         async def async_generator():
             nonlocal usage_stats
             nonlocal final_content
 
             try:
-                for chunk in response:
+                async for chunk in response:
                     if hasattr(chunk, "type") and chunk.type == "response.completed":
                         res = chunk.response
                         if res.output and len(res.output) > 0:

posthog/test/ai/openai/test_openai.py

@@ -541,7 +541,6 @@ def test_responses_api(mock_client):
 
         call_args = mock_client.capture.call_args[1]
         props = call_args["properties"]
-        print(props)
         assert props["$ai_provider"] == "openai"
         assert props["$ai_model"] == "gpt-4o-mini"
         assert props["$ai_input"] == [{"role": "user", "content": "Hello"}]