Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 2 deletions
Show Stats 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