Fix multiple stream_options

Author: Twixes

posthog/ai/providers/openai/openai.py

@@ -81,8 +81,10 @@ def _create_streaming(
         start_time = time.time()
         usage_stats: Dict[str, int] = {}
         accumulated_content = []
-        stream_options = {"include_usage": True}
-        response = self._client.chat.completions.create(**kwargs, stream_options=stream_options)
+        if "stream_options" not in kwargs:
+            kwargs["stream_options"] = {}
+        kwargs["stream_options"]["include_usage"] = True
+        response = super().create(**kwargs)
 
         def generator():
             nonlocal usage_stats

posthog/ai/providers/openai/openai_async.py

@@ -82,8 +82,10 @@ async def _create_streaming(
         start_time = time.time()
         usage_stats: Dict[str, int] = {}
         accumulated_content = []
-        stream_options = {"include_usage": True}
-        response = await self._client.chat.completions.create(**kwargs, stream_options=stream_options)
+        if "stream_options" not in kwargs:
+            kwargs["stream_options"] = {}
+        kwargs["stream_options"]["include_usage"] = True
+        response = await super().create(**kwargs)
 
         async def async_generator():
             nonlocal usage_stats, accumulated_content