fix: fix start error when search engine is not tavliy and env TAVILY_API_KEY not exist (#133)

Change-Id: I58e865a11e89acaa3c0b884578cd995d0e9b5422

Author: laundry2

src/tools/search.py

@@ -9,7 +9,7 @@
 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 @@
 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(