Skip to content

Commit b6e9f70

Browse files
add agent invocation origin assertion
1 parent c7552cf commit b6e9f70

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/integrations/langchain/test_langchain.py

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

219+
invoke_agent_span = next(x for x in tx["spans"] if x["op"] == "gen_ai.invoke_agent")
219220
chat_spans = list(x for x in tx["spans"] if x["op"] == "gen_ai.chat")
220221
tool_exec_span = next(x for x in tx["spans"] if x["op"] == "gen_ai.execute_tool")
221222

222223
assert len(chat_spans) == 2
223224

225+
assert invoke_agent_span["origin"] == "auto.ai.langchain"
224226
assert chat_spans[0]["origin"] == "auto.ai.langchain"
225227
assert chat_spans[1]["origin"] == "auto.ai.langchain"
226228
assert tool_exec_span["origin"] == "auto.ai.langchain"

0 commit comments

Comments
 (0)