From f77f7dc3ec6e88997f78da4144415820acb0537a Mon Sep 17 00:00:00 2001
From: Christophe Bornet <cbornet@hotmail.com>
Date: Wed, 6 Mar 2024 00:56:58 +0100
Subject: [PATCH] community[patch]: Fix VectorStoreQATool (#18529)

Fix #18460
---
 .../langchain_community/tools/vectorstore/tool.py      | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/libs/community/langchain_community/tools/vectorstore/tool.py b/libs/community/langchain_community/tools/vectorstore/tool.py
index c0cbdb67062..7f9b965f08c 100644
--- a/libs/community/langchain_community/tools/vectorstore/tool.py
+++ b/libs/community/langchain_community/tools/vectorstore/tool.py
@@ -56,7 +56,7 @@ class VectorStoreQATool(BaseVectorStoreTool, BaseTool):
         )
         return chain.invoke(
             {chain.input_key: query},
-            config={"callbacks": [run_manager.get_child() if run_manager else None]},
+            config={"callbacks": run_manager.get_child() if run_manager else None},
         )[chain.output_key]
 
     async def _arun(
@@ -73,9 +73,7 @@ class VectorStoreQATool(BaseVectorStoreTool, BaseTool):
         return (
             await chain.ainvoke(
                 {chain.input_key: query},
-                config={
-                    "callbacks": [run_manager.get_child() if run_manager else None]
-                },
+                config={"callbacks": run_manager.get_child() if run_manager else None},
             )
         )[chain.output_key]
 
@@ -114,7 +112,7 @@ class VectorStoreQAWithSourcesTool(BaseVectorStoreTool, BaseTool):
             chain.invoke(
                 {chain.question_key: query},
                 return_only_outputs=True,
-                callbacks=run_manager.get_child() if run_manager else None,
+                config={"callbacks": run_manager.get_child() if run_manager else None},
             )
         )
 
@@ -135,6 +133,6 @@ class VectorStoreQAWithSourcesTool(BaseVectorStoreTool, BaseTool):
             await chain.ainvoke(
                 {chain.question_key: query},
                 return_only_outputs=True,
-                callbacks=run_manager.get_child() if run_manager else None,
+                config={"callbacks": run_manager.get_child() if run_manager else None},
             )
         )