Update agent.py

agent.py

@@ -18,12 +18,14 @@ from langchain_community.utilities import WikipediaAPIWrapper
 from langchain_community.document_loaders import YoutubeLoader
 from langchain_community.document_loaders import UnstructuredExcelLoader
 from langchain_community.document_loaders import AssemblyAIAudioTranscriptLoader
+from langchain.text_splitter import CharacterTextSplitter
+from langchain_community.utilities import GoogleSerperAPIWrapper
 
 load_dotenv()
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 @tool
-def web_search(query: str) -> str:
+def duckduck_websearch(query: str) -> str:
     """Allows search through DuckDuckGo.
     Args:
         query: what you want to search
@@ -32,6 +34,16 @@ def web_search(query: str) -> str:
     results = search.invoke(query)
     return "\n".join(results)
 
+@tool
+def serper_websearch(query: str) -> str:
+    """Allows search through Serper.
+    Args:
+        query: what you want to search
+    """
+    search = GoogleSerperAPIWrapper(serper_api_key=os.getenv("SERPER_API_KEY"))
+    results = search.run(query)
+    return results
+
 @tool
 def visit_webpage(url: str) -> str:
     """Fetches raw HTML content of a web page.
@@ -54,6 +66,14 @@ def wiki_search(query: str) -> str:
     wikipediatool = WikipediaQueryRun(api_wrapper=api_wrapper)
     return wikipediatool.run({"query": query})
 
+@tool
+def text_splitter(text: str) -> List[str]:
+    """Splits text into chunks using LangChain's CharacterTextSplitter.
+    Args:
+        text: A string of text to split.
+    """
+    splitter = CharacterTextSplitter(chunk_size=450, chunk_overlap=10)
+    return splitter.split_text(text)
 
 @tool
 def youtube_transcript(video_url: str) -> str:
@@ -211,7 +231,7 @@ def divide(a: float, b: float) -> float:
 class BasicAgent:
     def __init__(self):
         self.model = ChatGoogleGenerativeAI(
-            model="gemini-2.0-flash",
+            model="gemini-2.0-flash-lite",
             temperature=0,
             max_tokens=128,
             timeout=None,
@@ -229,9 +249,11 @@ class BasicAgent:
                 If you are asked for a comma separated list, apply the above rules depending of whether the element to put in the list is a number or a string.
 
                 You have access to the following tools:
-                - web_search: web search the content of the query by passing the query as input
+                - serper_websearch: web search the content of the query by passing the query as input with Serper Search Engine
+                - duckduck_websearch: web search the content of the query by passing the query as input with DuckDuckGo Search Engine
                 - visit_webpage: visit the given webpage url by passing the url as input
                 - wiki_search: wiki search the content of the query by passing the query as input if the question asks for wiki search it
+                - text_splitter: split text into chunks
                 - youtube_transcript: fetch the transcript of the Youtube video by passing the video url as input if the question asks for watching a Youtube video
                 - read_file: read the content of the attached file by passing the TASK-ID as input
                 - excel_read: read the content of the attached excel file by passing the TASK-ID as input
@@ -239,10 +261,10 @@ class BasicAgent:
                 - mp3_listen: listen to the content of the attached mp3 file by passing the TASK-ID as input
                 - image_caption: understand the visual content of the attached image by passing the TASK-ID as input
                 - python_tool: run the python code
-
+                
                 If Task ID is included in the question, remember to call the relevant read tools [ie. read_file, excel_read, csv_read, mp3_listen, image_caption]
         """
-        self.tools = [web_search, visit_webpage, wiki_search, youtube_transcript, read_file, excel_read, csv_read, mp3_listen, image_caption, python_tool]
+        self.tools = [serper_websearch, duckduck_websearch, visit_webpage, wiki_search, text_splitter, youtube_transcript, read_file, excel_read, csv_read, mp3_listen, image_caption, python_tool]
         self.prompt = ChatPromptTemplate.from_messages([
             ("system", self.sys_prompt),
             ("human", "{input}")
@@ -262,11 +284,12 @@ class BasicAgent:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         # response = self.agent_exe.invoke({"input": f"Question: {question}"})
         # fixed_answer = response['message'][-1].content
-        time.sleep(15)
+        
         if file_name == "" or file_name is None:
             fixed_answer = self.agent.run(question)
         else:
             fixed_answer = self.agent.run(f'{question} with TASK-ID: {task_id}')
         # fixed_answer = "This is a default answer."
         print(f"Agent returning fixed answer: {fixed_answer}")
+        time.sleep(60)
         return fixed_answer