Skip to content

Commit b5c6e37

Browse files
committed
fix tests
1 parent 1140d93 commit b5c6e37

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integrations/langgraph/test_langgraph.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ def original_invoke(self, *args, **kwargs):
236236
assert len(invoke_spans) == 1
237237

238238
invoke_span = invoke_spans[0]
239-
assert "invoke_agent invoke test_graph" in invoke_span["description"]
239+
assert invoke_span["description"] == "invoke_agent"
240240
assert invoke_span["origin"] == "auto.ai.langgraph"
241241
assert invoke_span["data"][SPANDATA.GEN_AI_OPERATION_NAME] == "invoke_agent"
242242
assert invoke_span["data"][SPANDATA.GEN_AI_PIPELINE_NAME] == "test_graph"
@@ -656,7 +656,7 @@ def original_invoke(self, *args, **kwargs):
656656
invoke_span = invoke_spans[0]
657657

658658
if graph_name and graph_name.strip():
659-
assert f"invoke_agent invoke {graph_name}" in invoke_span["description"]
659+
assert invoke_span["description"] == "invoke_agent"
660660
assert invoke_span["data"][SPANDATA.GEN_AI_PIPELINE_NAME] == graph_name
661661
assert invoke_span["data"][SPANDATA.GEN_AI_AGENT_NAME] == graph_name
662662
else:

0 commit comments

Comments
 (0)