community: Fix VectorStoreQATool

Fix #18460
langchain-ai · Mar 4, 2024 · 148576c · 148576c
1 parent bc768a1
commit 148576c
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/libs/community/langchain_community/tools/vectorstore/tool.py b/libs/community/langchain_community/tools/vectorstore/tool.py
@@ -16,7 +16,7 @@
 
 
 class BaseVectorStoreTool(BaseModel):
-    """Base class for tools that use a VectorStore."""
+    "communi""Base class for tools that use a VectorStore."""
 
     vectorstore: VectorStore = Field(exclude=True)
     llm: BaseLanguageModel = Field(default_factory=lambda: OpenAI(temperature=0))
@@ -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},
             )
         )