Add tags + data passing functionality to @ai_track (#3071)

Co-authored-by: Anton Pirker <[email protected]>

Author: colin-sentry

sentry_sdk/ai/monitoring.py

@@ -31,7 +31,12 @@ def wrapped(*args, **kwargs):
             # type: (Any, Any) -> Any
             curr_pipeline = _ai_pipeline_name.get()
             op = span_kwargs.get("op", "ai.run" if curr_pipeline else "ai.pipeline")
+
             with start_span(description=description, op=op, **span_kwargs) as span:
+                for k, v in kwargs.pop("sentry_tags", {}).items():
+                    span.set_tag(k, v)
+                for k, v in kwargs.pop("sentry_data", {}).items():
+                    span.set_data(k, v)
                 if curr_pipeline:
                     span.set_data("ai.pipeline.name", curr_pipeline)
                     return f(*args, **kwargs)