From 9ff249a38d425a410efe61145ab861f32f0cfc55 Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Mon, 17 Jun 2024 05:52:24 -0700 Subject: [PATCH] standard-tests[patch]: don't require str chunk contents (#22965) --- .../langchain_standard_tests/integration_tests/chat_models.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py b/libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py index 9eabc6239e..0dfc9a0481 100644 --- a/libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py +++ b/libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py @@ -73,7 +73,6 @@ class ChatModelIntegrationTests(ABC): for token in model.stream("Hello"): assert token is not None assert isinstance(token, AIMessageChunk) - assert isinstance(token.content, str) num_tokens += len(token.content) assert num_tokens > 0 @@ -85,7 +84,6 @@ class ChatModelIntegrationTests(ABC): async for token in model.astream("Hello"): assert token is not None assert isinstance(token, AIMessageChunk) - assert isinstance(token.content, str) num_tokens += len(token.content) assert num_tokens > 0