feat: support duckduckgo search engine

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

View File

@@ -1,11 +1,18 @@
from .crawl import crawl_tool
from .python_repl import python_repl_tool
from .search import tavily_tool
from .bash_tool import bash_tool
from .search import tavily_search_tool, duckduckgo_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,
}
web_search_tool = search_tool_mappings.get(SELECTED_SEARCH_ENGINE, tavily_search_tool)
__all__ = [
"bash_tool",
"crawl_tool",
"tavily_tool",
"web_search_tool",
"python_repl_tool",
]