format

Author: shellmayr

sentry_sdk/integrations/cohere.py

@@ -167,16 +167,16 @@ def new_chat(*args: "Any", **kwargs: "Any") -> "Any":
             if should_send_default_pii() and integration.include_prompts:
                 messages = []
                 for x in kwargs.get("chat_history", []):
-                    messages.append({
-                        "role": getattr(x, "role", "").lower(),
-                        "content": getattr(x, "message", ""),
-                    })
+                    messages.append(
+                        {
+                            "role": getattr(x, "role", "").lower(),
+                            "content": getattr(x, "message", ""),
+                        }
+                    )
                 messages.append({"role": "user", "content": message})
                 messages = normalize_message_roles(messages)
                 scope = sentry_sdk.get_current_scope()
-                messages_data = truncate_and_annotate_messages(
-                    messages, span, scope
-                )
+                messages_data = truncate_and_annotate_messages(messages, span, scope)
                 if messages_data is not None:
                     set_data_normalized(
                         span,
@@ -263,7 +263,9 @@ def new_embed(*args: "Any", **kwargs: "Any") -> "Any":
                     )
 
             if "model" in kwargs:
-                set_data_normalized(span, SPANDATA.GEN_AI_REQUEST_MODEL, kwargs["model"])
+                set_data_normalized(
+                    span, SPANDATA.GEN_AI_REQUEST_MODEL, kwargs["model"]
+                )
             try:
                 res = f(*args, **kwargs)
             except Exception as e:

sentry_sdk/integrations/cohere_v2.py

@@ -87,7 +87,11 @@ def _extract_messages_v2(messages):
             text = content
         elif isinstance(content, list):
             text = " ".join(
-                (item.get("text", "") if isinstance(item, dict) else getattr(item, "text", ""))
+                (
+                    item.get("text", "")
+                    if isinstance(item, dict)
+                    else getattr(item, "text", "")
+                )
                 for item in content
                 if (isinstance(item, dict) and "text" in item) or hasattr(item, "text")
             )
@@ -125,9 +129,7 @@ def collect_v2_response_fields(span, res, include_pii):
                 and res.message.content
             ):
                 texts = [
-                    item.text
-                    for item in res.message.content
-                    if hasattr(item, "text")
+                    item.text for item in res.message.content if hasattr(item, "text")
                 ]
                 if texts:
                     set_data_normalized(span, SPANDATA.GEN_AI_RESPONSE_TEXT, texts)
@@ -189,9 +191,7 @@ def new_chat(*args, **kwargs):
                 messages = _extract_messages_v2(kwargs.get("messages", []))
                 messages = normalize_message_roles(messages)
                 scope = sentry_sdk.get_current_scope()
-                messages_data = truncate_and_annotate_messages(
-                    messages, span, scope
-                )
+                messages_data = truncate_and_annotate_messages(messages, span, scope)
                 if messages_data is not None:
                     set_data_normalized(
                         span,
@@ -228,9 +228,7 @@ def new_iterator():
                                 msg = getattr(x.delta, "message", None)
                                 if msg is not None:
                                     content = getattr(msg, "content", None)
-                                    if content is not None and hasattr(
-                                        content, "text"
-                                    ):
+                                    if content is not None and hasattr(content, "text"):
                                         collected_text.append(content.text)
 
                             if isinstance(x, MessageEndV2ChatStreamResponse):