wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-24 06:34:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Refactors the retriever function to use async/await (#821)

Browse Source 
* refactor: Refactors the retriever function to use async/await
This commit is contained in:
Xun
2026-01-20 19:56:26 +08:00
committed by GitHub
parent 2ed0eeb107
commit 0e64c52975
10 changed files with 196 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/tools/retriever.py
View File

@@ -47,7 +47,15 @@ class RetrieverTool(BaseTool):
keywords: str,
run_manager: Optional[AsyncCallbackManagerForToolRun] = None,
) -> list[Document]:
return self._run(keywords, run_manager.get_sync())
logger.info(
f"Retriever tool query: {keywords}", extra={"resources": self.resources}
)
documents = await self.retriever.query_relevant_documents_async(
keywords, self.resources
)
if not documents:
return "No results found from the local knowledge base."
return [doc.to_dict() for doc in documents]
def get_retriever_tool(resources: List[Resource]) -> RetrieverTool | None: