diff --git a/libs/community/langchain_community/chat_models/zhipuai.py b/libs/community/langchain_community/chat_models/zhipuai.py index 9ec8ffa728c..f3573129dd7 100644 --- a/libs/community/langchain_community/chat_models/zhipuai.py +++ b/libs/community/langchain_community/chat_models/zhipuai.py @@ -496,9 +496,10 @@ class ChatZhipuAI(BaseChatModel): chunk = ChatGenerationChunk( message=chunk, generation_info=generation_info ) - yield chunk if run_manager: run_manager.on_llm_new_token(chunk.text, chunk=chunk) + yield chunk + if finish_reason is not None: break @@ -582,8 +583,9 @@ class ChatZhipuAI(BaseChatModel): chunk = ChatGenerationChunk( message=chunk, generation_info=generation_info ) - yield chunk if run_manager: await run_manager.on_llm_new_token(chunk.text, chunk=chunk) + yield chunk + if finish_reason is not None: break