fix: try stricter tools

app.py

@@ -13,12 +13,6 @@ 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
-from llama_index.core.node_parser import SentenceSplitter
-from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 from llama_index.core.schema import Document
 from llama_index.core import get_response_synthesizer
 import pandas as pd
@@ -38,12 +32,8 @@ class BasicAgent:
 
         # self.llm = Ollama(model="qwen2.5:7b", request_timeout=500)
         self.llm = Gemini(model_name="models/gemini-2.0-flash")
+
         def get_answers_from_youtube_transcript(youtube_link: str) -> str:
-            """Fetches transcript of the given youtube_link and returns matching answers based on query.
-                Args:
-                youtube_link (str): youtube video link for which we need to answer questions on.
-                query (str): question to answer from video transcript.
-            """
             try:
                 loader = YoutubeTranscriptReader()
                 documents = loader.load_data(
@@ -52,7 +42,7 @@ class BasicAgent:
 
                 text = documents[0].text_resource.text
 
-                return "Here's the transcript from the video, examine and formulate answer based on what is said in the transcript: \n" + text
+                return { "video_transcript": text }
             except Exception as e:
                 print("error", e)
 
@@ -62,29 +52,30 @@ 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 = 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
+                raw_results = DuckDuckGoSearchToolSpec().duckduckgo_full_search(query, max_results=5)
+                texts = [res['body'] for res in raw_results]
+                full_text = "\n".join(texts)
+                return { "web_search_results": full_text }
 
             except Exception as e:
                 return f"An error occurred: {e}"
@@ -103,8 +94,7 @@ class BasicAgent:
                 if text == "":
                     text = WikipediaToolSpec().search_data(query)
                     
-                return "Here's the response from wikipedia search: " + text
-
+                return { "wiki_search_results": text }
             except Exception as e:
                 return f"An error occurred: {e}"
 
@@ -115,7 +105,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], llm=self.llm)
+        self.agent = AgentWorkflow.from_tools_or_functions([duckduckgo_search_tool, youtube_transcript_answer_tool, wikipedia_search_tool, web_page_reader_tool], llm=self.llm, system_prompt="You're an ai agent designed for question answering. Keep your answers concise or even one word when possible. You have access to a bunch of tools, utilise them well to reach answers.")
         print("BasicAgent initialized.")
 
     async def run_agent(self, question: str):