added required packages

agent.py

@@ -1,11 +1,15 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel
 from smolagents import tool
+from smolagents.tools import WikipediaTool, ArxivTool, ImageReaderTool, PythonREPLTool
 import os
 
-# Initialize model
-model = InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct", token = os.environ.get("HF_TOKEN"))
+# Initialize model with token
+model = InferenceClientModel(
+    model_id="Qwen/Qwen2.5-72B-Instruct",
+    token=os.environ.get("HF_TOKEN")
+)
 
-# Custom tools with corrected docstrings
+# Define math tools with detailed docstrings
 @tool
 def add(a: int, b: int) -> int:
     """Add two numbers.
@@ -37,7 +41,7 @@ def subtract(a: int, b: int) -> int:
     return a - b
 
 @tool
-def divide(a: int, b: int) -> int:
+def divide(a: int, b: int) -> float:
     """Divide two numbers.
     
     Args:
@@ -58,15 +62,37 @@ def modulus(a: int, b: int) -> int:
     """
     return a % b
 
-# Create agent with tools
+# Create the agent with all required tools
 agent = CodeAgent(
-    tools=[DuckDuckGoSearchTool(), add, multiply, subtract, divide, modulus],
+    tools=[
+        DuckDuckGoSearchTool(),
+        WikipediaTool(),
+        ArxivTool(),
+        ImageReaderTool(),
+        PythonREPLTool(),
+        add,
+        multiply,
+        subtract,
+        divide,
+        modulus
+    ],
     model=model,
+    planning_interval=3,  # Enable planning every 3 steps
+    max_steps=15,         # Limit max steps to prevent infinite loops
+    verbosity_level=1     # Enable some logging for debugging
 )
 
+# Function to run the agent and ensure correct output format
 def answer_question(question: str) -> str:
-    result = agent.run(question)
-    # Ensure output starts with "FINAL ANSWER:"
-    if not result.strip().startswith("FINAL ANSWER:"):
-        result = f"FINAL ANSWER: {result.strip()}"
-    return result
+    try:
+        result = agent.run(question)
+        if not result.strip().startswith("FINAL ANSWER:"):
+            result = f"FINAL ANSWER: {result.strip()}"
+        return result
+    except Exception as e:
+        return f"FINAL ANSWER: [AGENT ERROR: {str(e)}]"
+
+# Example test
+if __name__ == "__main__":
+    test_question = "What is the capital of France?"
+    print(answer_question(test_question))