wip

Author: shellmayr

sentry_sdk/integrations/cohere/v1.py

@@ -142,6 +142,7 @@ def new_chat(*args, **kwargs):
             set_data_normalized(span, SPANDATA.GEN_AI_OPERATION_NAME, "chat")
             if model:
                 set_data_normalized(span, SPANDATA.GEN_AI_REQUEST_MODEL, model)
+                set_data_normalized(span, SPANDATA.GEN_AI_RESPONSE_MODEL, model)
 
             if should_send_default_pii() and integration.include_prompts:
                 messages = []

sentry_sdk/integrations/cohere/v2.py

@@ -186,6 +186,7 @@ def new_chat(*args, **kwargs):
             set_data_normalized(span, SPANDATA.GEN_AI_OPERATION_NAME, "chat")
             if model:
                 set_data_normalized(span, SPANDATA.GEN_AI_REQUEST_MODEL, model)
+                set_data_normalized(span, SPANDATA.GEN_AI_RESPONSE_MODEL, model)
 
             if should_send_default_pii() and integration.include_prompts:
                 messages = _extract_messages_v2(kwargs.get("messages", []))