Skip to content

Commit 6fd0c69

Browse files
Merge branch 'webb/langchain/span-origin' into webb/langchain/remove-mocks
2 parents 3ac4da9 + b6e9f70 commit 6fd0c69

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

tests/integrations/langchain/test_langchain.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -336,11 +336,13 @@ def test_langchain_agent(
336336
assert tx["type"] == "transaction"
337337
assert tx["contexts"]["trace"]["origin"] == "manual"
338338

339+
invoke_agent_span = next(x for x in tx["spans"] if x["op"] == "gen_ai.invoke_agent")
339340
chat_spans = list(x for x in tx["spans"] if x["op"] == "gen_ai.chat")
340341
tool_exec_span = next(x for x in tx["spans"] if x["op"] == "gen_ai.execute_tool")
341342

342343
assert len(chat_spans) == 2
343344

345+
assert invoke_agent_span["origin"] == "auto.ai.langchain"
344346
assert chat_spans[0]["origin"] == "auto.ai.langchain"
345347
assert chat_spans[1]["origin"] == "auto.ai.langchain"
346348
assert tool_exec_span["origin"] == "auto.ai.langchain"

0 commit comments

Comments
 (0)