github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-12 14:23:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow for async use of SelfAskWithSearchChain (#5394)

Browse Source 
# Allow for async use of SelfAskWithSearchChain


Co-authored-by: Dev 2049 <dev.dev2049@gmail.com>
This commit is contained in:
Mark Pors 2023-05-31 02:02:39 +02:00 committed by GitHub
parent 8121e04200
commit 0a44bfdca3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/agents/self_ask_with_search/base.py
View File

@ -77,7 +77,10 @@ class SelfAskWithSearchChain(AgentExecutor):
): ):
"""Initialize with just an LLM and a search chain.""" """Initialize with just an LLM and a search chain."""
search_tool = Tool( search_tool = Tool(
name="Intermediate Answer", func=search_chain.run, description="Search" name="Intermediate Answer",
func=search_chain.run,
coroutine=search_chain.arun,
description="Search",
) )
agent = SelfAskWithSearchAgent.from_llm_and_tools(llm, [search_tool]) agent = SelfAskWithSearchAgent.from_llm_and_tools(llm, [search_tool])
super().__init__(agent=agent, tools=[search_tool], **kwargs) super().__init__(agent=agent, tools=[search_tool], **kwargs)