Skip to content

Commit 1b4b2ba

Browse files
do not gate token assertions
1 parent 5d2353a commit 1b4b2ba

1 file changed

Lines changed: 6 additions & 10 deletions

File tree

tests/integrations/langchain/test_langchain.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -333,17 +333,13 @@ def test_tool_execution_span(
333333
assert chat_spans[1]["origin"] == "auto.ai.langchain"
334334
assert tool_exec_span["origin"] == "auto.ai.langchain"
335335

336-
# Token usage is only available in newer versions of langchain (v0.2+)
337-
# where usage_metadata is supported on AIMessageChunk
338-
if "gen_ai.usage.input_tokens" in chat_spans[0]["data"]:
339-
assert chat_spans[0]["data"]["gen_ai.usage.input_tokens"] == 142
340-
assert chat_spans[0]["data"]["gen_ai.usage.output_tokens"] == 50
341-
assert chat_spans[0]["data"]["gen_ai.usage.total_tokens"] == 192
336+
assert chat_spans[0]["data"]["gen_ai.usage.input_tokens"] == 142
337+
assert chat_spans[0]["data"]["gen_ai.usage.output_tokens"] == 50
338+
assert chat_spans[0]["data"]["gen_ai.usage.total_tokens"] == 192
342339

343-
if "gen_ai.usage.input_tokens" in chat_spans[1]["data"]:
344-
assert chat_spans[1]["data"]["gen_ai.usage.input_tokens"] == 89
345-
assert chat_spans[1]["data"]["gen_ai.usage.output_tokens"] == 28
346-
assert chat_spans[1]["data"]["gen_ai.usage.total_tokens"] == 117
340+
assert chat_spans[1]["data"]["gen_ai.usage.input_tokens"] == 89
341+
assert chat_spans[1]["data"]["gen_ai.usage.output_tokens"] == 28
342+
assert chat_spans[1]["data"]["gen_ai.usage.total_tokens"] == 117
347343

348344
if send_default_pii and include_prompts:
349345
assert "word" in tool_exec_span["data"][SPANDATA.GEN_AI_TOOL_INPUT]

0 commit comments

Comments
 (0)