GitHub-Proxy/deer-flow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/bytedance/deer-flow synced 2025-08-19 06:35:52 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
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,13 +19,16 @@ 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)
tavily_search_tool = LoggedTavilySearch( if os.getenv("SEARCH_API", "") == SearchEngine.TAVILY.value:
name="web_search", tavily_search_tool = LoggedTavilySearch(
max_results=SEARCH_MAX_RESULTS, name="web_search",
include_raw_content=True, max_results=SEARCH_MAX_RESULTS,
include_images=True, include_raw_content=True,
include_image_descriptions=True, include_images=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(