From e412042482321755926d1be059fa3250abf26285 Mon Sep 17 00:00:00 2001 From: Meenakshi Ruia Date: Wed, 25 Dec 2024 23:54:06 +0530 Subject: [PATCH] community[patch]/chat_models/bedrock.py yielded after on_llm_new_token --- libs/community/langchain_community/chat_models/bedrock.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/community/langchain_community/chat_models/bedrock.py b/libs/community/langchain_community/chat_models/bedrock.py index 35ad534c165d4..8f4943951d5d5 100644 --- a/libs/community/langchain_community/chat_models/bedrock.py +++ b/libs/community/langchain_community/chat_models/bedrock.py @@ -266,7 +266,7 @@ def _stream( delta = chunk.text cg_chunk = ChatGenerationChunk(message=AIMessageChunk(content=delta)) if run_manager: - run_manager.on_llm_new_token(delta, chunk = cg_chunk ) + run_manager.on_llm_new_token(delta, chunk=cg_chunk ) yield cg_chunk def _generate(