Fix pydantic issue in test

Author: colin-sentry

sentry_sdk/integrations/openai.py

@@ -146,8 +146,7 @@ def new_chat_completion(*args, **kwargs):
             elif hasattr(res, "_iterator"):
                 data_buf: list[list[str]] = []  # one for each choice
 
-                # type: Iterator[ChatCompletionChunk]
-                old_iterator = res._iterator
+                old_iterator = res._iterator  # type: Iterator[ChatCompletionChunk]
 
                 def new_iterator():
                     # type: () -> Iterator[ChatCompletionChunk]

tests/integrations/openai/test_openai.py

@@ -73,14 +73,22 @@ def test_streaming_chat_completion(sentry_init, capture_events):
     returned_stream._iterator = [
         ChatCompletionChunk(
             id="1",
-            choices=[DeltaChoice(index=0, delta=ChoiceDelta(content="hel"))],
+            choices=[
+                DeltaChoice(
+                    index=0, delta=ChoiceDelta(content="hel"), finish_reason=None
+                )
+            ],
             created=100000,
             model="model-id",
             object="chat.completion.chunk",
         ),
         ChatCompletionChunk(
             id="1",
-            choices=[DeltaChoice(index=1, delta=ChoiceDelta(content="lo "))],
+            choices=[
+                DeltaChoice(
+                    index=1, delta=ChoiceDelta(content="lo "), finish_reason=None
+                )
+            ],
             created=100000,
             model="model-id",
             object="chat.completion.chunk",