From 3d5e579ebd29a31cbdd2b28ed2cb8989a640cf55 Mon Sep 17 00:00:00 2001
From: laundry <40748509+laundry2@users.noreply.github.com>
Date: Wed, 14 May 2025 14:45:36 +0800
Subject: [PATCH] fix: fix start error when search engine is not tavliy and env
 TAVILY_API_KEY not exist (#133)

Change-Id: I58e865a11e89acaa3c0b884578cd995d0e9b5422
---
 src/tools/search.py | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/src/tools/search.py b/src/tools/search.py
index e4c92ad..992a6b5 100644
--- a/src/tools/search.py
+++ b/src/tools/search.py
@@ -9,7 +9,7 @@ from langchain_community.tools import BraveSearch, DuckDuckGoSearchResults
 from langchain_community.tools.arxiv import ArxivQueryRun
 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 (
     TavilySearchResultsWithImages,
 )
@@ -19,13 +19,16 @@ from src.tools.decorators import create_logged_tool
 logger = logging.getLogger(__name__)
 
 LoggedTavilySearch = create_logged_tool(TavilySearchResultsWithImages)
-tavily_search_tool = LoggedTavilySearch(
-    name="web_search",
-    max_results=SEARCH_MAX_RESULTS,
-    include_raw_content=True,
-    include_images=True,
-    include_image_descriptions=True,
-)
+if os.getenv("SEARCH_API", "") == SearchEngine.TAVILY.value:
+    tavily_search_tool = LoggedTavilySearch(
+        name="web_search",
+        max_results=SEARCH_MAX_RESULTS,
+        include_raw_content=True,
+        include_images=True,
+        include_image_descriptions=True,
+    )
+else:
+    tavily_search_tool = None
 
 LoggedDuckDuckGoSearch = create_logged_tool(DuckDuckGoSearchResults)
 duckduckgo_search_tool = LoggedDuckDuckGoSearch(