File tree Expand file tree Collapse file tree
tests/integrations/langchain Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ]
You can’t perform that action at this time.
0 commit comments