fix bug

app.py

@@ -9,9 +9,8 @@ import time
 from typing import List, Dict, Any, Optional, Union, Tuple
 
 # --- Import necessary libraries ---
-from smolagents import CodeAgent
+from smolagents import CodeAgent, tool
 from smolagents.models import LiteLLMModel
-from llama_index.core.tools import FunctionTool
 from langgraph.graph import StateGraph, END
 
 # --- Constants ---
@@ -257,33 +256,61 @@ class GAIAAgent:
     
     def setup_tools(self):
         """Set up tools for the agent"""
-        # Use FunctionTool interface from llama_index but integrate with smolagents
+        # Create tools using smolagents @tool decorator
+        
+        @tool
+        def calculator(expression: str) -> str:
+            """Calculate mathematical expressions like '2 + 2' or '(15 * 3) / 2'
+            
+            Args:
+                expression: The mathematical expression to calculate
+            """
+            return GAIAToolkit.calculator(expression)
+        
+        @tool
+        def search_web(query: str) -> str:
+            """Search for information related to a query
+            
+            Args:
+                query: The search query
+            """
+            return GAIAToolkit.search_web(query)
+        
+        @tool
+        def file_reader(file_id: str) -> str:
+            """Read file content given a file ID
+            
+            Args:
+                file_id: The ID of the file to read
+            """
+            return GAIAToolkit.file_reader(file_id)
+        
+        @tool
+        def analyze_text(text: str) -> str:
+            """Analyze text to extract statistics and key information
+            
+            Args:
+                text: The text to analyze
+            """
+            result = GAIAToolkit.analyze_text(text)
+            return str(result)
+        
+        @tool
+        def extract_answer(reasoning: str) -> str:
+            """Extract the final answer from reasoning
+            
+            Args:
+                reasoning: The reasoning text to extract the answer from
+            """
+            return GAIAToolkit.extract_answer(reasoning)
+        
+        # Assign the tools to the agent
         self.tools = [
-            FunctionTool.from_defaults(
-                name="calculator",
-                description="Calculate mathematical expressions like '2 + 2' or '(15 * 3) / 2'",
-                fn=GAIAToolkit.calculator
-            ),
-            FunctionTool.from_defaults(
-                name="search_web",
-                description="Search for information related to a query",
-                fn=GAIAToolkit.search_web
-            ),
-            FunctionTool.from_defaults(
-                name="file_reader",
-                description="Read file content given a file ID",
-                fn=GAIAToolkit.file_reader
-            ),
-            FunctionTool.from_defaults(
-                name="analyze_text",
-                description="Analyze text to extract statistics and key information",
-                fn=GAIAToolkit.analyze_text
-            ),
-            FunctionTool.from_defaults(
-                name="extract_answer",
-                description="Extract the final answer from reasoning",
-                fn=GAIAToolkit.extract_answer
-            )
+            calculator,
+            search_web,
+            file_reader,
+            analyze_text,
+            extract_answer
         ]
     
     def create_system_prompt(self) -> str:
@@ -356,32 +383,24 @@ class GAIAAgent:
     def analyze_and_plan(self, question: str):
         """Analyze the question and plan approach"""
         analyze_prompt = f"""Analyze the following question:
-
-{question}
-
-Identify:
-1. Question type (calculation, information retrieval, text analysis, etc.)
-2. Key tools needed
-3. Solution steps
-
-Provide only a concise analysis, don't attempt to answer the question.
-"""
-        
+                            {question}
+                            Identify:
+                            1. Question type (calculation, information retrieval, text analysis, etc.)
+                            2. Key tools needed
+                            3. Solution steps
+                            Provide only a concise analysis, don't attempt to answer the question.
+                            """
         analysis = self.model.generate(analyze_prompt).strip()
         self.workflow_states["analysis"] = analysis
         
         plan_prompt = f"""Based on the question analysis:
-
-{analysis}
-
-Formulate a concise step-by-step plan to answer the question:
-
-{question}
-
-Use available tools: calculator, search_web, file_reader, analyze_text.
-List specific steps, don't attempt to answer the question.
-"""
-        
+                            {analysis}
+                            Formulate a concise step-by-step plan to answer the question:
+                            {question}
+                            Use available tools: calculator, search_web, file_reader, analyze_text.
+                            List specific steps, don't attempt to answer the question.
+                            """
+                                    
         plan = self.model.generate(plan_prompt).strip()
         self.workflow_states["plan"] = plan