wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-17 19:44:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: rag retrieving tool call result display (#263)

Browse Source 
* feat: local search tool call result display

* chore: add file copyright

* fix: miss edit plan interrupt feedback

* feat: disable pasting html into input box
This commit is contained in:
JeffJiang
2025-05-29 19:52:34 +08:00
committed by GitHub
parent 7e9fbed918
commit 4ddd659d8d
12 changed files with 10267 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/tools/retriever.py
View File

@@ -1,3 +1,6 @@
# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
# SPDX-License-Identifier: MIT
import logging
from typing import List, Optional, Type
from langchain_core.tools import BaseTool
@@ -19,9 +22,7 @@ class RetrieverInput(BaseModel):
class RetrieverTool(BaseTool):
name: str = "local_search_tool"
description: str = (
"Useful for retrieving information from the file with `rag://` uri prefix, it should be higher priority than the web search or writing code. Input should be a search keywords."
)
description: str = "Useful for retrieving information from the file with `rag://` uri prefix, it should be higher priority than the web search or writing code. Input should be a search keywords."
args_schema: Type[BaseModel] = RetrieverInput
retriever: Retriever = Field(default_factory=Retriever)