github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-14 15:16:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
Chester Curme 2025-05-08 13:01:28 -04:00
parent e01c52d2d0
commit 3516c2c394
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libs/partners/anthropic/tests/integration_tests/test_chat_models.py
View File

@ -38,17 +38,12 @@ async def test_astream() -> None:
async def test_stream_usage() -> 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"): async for token in model.astream("hi"):
assert isinstance(token, AIMessageChunk) assert isinstance(token, AIMessageChunk)
assert token.usage_metadata is None
# check we override with kwarg async for token in model.astream("hi"):
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 isinstance(token, AIMessageChunk)
assert token.usage_metadata is None
async def test_async_stream_twice() -> None: async def test_async_stream_twice() -> None: