community[patch]: callback before yield for cloudflare

langchain-ai · Sep 27, 2024 · e6773ae · e6773ae
1 parent c6350d6
commit e6773ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/community/langchain_community/llms/cloudflare_workersai.py b/libs/community/langchain_community/llms/cloudflare_workersai.py
@@ -105,9 +105,9 @@ def _stream(
                     logger.debug(chunk)
                     raise e
                 if data is not None and "response" in data:
+                    if run_manager:
+                        run_manager.on_llm_new_token(data["response"])
                     yield GenerationChunk(text=data["response"])
-                if run_manager:
-                    run_manager.on_llm_new_token(data["response"])
         logger.debug("stream end")
         self.streaming = original_steaming