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

feat: support arxiv & brave search

Browse Source
This commit is contained in:
He Tao
2025-04-11 15:37:55 +08:00
parent 23298abd14
commit 3a342a62ba
7 changed files with 99 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/tools/__init__.py
View File

@@ -1,12 +1,19 @@
from .crawl import crawl_tool
from .python_repl import python_repl_tool
from .search import tavily_search_tool, duckduckgo_search_tool
from .search import (
tavily_search_tool,
duckduckgo_search_tool,
brave_search_tool,
arxiv_search_tool,
)
from src.config import SELECTED_SEARCH_ENGINE, SearchEngine
# Map search engine names to their respective tools
search_tool_mappings = {
SearchEngine.TAVILY.value: tavily_search_tool,
SearchEngine.DUCKDUCKGO.value: duckduckgo_search_tool,
SearchEngine.BRAVE_SEARCH.value: brave_search_tool,
SearchEngine.ARXIV.value: arxiv_search_tool,
}
web_search_tool = search_tool_mappings.get(SELECTED_SEARCH_ENGINE, tavily_search_tool)