use span.set_data & move system & op name out of try block to capture in exceptions

Author: shellmayr

sentry_sdk/integrations/cohere.py

@@ -152,6 +152,11 @@ def new_chat(*args: "Any", **kwargs: "Any") -> "Any":
             origin=CohereIntegration.origin,
         )
         span.__enter__()
+
+        with capture_internal_exceptions():
+            span.set_data(SPANDATA.GEN_AI_SYSTEM, "cohere")
+            span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "chat")
+
         try:
             res = f(*args, **kwargs)
         except Exception as e:
@@ -162,9 +167,6 @@ def new_chat(*args: "Any", **kwargs: "Any") -> "Any":
             reraise(*exc_info)
 
         with capture_internal_exceptions():
-            span.set_data(SPANDATA.GEN_AI_SYSTEM, "cohere")
-            span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "chat")
-
             if should_send_default_pii() and integration.include_prompts:
                 messages = []
                 for x in kwargs.get("chat_history", []):
@@ -241,8 +243,8 @@ def new_embed(*args: "Any", **kwargs: "Any") -> "Any":
             name=f"embeddings {model}".strip(),
             origin=CohereIntegration.origin,
         ) as span:
-            set_data_normalized(span, SPANDATA.GEN_AI_SYSTEM, "cohere")
-            set_data_normalized(span, SPANDATA.GEN_AI_OPERATION_NAME, "embeddings")
+            span.set_data(SPANDATA.GEN_AI_SYSTEM, "cohere")
+            span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "embeddings")
 
             if "texts" in kwargs and (
                 should_send_default_pii() and integration.include_prompts