File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed
tests/integrations/langchain Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff 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"
You can’t perform that action at this time.
0 commit comments