fix(cohere): Response streaming attribute

Author: alexander-alderman-webb

sentry_sdk/integrations/cohere.py

@@ -181,7 +181,7 @@ def new_chat(*args: "Any", **kwargs: "Any") -> "Any":
             for k, v in COLLECTED_CHAT_PARAMS.items():
                 if k in kwargs:
                     set_data_normalized(span, v, kwargs[k])
-            set_data_normalized(span, SPANDATA.AI_STREAMING, False)
+            span.set_data(SPANDATA.AI_STREAMING, streaming)
 
             if streaming:
                 old_iterator = res

tests/integrations/cohere/test_cohere.py

@@ -54,6 +54,7 @@ def test_nonstreaming_chat(
     assert tx["type"] == "transaction"
     span = tx["spans"][0]
     assert span["op"] == "ai.chat_completions.create.cohere"
+    assert span["data"][SPANDATA.AI_STREAMING] is False
     assert span["data"][SPANDATA.AI_MODEL_ID] == "some-model"
 
     if send_default_pii and include_prompts:
@@ -131,6 +132,7 @@ def test_streaming_chat(sentry_init, capture_events, send_default_pii, include_p
     assert tx["type"] == "transaction"
     span = tx["spans"][0]
     assert span["op"] == "ai.chat_completions.create.cohere"
+    assert span["data"][SPANDATA.AI_STREAMING] is True
     assert span["data"][SPANDATA.AI_MODEL_ID] == "some-model"
 
     if send_default_pii and include_prompts: