Skip to content

Commit 82c1de4

Browse files
feat(langchain): Record run_name as gen_ai.function_id on Invoke Agent Spans (#5926)
Set `run_name` as the `gen_ai.function_id` attribute instead of `gen_ai.agent.name`. Add tests for `AgentExecutor.invoke()` based on the existing `AgentExecutor.stream()` test. Add tests that use `create_openai_tools_agent().with_config()` to test all branches for extracting the `run_name` LangChain attribute.
1 parent c0b3869 commit 82c1de4

2 files changed

Lines changed: 472 additions & 3 deletions

File tree

sentry_sdk/integrations/langchain.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -975,7 +975,7 @@ def new_invoke(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
975975
origin=LangchainIntegration.origin,
976976
) as span:
977977
if run_name:
978-
span.set_data(SPANDATA.GEN_AI_AGENT_NAME, run_name)
978+
span.set_data(SPANDATA.GEN_AI_FUNCTION_ID, run_name)
979979

980980
span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "invoke_agent")
981981
span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, False)
@@ -1035,7 +1035,7 @@ def new_stream(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
10351035
span.__enter__()
10361036

10371037
if run_name:
1038-
span.set_data(SPANDATA.GEN_AI_AGENT_NAME, run_name)
1038+
span.set_data(SPANDATA.GEN_AI_FUNCTION_ID, run_name)
10391039

10401040
span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "invoke_agent")
10411041
span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)

0 commit comments

Comments
 (0)