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

fix: fix start error when search engine is not tavliy and env TAVILY_API_KEY not exist (#133)

Browse Source 
Change-Id: I58e865a11e89acaa3c0b884578cd995d0e9b5422
This commit is contained in:
laundry
2025-05-14 14:45:36 +08:00
committed by GitHub
parent a14ca92c36
commit 3d5e579ebd
1 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/tools/search.py
View File

@@ -9,7 +9,7 @@ from langchain_community.tools import BraveSearch, DuckDuckGoSearchResults
from langchain_community.tools.arxiv import ArxivQueryRun from langchain_community.tools.arxiv import ArxivQueryRun
from langchain_community.utilities import ArxivAPIWrapper, BraveSearchWrapper from langchain_community.utilities import ArxivAPIWrapper, BraveSearchWrapper
from src.config import SEARCH_MAX_RESULTS from src.config import SEARCH_MAX_RESULTS, SearchEngine
from src.tools.tavily_search.tavily_search_results_with_images import ( from src.tools.tavily_search.tavily_search_results_with_images import (
TavilySearchResultsWithImages, TavilySearchResultsWithImages,
) )
@@ -19,6 +19,7 @@ from src.tools.decorators import create_logged_tool
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
LoggedTavilySearch = create_logged_tool(TavilySearchResultsWithImages) LoggedTavilySearch = create_logged_tool(TavilySearchResultsWithImages)
if os.getenv("SEARCH_API", "") == SearchEngine.TAVILY.value:
tavily_search_tool = LoggedTavilySearch( tavily_search_tool = LoggedTavilySearch(
name="web_search", name="web_search",
max_results=SEARCH_MAX_RESULTS, max_results=SEARCH_MAX_RESULTS,
@@ -26,6 +27,8 @@ tavily_search_tool = LoggedTavilySearch(
include_images=True, include_images=True,
include_image_descriptions=True, include_image_descriptions=True,
) )
else:
tavily_search_tool = None
LoggedDuckDuckGoSearch = create_logged_tool(DuckDuckGoSearchResults) LoggedDuckDuckGoSearch = create_logged_tool(DuckDuckGoSearchResults)
duckduckgo_search_tool = LoggedDuckDuckGoSearch( duckduckgo_search_tool = LoggedDuckDuckGoSearch(