diff --git a/docs/configuration_guide.md b/docs/configuration_guide.md index 215ab37..f02f53a 100644 --- a/docs/configuration_guide.md +++ b/docs/configuration_guide.md @@ -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 diff --git a/src/tools/search.py b/src/tools/search.py index 73c4128..5f1ef8b 100644 --- a/src/tools/search.py +++ b/src/tools/search.py @@ -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, )