community[patch]/chat_models/bedrock.py yielded after on_llm_new_token

langchain-ai · Dec 25, 2024 · e412042 · e412042
1 parent 80d58b5
commit e412042
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(