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 @@ -448,11 +448,13 @@ def test_langchain_openai_tools_agent(
448448 assert tx ["type" ] == "transaction"
449449 assert tx ["contexts" ]["trace" ]["origin" ] == "manual"
450450
451+ invoke_agent_span = next (x for x in tx ["spans" ] if x ["op" ] == "gen_ai.invoke_agent" )
451452 chat_spans = list (x for x in tx ["spans" ] if x ["op" ] == "gen_ai.chat" )
452453 tool_exec_span = next (x for x in tx ["spans" ] if x ["op" ] == "gen_ai.execute_tool" )
453454
454455 assert len (chat_spans ) == 2
455456
457+ assert invoke_agent_span ["origin" ] == "auto.ai.langchain"
456458 assert chat_spans [0 ]["origin" ] == "auto.ai.langchain"
457459 assert chat_spans [1 ]["origin" ] == "auto.ai.langchain"
458460 assert tool_exec_span ["origin" ] == "auto.ai.langchain"
You can’t perform that action at this time.
0 commit comments