From e367ee71e722001987d75fa0140bb95cf40d3ec6 Mon Sep 17 00:00:00 2001
From: blob42 <spike@w530>
Date: Sun, 26 Mar 2023 04:11:00 +0200
Subject: [PATCH] includ tool name for on_tool_end callback

---
 langchain/tools/base.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/langchain/tools/base.py b/langchain/tools/base.py
index 9a40cde1..f53101c2 100644
--- a/langchain/tools/base.py
+++ b/langchain/tools/base.py
@@ -70,7 +70,7 @@ class BaseTool(BaseModel):
             self.callback_manager.on_tool_error(e, verbose=verbose)
             raise e
         self.callback_manager.on_tool_end(
-            observation, verbose=verbose, color=color, **kwargs
+            observation, verbose=verbose, color=color, name=self.name, **kwargs
         )
         return observation
 
@@ -112,10 +112,10 @@ class BaseTool(BaseModel):
             raise e
         if self.callback_manager.is_async:
             await self.callback_manager.on_tool_end(
-                observation, verbose=verbose, color=color, **kwargs
+                observation, verbose=verbose, color=color, name=self.name, **kwargs
             )
         else:
             self.callback_manager.on_tool_end(
-                observation, verbose=verbose, color=color, **kwargs
+                observation, verbose=verbose, color=color, name=self.name, **kwargs
             )
         return observation