Update app.py

app.py

@@ -13,11 +13,15 @@ from langchain import HuggingFaceHub
 import torch
 import streamlit as st
 from langchain.utilities import SerpAPIWrapper
+from langchain.tools import DuckDuckGoSearchRun
 import os
 hf_token = os.environ['HF_TOKEN']
 serp_token = os.environ['SERP_TOKEN']
 repo_id = "sentence-transformers/all-mpnet-base-v2"
 
+
+
+
 HUGGINGFACEHUB_API_TOKEN= hf_token
 hf = HuggingFaceHubEmbeddings(
     repo_id=repo_id,
@@ -103,14 +107,19 @@ def main():
             description="""Use this tool to answer document related questions. The input to this tool should be the question.""",
         )
 
-        search = SerpAPIWrapper(serpapi_api_key=serp_token)
-
-        google_searchtool= Tool(
-                name="Current Search",
-                func=search.run,
-                description="use this tool to answer real time or current search related questions.",
-            )
-
+        # search = SerpAPIWrapper(serpapi_api_key=serp_token)
+
+        # google_searchtool= Tool(
+        #         name="Current Search",
+        #         func=search.run,
+        #         description="use this tool to answer real time or current search related questions.",
+        #     )
+        search = DuckDuckGoSearchRun()
+        search_tool = Tool(
+            name="search",
+            func=search,
+            description="use this tool to answer real time or current search related questions."
+)
         # tool for asking human
         human_ask_tool = CustomAskHumanTool()
         # agent prompt
@@ -120,7 +129,7 @@ def main():
         # initialize agent
         agent = initialize_agent(
             agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
-                    tools=[db_search_tool,google_searchtool],
+                    tools=[db_search_tool,search_tool],
             llm=llm,
             verbose=True,
             max_iterations=5,