From 4530d851e4812b87696802f628326c4a1d2fa743 Mon Sep 17 00:00:00 2001
From: Riccardo Schirone <562321+ret2libc@users.noreply.github.com>
Date: Tue, 25 Jun 2024 14:42:03 +0200
Subject: [PATCH] Merge pull request #22662

* core: runnables: special handling GeneratorExit because no error
---
 libs/core/langchain_core/runnables/base.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/libs/core/langchain_core/runnables/base.py b/libs/core/langchain_core/runnables/base.py
index 6ea52daf4c7..4950302777f 100644
--- a/libs/core/langchain_core/runnables/base.py
+++ b/libs/core/langchain_core/runnables/base.py
@@ -1892,6 +1892,8 @@ class Runnable(Generic[Input, Output], ABC):
                             final_input_supported = False
                 else:
                     final_input = ichunk
+        except GeneratorExit:
+            run_manager.on_chain_end(final_output, inputs=final_input)
         except BaseException as e:
             run_manager.on_chain_error(e, inputs=final_input)
             raise