From 29be36095476f33935cfceba79814c9ff94406e4 Mon Sep 17 00:00:00 2001
From: wushiai1109 <55396274+wushiai1109@users.noreply.github.com>
Date: Tue, 27 May 2025 18:58:14 +0800
Subject: [PATCH] Update nodes.py (#242)

SELECTED_SEARCH_ENGINE impossible equal to SearchEngine.ARXIV, should be SearchEngine.ARXIV.value, or use the encapsulated get_web_search_tool
---
 src/graph/nodes.py | 21 +++------------------
 1 file changed, 3 insertions(+), 18 deletions(-)

diff --git a/src/graph/nodes.py b/src/graph/nodes.py
index 892faef..6f95f10 100644
--- a/src/graph/nodes.py
+++ b/src/graph/nodes.py
@@ -50,24 +50,9 @@ def background_investigation_node(
     logger.info("background investigation node is running.")
     configurable = Configuration.from_runnable_config(config)
     query = state["messages"][-1].content
-    if SELECTED_SEARCH_ENGINE == SearchEngine.TAVILY:
-        searched_content = LoggedTavilySearch(
-            max_results=configurable.max_search_results
-        ).invoke({"query": query})
-        background_investigation_results = None
-        if isinstance(searched_content, list):
-            background_investigation_results = [
-                {"title": elem["title"], "content": elem["content"]}
-                for elem in searched_content
-            ]
-        else:
-            logger.error(
-                f"Tavily search returned malformed response: {searched_content}"
-            )
-    else:
-        background_investigation_results = get_web_search_tool(
-            configurable.max_search_results
-        ).invoke(query)
+    background_investigation_results = get_web_search_tool(
+        configurable.max_search_results
+    ).invoke(query)
     return Command(
         update={
             "background_investigation_results": json.dumps(