remove unneeded default

DataDog · Jan 24, 2025 · 3e6a796 · 3e6a796
1 parent 713f2e7
commit 3e6a796
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ddtrace/contrib/internal/openai/utils.py b/ddtrace/contrib/internal/openai/utils.py
@@ -89,7 +89,7 @@ def _extract_token_chunk(self, chunk):
         """Attempt to extract the token chunk (last chunk in the stream) from the streamed response."""
         if not self._dd_span._get_ctx_item("_dd.auto_extract_token_chunk"):
             return
-        choices = getattr(chunk, "choices", None)
+        choices = getattr(chunk, "choices")
         if not choices:
             return
         choice = choices[0]
@@ -155,7 +155,7 @@ async def _extract_token_chunk(self, chunk):
         """Attempt to extract the token chunk (last chunk in the stream) from the streamed response."""
         if not self._dd_span._get_ctx_item("_dd.auto_extract_token_chunk"):
             return
-        choices = getattr(chunk, "choices", None)
+        choices = getattr(chunk, "choices")
         if not choices:
             return
         choice = choices[0]