wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-03 06:12:14 +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

8
docs/configuration_guide.md
View File

@@ -188,7 +188,7 @@ DeerFlow allows you to control which domains are included or excluded in Tavily
`Tips`: it only supports Tavily currently.
You can configure domain filtering in your `conf.yaml` file as follows:
You can configure domain filtering and search results in your `conf.yaml` file as follows:
```yaml
SEARCH_ENGINE:
@@ -202,6 +202,12 @@ SEARCH_ENGINE:
exclude_domains:
- unreliable-site.com
- spam-domain.net
# Include images in search results, default: true
include_images: false
# Include image descriptions in search results, default: true
include_image_descriptions: false
# Include raw content in search results, default: true
include_raw_content: false
```
## RAG (Retrieval-Augmented Generation) Configuration

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,
)