diff --git a/libs/partners/anthropic/tests/integration_tests/test_chat_models.py b/libs/partners/anthropic/tests/integration_tests/test_chat_models.py index 43b50b059a9..caab0ae3202 100644 --- a/libs/partners/anthropic/tests/integration_tests/test_chat_models.py +++ b/libs/partners/anthropic/tests/integration_tests/test_chat_models.py @@ -38,17 +38,12 @@ async def test_astream() -> None: async def test_stream_usage() -> None: - model = ChatAnthropic(model_name=MODEL_NAME, stream_usage=False) # type: ignore[call-arg] + model = ChatAnthropic(model_name=MODEL_NAME) # type: ignore[call-arg] async for token in model.astream("hi"): assert isinstance(token, AIMessageChunk) - assert token.usage_metadata is None - # check we override with kwarg - model = ChatAnthropic(model_name=MODEL_NAME) # type: ignore[call-arg] - assert model.stream_usage - async for token in model.astream("hi", stream_usage=False): + async for token in model.astream("hi"): assert isinstance(token, AIMessageChunk) - assert token.usage_metadata is None async def test_async_stream_twice() -> None: