github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 15:43:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: Fix VectorStoreQATool (#18529)

Browse Source 
Fix #18460
This commit is contained in:
Christophe Bornet 2024-03-06 00:56:58 +01:00 committed by GitHub
parent 539a13dbda
commit f77f7dc3ec
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
libs/community/langchain_community/tools/vectorstore/tool.py
View File

@ -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},
)
)