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

test stream twice

Browse Source
This commit is contained in:
Chester Curme 2025-05-08 10:35:31 -04:00
parent 2d202f9762
commit ad68ddbf40
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/partners/anthropic/Makefile
View File

@ -14,7 +14,7 @@ test tests:
uv run --group test pytest -vvv --disable-socket --allow-unix-socket $(TEST_FILE)
integration_test integration_tests:
uv run --group test --group test_integration pytest -vvv --timeout 30 $(TEST_FILE)
uv run --group test --group test_integration pytest -vvv --timeout 30 tests/integration_tests/test_chat_models.py::test_async_stream_twice
test_watch:
uv run --group test ptw --snapshot-update --now . -- -vv $(TEST_FILE)

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

@ -145,6 +145,15 @@ async def test_stream_usage_override() -> None:
assert token.usage_metadata is None
async def test_async_stream_twice() -> None:
model = ChatAnthropic(model_name=MODEL_NAME)
async for token in model.astream("hi"):
assert isinstance(token, AIMessageChunk)
async for token in model.astream("hi"):
assert isinstance(token, AIMessageChunk)
async def test_abatch() -> None:
"""Test streaming tokens from ChatAnthropicMessages."""
llm = ChatAnthropicMessages(model_name=MODEL_NAME) # type: ignore[call-arg, call-arg]