Skip to content

Commit baa8a99

Browse files
committed
fix test
1 parent 2f62e58 commit baa8a99

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/integrations/langchain/test_langchain.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -176,9 +176,7 @@ def test_langchain_agent(
176176

177177
tx = events[0]
178178
assert tx["type"] == "transaction"
179-
chat_spans = list(
180-
x for x in tx["spans"] if x["op"] == "ai.chat_completions.create.langchain"
181-
)
179+
chat_spans = list(x for x in tx["spans"] if x["op"] == "gen_ai.chat")
182180
tool_exec_span = next(x for x in tx["spans"] if x["op"] == "ai.tool.langchain")
183181

184182
assert len(chat_spans) == 2

0 commit comments

Comments
 (0)