Skip to content

Commit aaf1662

Browse files
committed
fixed type hinting issue
1 parent 4b235d4 commit aaf1662

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/test_interface/test_agent_integration/test_langgraph_integration.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -420,7 +420,7 @@ def agent_node(state: State) -> State:
420420
def test_langgraph_adapter_no_cost_without_model_id():
421421
"""Test that LangGraph adapter cannot auto-detect model_id (by design)."""
422422
from maseval.interface.agents.langgraph import LangGraphAgentAdapter
423-
from maseval.core.usage import StaticPricingCalculator
423+
from maseval.core.usage import TokenUsage as MasevalTokenUsage, StaticPricingCalculator
424424
from langgraph.graph import StateGraph, END
425425
from typing_extensions import TypedDict
426426
from langchain_core.messages import AIMessage
@@ -448,5 +448,6 @@ def agent_node(state: State) -> State:
448448
adapter.run("Test")
449449

450450
usage = adapter.gather_usage()
451+
assert isinstance(usage, MasevalTokenUsage)
451452
assert usage.cost == 0.0
452453
assert usage.input_tokens == 100

0 commit comments

Comments
 (0)