From 6acd55b2160a8d48469bb169b181f316b707caba Mon Sep 17 00:00:00 2001 From: Chester Curme Date: Thu, 8 May 2025 10:50:14 -0400 Subject: [PATCH] x --- .../anthropic/tests/integration_tests/test_chat_models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 dbde0b701a2..73f3c0d02b1 100644 --- a/libs/partners/anthropic/tests/integration_tests/test_chat_models.py +++ b/libs/partners/anthropic/tests/integration_tests/test_chat_models.py @@ -144,11 +144,12 @@ async def test_stream_usage() -> None: async def test_async_stream_twice() -> None: - model = ChatAnthropic(model_name=MODEL_NAME, stream_usage=False) + model = ChatAnthropic(model_name=MODEL_NAME, stream_usage=False) # 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):