File tree Expand file tree Collapse file tree
tests/test_interface/test_agent_integration Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -420,7 +420,7 @@ def agent_node(state: State) -> State:
420420def 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
You can’t perform that action at this time.
0 commit comments