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

feat: support duckduckgo search engine

Browse Source
This commit is contained in:
He Tao
2025-04-10 11:45:04 +08:00
parent 03798ded08
commit 1195612c47
13 changed files with 88 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/tools/search.py
View File

@@ -1,10 +1,18 @@
import logging
from langchain_community.tools.tavily_search import TavilySearchResults
from src.config import TAVILY_MAX_RESULTS
from langchain_community.tools import DuckDuckGoSearchResults
from src.config import SEARCH_MAX_RESULTS
from .decorators import create_logged_tool
logger = logging.getLogger(__name__)
# Initialize Tavily search tool with logging
LoggedTavilySearch = create_logged_tool(TavilySearchResults)
tavily_tool = LoggedTavilySearch(name="tavily_search", max_results=TAVILY_MAX_RESULTS)
tavily_search_tool = LoggedTavilySearch(
name="web_search", max_results=SEARCH_MAX_RESULTS
)
LoggedDuckDuckGoSearch = create_logged_tool(DuckDuckGoSearchResults)
duckduckgo_search_tool = LoggedDuckDuckGoSearch(
name="web_search", max_results=SEARCH_MAX_RESULTS
)