port some more attributes

Author: shellmayr

sentry_sdk/consts.py

@@ -807,7 +807,9 @@ class OP:
     GEN_AI_EXECUTE_TOOL = "gen_ai.execute_tool"
     GEN_AI_HANDOFF = "gen_ai.handoff"
     GEN_AI_INVOKE_AGENT = "gen_ai.invoke_agent"
+    GEN_AI_PIPELINE = "gen_ai.pipeline"
     GEN_AI_RESPONSES = "gen_ai.responses"
+    GEN_AI_RUN = "gen_ai.run"
     GRAPHQL_EXECUTE = "graphql.execute"
     GRAPHQL_MUTATION = "graphql.mutation"
     GRAPHQL_PARSE = "graphql.parse"
@@ -833,10 +835,6 @@ class OP:
     HUGGINGFACE_HUB_CHAT_COMPLETIONS_CREATE = (
         "ai.chat_completions.create.huggingface_hub"
     )
-    LANGCHAIN_PIPELINE = "ai.pipeline.langchain"
-    LANGCHAIN_RUN = "ai.run.langchain"
-    LANGCHAIN_TOOL = "ai.tool.langchain"
-    LANGCHAIN_AGENT = "ai.agent.langchain"
     QUEUE_PROCESS = "queue.process"
     QUEUE_PUBLISH = "queue.publish"
     QUEUE_SUBMIT_ARQ = "queue.submit.arq"

sentry_sdk/integrations/langchain.py

@@ -186,7 +186,7 @@ def on_llm_start(
             watched_span = self._create_span(
                 run_id,
                 kwargs.get("parent_run_id"),
-                op=OP.LANGCHAIN_RUN,
+                op=OP.GEN_AI_,
                 name=kwargs.get("name") or "Langchain LLM call",
                 origin=LangchainIntegration.origin,
             )
@@ -309,9 +309,9 @@ def on_chain_start(self, serialized, inputs, *, run_id, **kwargs):
                 run_id,
                 kwargs.get("parent_run_id"),
                 op=(
-                    OP.LANGCHAIN_RUN
+                    OP.GEN_AI_RUN
                     if kwargs.get("parent_run_id") is not None
-                    else OP.LANGCHAIN_PIPELINE
+                    else OP.GEN_AI_PIPELINE
                 ),
                 name=kwargs.get("name") or "Chain execution",
                 origin=LangchainIntegration.origin,
@@ -347,7 +347,7 @@ def on_agent_action(self, action, *, run_id, **kwargs):
             watched_span = self._create_span(
                 run_id,
                 kwargs.get("parent_run_id"),
-                op=OP.LANGCHAIN_AGENT,
+                op=OP.GEN_AI_INVOKE_AGENT,
                 name=action.tool or "AI tool usage",
                 origin=LangchainIntegration.origin,
             )
@@ -384,7 +384,7 @@ def on_tool_start(self, serialized, input_str, *, run_id, **kwargs):
             watched_span = self._create_span(
                 run_id,
                 kwargs.get("parent_run_id"),
-                op=OP.LANGCHAIN_TOOL,
+                op=OP.GEN_AI_EXECUTE_TOOL,
                 name=serialized.get("name") or kwargs.get("name") or "AI tool usage",
                 origin=LangchainIntegration.origin,
             )

tests/integrations/langchain/test_langchain.py

@@ -177,12 +177,12 @@ def test_langchain_agent(
     tx = events[0]
     assert tx["type"] == "transaction"
     chat_spans = list(x for x in tx["spans"] if x["op"] == "gen_ai.chat")
-    tool_exec_span = next(x for x in tx["spans"] if x["op"] == "ai.tool.langchain")
+    tool_exec_span = next(x for x in tx["spans"] if x["op"] == "gen_ai.execute_tool")
 
     assert len(chat_spans) == 2
 
     # We can't guarantee anything about the "shape" of the langchain execution graph
-    assert len(list(x for x in tx["spans"] if x["op"] == "ai.run.langchain")) > 0
+    assert len(list(x for x in tx["spans"] if x["op"] == "gen_ai.run")) > 0
 
     if use_unknown_llm_type:
         assert "gen_ai.usage.input_tokens" in chat_spans[0]["data"]