add Calculator tool to BasicAgent and update requirements

app.py

@@ -6,9 +6,11 @@ import requests, textwrap
 
 from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
 
-from langchain_community.tools import DuckDuckGoSearchRun
-from langchain.agents import initialize_agent, AgentType
+from langchain_community.tools import DuckDuckGoSearchRun, Calculator
+from langchain.agents import initialize_agent
 from langchain_core.prompts import ChatPromptTemplate
+from langchain_community.tools import Calculator
+
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -23,6 +25,7 @@ You are GAIA-Bot, a fully-autonomous problem-solver.
 You have access to these tools:
 - gaia_file_fetch    – download the file attached to the current task_id.
 - duckduckgo_search  – look up current facts on the public web.
+- calculator          – evaluate math expressions, percentages, date differences.
 
 If you require a file attached to the current task, download it with
 `requests_get("https://agents-course-unit4-scoring.hf.space/files/{task_id}")`
@@ -44,7 +47,7 @@ def gaia_file_fetch(task_id: str) -> str:
     • Input:  task_id (string)  
     • Output: text content (first 20 000 chars) or a notice if the file is binary.
     """
-    url = f"https://{GAIA_HOST}/files/{task_id}"
+    url = f"https://{DEFAULT_API_URL}/files/{task_id}"
     resp = requests.get(url, timeout=15)
     if resp.status_code != 200:
         return f"ERROR {resp.status_code}: could not fetch file."
@@ -74,6 +77,7 @@ class BasicAgent:
         tools = [
             gaia_file_fetch,
             DuckDuckGoSearchRun(),
+            Calculator()
         ]
 
         prompt = ChatPromptTemplate.from_messages([
@@ -85,7 +89,8 @@ class BasicAgent:
             tools=tools,
             llm=self.llm,
             prompt=prompt,
-            max_iterations=5,
+            max_iterations=15,
+            max_execution_time=120,  # seconds
             verbose=False,
         )
 

requirements.txt

@@ -1,6 +1,6 @@
 langchain
 langchain-huggingface
-langchain-community      # for DuckDuckGoSearchRun
+langchain-community      # for DuckDuckGoSearchRun, Calculator
 huggingface-hub
 duckduckgo-search
 gradio