github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-28 23:07:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

standard-tests[patch]: don't require str chunk contents (#22965)

Browse Source
This commit is contained in:
Bagatur
2024-06-17 05:52:24 -07:00
committed by GitHub
parent 892bd4c29b
commit 9ff249a38d
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py
View File

@@ -73,7 +73,6 @@ class ChatModelIntegrationTests(ABC):
for token in model.stream("Hello"): for token in model.stream("Hello"):
assert token is not None assert token is not None
assert isinstance(token, AIMessageChunk) assert isinstance(token, AIMessageChunk)
assert isinstance(token.content, str)
num_tokens += len(token.content) num_tokens += len(token.content)
assert num_tokens > 0 assert num_tokens > 0
@@ -85,7 +84,6 @@ class ChatModelIntegrationTests(ABC):
async for token in model.astream("Hello"): async for token in model.astream("Hello"):
assert token is not None assert token is not None
assert isinstance(token, AIMessageChunk) assert isinstance(token, AIMessageChunk)
assert isinstance(token.content, str)
num_tokens += len(token.content) num_tokens += len(token.content)
assert num_tokens > 0 assert num_tokens > 0