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 @@ -217,12 +217,14 @@ def test_langchain_agent(
217217 assert tx ["contexts" ]["trace" ]["origin" ] == "manual"
218218
219219 chat_spans = list (x for x in tx ["spans" ] if x ["op" ] == "gen_ai.chat" )
220+ invoke_agent_span = next (x for x in tx ["spans" ] if x ["op" ] == "gen_ai.invoke_agent" )
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
224225 assert chat_spans [0 ]["origin" ] == "auto.ai.langchain"
225226 assert chat_spans [1 ]["origin" ] == "auto.ai.langchain"
227+ assert invoke_agent_span ["origin" ] == "auto.ai.langchain"
226228 assert tool_exec_span ["origin" ] == "auto.ai.langchain"
227229
228230 # We can't guarantee anything about the "shape" of the langchain execution graph
You can’t perform that action at this time.
0 commit comments