github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-25 08:03:39 +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( return chain.invoke(
{chain.input_key: query}, {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] )[chain.output_key]
async def _arun( async def _arun(
@ -73,9 +73,7 @@ class VectorStoreQATool(BaseVectorStoreTool, BaseTool):
return ( return (
await chain.ainvoke( await chain.ainvoke(
{chain.input_key: query}, {chain.input_key: query},
config={ config={"callbacks": run_manager.get_child() if run_manager else None},
"callbacks": [run_manager.get_child() if run_manager else None]
},
) )
)[chain.output_key] )[chain.output_key]
@ -114,7 +112,7 @@ class VectorStoreQAWithSourcesTool(BaseVectorStoreTool, BaseTool):
chain.invoke( chain.invoke(
{chain.question_key: query}, {chain.question_key: query},
return_only_outputs=True, 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( await chain.ainvoke(
{chain.question_key: query}, {chain.question_key: query},
return_only_outputs=True, 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},
) )
) )