fix: try google search

app.py

@@ -13,6 +13,7 @@ from llama_index.core.tools import FunctionTool
 from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
 from llama_index.core.agent.workflow import AgentWorkflow
 from llama_index.llms.gemini import Gemini
+from llama_index.tools.google import GoogleSearchToolSpec
 from llama_index.retrievers.bm25 import BM25Retriever
 from llama_index.core.query_engine import RetrieverQueryEngine
 from llama_index.core.tools import QueryEngineTool
@@ -61,29 +62,28 @@ class BasicAgent:
             description="Fetches transcript of the given youtube_link and returns matching answers based on query. To be called when video link is given.",
         )
 
-        def web_page_reader(url: str, query: str) -> str:
-            try:
-                documents = SimpleWebPageReader(html_to_text=True).load_data(
-                    [url]
-                )
-                index = SummaryIndex.from_documents(documents)
-                query_engine = index.as_query_engine()
-                response = query_engine.query(query)
-                return response.response
-            except Exception as e:
-                print("error in webpage", e)
-
-        web_page_reader_tool = FunctionTool.from_defaults(
-            web_page_reader,
-            name="web_page_reader",
-            description="Visits the wepage on given url and returns response on the passed query"
-        )
+        # def web_page_reader(url: str, query: str) -> str:
+        #     try:
+        #         documents = SimpleWebPageReader(html_to_text=True).load_data(
+        #             [url]
+        #         )
+        #         index = SummaryIndex.from_documents(documents)
+        #         query_engine = index.as_query_engine()
+        #         response = query_engine.query(query)
+        #         return response.response
+        #     except Exception as e:
+        #         print("error in webpage", e)
+
+        # web_page_reader_tool = FunctionTool.from_defaults(
+        #     web_page_reader,
+        #     name="web_page_reader",
+        #     description="Visits the wepage on given url and returns response on the passed query"
+        # )
 
         def duck_duck_go_search_tool(query: str) -> str:
             try:
-                raw_results = DuckDuckGoSearchToolSpec().duckduckgo_full_search(query, max_results=5)
-                texts = [res['body'] for res in raw_results]
-                full_text = "\n".join(texts)
+                raw_results = GoogleSearchToolSpec(key="AIzaSyBCp55z6a9hRqYoJZLluK-rku7uAST4qdk", engine="042d23e47f6584359").google_search(query=query)
+                full_text = raw_results[0].text
                 return "Here's the responses from the search: " + full_text
 
             except Exception as e:
@@ -115,7 +115,7 @@ class BasicAgent:
                 description="Searches wikipedia and converts results into a high quality answer."
             )
 
-        self.agent = AgentWorkflow.from_tools_or_functions([duckduckgo_search_tool, youtube_transcript_answer_tool, wikipedia_search_tool, web_page_reader_tool], llm=self.llm)
+        self.agent = AgentWorkflow.from_tools_or_functions([duckduckgo_search_tool, youtube_transcript_answer_tool, wikipedia_search_tool], llm=self.llm)
         print("BasicAgent initialized.")
 
     async def run_agent(self, question: str):
@@ -186,7 +186,7 @@ async def run_and_submit_all( profile: gr.OAuthProfile | None):
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
     for item in questions_data:
-        await asyncio.sleep(60)
+        await asyncio.sleep(10)
         task_id = item.get("task_id")
         question_text = item.get("question")
         if not task_id or question_text is None: