update

Author: antonpirker

sentry_sdk/api.py

@@ -498,4 +498,4 @@ def update_current_span(op=None, name=None, attributes=None):
         current_span.description = name
 
     if attributes is not None:
-        current_span.set_data(attributes)
+        current_span.update_data(attributes)

sentry_sdk/tracing.py

@@ -599,25 +599,9 @@ def set_tag(self, key, value):
         # type: (str, Any) -> None
         self._tags[key] = value
 
-    def set_data(self, key=None, value=None):
-        # type: (Optional[Union[str, Dict[str, Any]]], Optional[Any]) -> None
-        """Set data on the span.
-
-        Can be called in two ways:
-        - set_data(key, value) - sets a single key-value pair
-        - set_data({"key": "value"}) - sets multiple key-value pairs from a dict
-        """
-        if key is None:
-            return
-
-        if isinstance(key, dict):
-            # Dictionary calling pattern: set_data({"key": "value"})
-            for k, v in key.items():
-                self._data[k] = v
-
-        elif isinstance(key, str):
-            # Traditional calling pattern: set_data(key, value)
-            self._data[key] = value
+    def set_data(self, key, value):
+        # type: (str, Any) -> None
+        self._data[key] = value
 
     def update_data(self, data):
         # type: (Dict[str, Any]) -> None
@@ -1292,8 +1276,8 @@ def set_tag(self, key, value):
         # type: (str, Any) -> None
         pass
 
-    def set_data(self, key=None, value=None):
-        # type: (Optional[Union[str, Dict[str, Any]]], Optional[Any]) -> None
+    def set_data(self, key, value):
+        # type: (str, Any) -> None
         pass
 
     def update_data(self, data):

sentry_sdk/tracing_utils.py

@@ -1016,7 +1016,7 @@ def _set_input_attributes(span, template, send_pii, name, f, args, kwargs):
             attributes[SPANDATA.GEN_AI_TOOL_DESCRIPTION] = docstring
 
     attributes.update(_get_input_attributes(template, send_pii, args, kwargs))
-    span.set_data(attributes)
+    span.udpate_data(attributes)
 
 
 def _set_output_attributes(span, template, send_pii, result):
@@ -1035,7 +1035,7 @@ def _set_output_attributes(span, template, send_pii, result):
         attributes[SPANDATA.GEN_AI_TOOL_OUTPUT] = result
 
     attributes.update(_get_output_attributes(template, send_pii, result))
-    span.set_data(attributes)
+    span.udpate_data(attributes)
 
 
 def create_span_decorator(template, op=None, name=None, attributes=None):
@@ -1088,7 +1088,7 @@ async def async_wrapper(*args, **kwargs):
                 result = await f(*args, **kwargs)
 
                 _set_output_attributes(span, template, send_pii, result)
-                span.set_data(attributes)
+                span.udpate_data(attributes)
 
                 return result
 
@@ -1128,7 +1128,7 @@ def sync_wrapper(*args, **kwargs):
                 result = f(*args, **kwargs)
 
                 _set_output_attributes(span, template, send_pii, result)
-                span.set_data(attributes)
+                span.udpate_data(attributes)
 
                 return result