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

feat:support config tavily search results (#591)

Browse Source 
* feat:support config tavily search results

* feat: support config tavily search results

* feat: update the default value of include_images

* fix: fix the test

---------

Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
This commit is contained in:
Fancy-hjyp
2025-09-22 18:26:50 +08:00
committed by GitHub
parent 150a730f98
commit 6bb0b95579
2 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/tools/search.py
View File

@@ -47,6 +47,11 @@ def get_web_search_tool(max_search_results: int):
# Only get and apply include/exclude domains for Tavily
include_domains: Optional[List[str]] = search_config.get("include_domains", [])
exclude_domains: Optional[List[str]] = search_config.get("exclude_domains", [])
include_raw_content = search_config.get("include_raw_content", True)
include_images: Optional[bool] = search_config.get("include_images", True)
include_image_descriptions: Optional[bool] = (
include_images and search_config.get("include_image_descriptions", True)
)
logger.info(
f"Tavily search configuration loaded: include_domains={include_domains}, exclude_domains={exclude_domains}"
@@ -55,9 +60,9 @@ def get_web_search_tool(max_search_results: int):
return LoggedTavilySearch(
name="web_search",
max_results=max_search_results,
include_raw_content=True,
include_images=True,
include_image_descriptions=True,
include_raw_content=include_raw_content,
include_images=include_images,
include_image_descriptions=include_image_descriptions,
include_domains=include_domains,
exclude_domains=exclude_domains,
)