From bdce9a47d00828e012cb31650f0ed4e61378bd63 Mon Sep 17 00:00:00 2001 From: Sam Jove Date: Thu, 29 Aug 2024 09:29:28 -0400 Subject: [PATCH] community[patch]: callback before yield for _astream (gigachat) (#25834) Description: Moves yield to after callback for _astream for gigachat in the community package Issue: #16913 --------- Co-authored-by: Chester Curme --- libs/community/langchain_community/chat_models/gigachat.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/libs/community/langchain_community/chat_models/gigachat.py b/libs/community/langchain_community/chat_models/gigachat.py index 1e20fa84dfb..62285c98868 100644 --- a/libs/community/langchain_community/chat_models/gigachat.py +++ b/libs/community/langchain_community/chat_models/gigachat.py @@ -268,6 +268,7 @@ class GigaChat(_BaseGigaChat, BaseChatModel): dict(finish_reason=finish_reason) if finish_reason is not None else None ) - yield ChatGenerationChunk(message=chunk, generation_info=generation_info) if run_manager: await run_manager.on_llm_new_token(content) + + yield ChatGenerationChunk(message=chunk, generation_info=generation_info)