community[patch]: Fix VectorStoreQATool (#18529)

Fix #18460
langchain-ai · Mar 5, 2024 · f77f7dc · f77f7dc
1 parent 539a13d
commit f77f7dc
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/libs/community/langchain_community/tools/vectorstore/tool.py b/libs/community/langchain_community/tools/vectorstore/tool.py
@@ -56,7 +56,7 @@ def _run(
         )
         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 @@ async def _arun(
         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 @@ def _run(
             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 @@ async def _arun(
             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},
             )
         )