Update app.py

app.py

@@ -53,8 +53,6 @@ def search_tavily(state):
         
     search_docs = tavily_search.invoke(state['question'])
     
-    print("Tavily search returned:", search_docs)
-    
     # Format
     formatted_search_docs = "\n\n---\n\n".join(
         [
@@ -72,8 +70,6 @@ def search_wikipedia(state):
     # Search
     search_docs = WikipediaLoader(query=state['question'], 
                                   load_max_docs=2).load()
-
-    print("Wikipedia search returned:", search_docs)
     
     # Format
     formatted_search_docs = "\n\n---\n\n".join(
@@ -85,25 +81,6 @@ def search_wikipedia(state):
 
     return {"context": [formatted_search_docs]}
 
-def search_DuckDuckGo(state):
-
-    """ Retrive answer from DuckDuckGoSearch."""
-
-    ddg_search = DuckDuckGoSearchRun(max_results=2)
-    search_docs = ddg_search.invoke(state['question'])
-
-    print("DuckDuckGo search returned:", search_docs)
-    
-    # Format
-    formatted_search_docs = "\n\n---\n\n".join(
-        [
-            f'<Document href="{doc["url"]}"/>\n{doc["content"]}\n</Document>'
-            for doc in search_docs
-        ]
-    )
-
-    return {"context": [formatted_search_docs]}
-
 def generate_answer(state):
 
     """Node to give answer to the question"""
@@ -126,15 +103,12 @@ builder = StateGraph(State)
 
 builder.add_node("search_tavily",search_tavily)
 builder.add_node("search_wikipedia", search_wikipedia)
-builder.add_node("search_DuckDuckGo", search_DuckDuckGo)
 builder.add_node("generate_answer", generate_answer)
 
 builder.add_edge(START, "search_wikipedia")
 builder.add_edge(START, "search_tavily")
-builder.add_edge(START, "search_DuckDuckGo")
 builder.add_edge("search_wikipedia", "generate_answer")
 builder.add_edge("search_tavily", "generate_answer")
-builder.add_edge("search_DuckDuckGo", "generate_answer")
 graph = builder.compile()