agent name

Author: antonpirker

sentry_sdk/integrations/langchain.py

@@ -778,7 +778,11 @@ def new_stream(self, *args, **kwargs):
             return f(self, *args, **kwargs)
 
         try:
-            agent_name = self.agent.runnable.config.get("run_name")
+            agent_name = None
+            if len(args) > 1:
+                agent_name = args[1].get("run_name")
+            if agent_name is None:
+                agent_name = self.agent.runnable.config.get("run_name")
         except Exception:
             agent_name = ""
 
@@ -793,6 +797,9 @@ def new_stream(self, *args, **kwargs):
         span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "invoke_agent")
         span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
 
+        if agent_name != "":
+            span.set_data(SPANDATA.GEN_AI_AGENT_NAME, agent_name)
+
         input = args[0].get("input") if len(args) > 1 else None
         if input is not None:
             set_data_normalized(