mirror of
https://git.mirrors.martin98.com/https://github.com/bytedance/deer-flow
synced 2025-10-04 17:16:33 +08:00
19 lines
607 B
Python
19 lines
607 B
Python
import logging
|
|
from langchain_community.tools.tavily_search import TavilySearchResults
|
|
from langchain_community.tools import DuckDuckGoSearchResults
|
|
from src.config import SEARCH_MAX_RESULTS
|
|
from .decorators import create_logged_tool
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
LoggedTavilySearch = create_logged_tool(TavilySearchResults)
|
|
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
|
|
)
|