This commit is contained in:
Chester Curme 2025-05-08 10:45:55 -04:00
parent f23b8008ce
commit 50ce3bf5d9

View File

@ -149,10 +149,13 @@ async def test_async_stream_twice() -> None:
model = ChatAnthropic(model_name=MODEL_NAME, stream_usage=False)
async for token in model.astream("hi"):
assert isinstance(token, AIMessageChunk)
assert token.usage_metadata is None
model = ChatAnthropic(model_name=MODEL_NAME) # type: ignore[call-arg]
assert model.stream_usage
async for token in model.astream("hi", stream_usage=False):
assert isinstance(token, AIMessageChunk)
assert token.usage_metadata is None
async def test_abatch() -> None: