Update app.py

app.py

@@ -29,12 +29,6 @@ class BasicAgent:
         if not HF_TOKEN:
             raise ValueError("HF_TOKEN environment variable not set. Please set your Hugging Face API token.")
         
-        # Initialize tools
-        print("In constructor of basic agent")
-        self.calculator = create_calculator_tool()
-        print("Calculator tool created")
-        self.search_tool = create_search_tool()
-        print("Search tool created")
         # Set up headers for API calls
         self.headers = {
             "Authorization": f"Bearer {HF_TOKEN}",
@@ -54,22 +48,17 @@ class BasicAgent:
         print("Stategraph created")
         # Add nodes
         workflow.add_node("analyze", self._analyze_question)
-        workflow.add_node("calculator", self._use_calculator)
         workflow.add_node("search", self._use_search)
         workflow.add_node("final_answer", self._generate_final_answer)
         
         # Define edges
-        workflow.add_edge("analyze", "calculator")
         workflow.add_edge("analyze", "search")
         workflow.add_edge("analyze", "final_answer")
-        workflow.add_edge("calculator", "final_answer")
         workflow.add_edge("search", "final_answer")
         
         # Define conditional edges
         def router(state: AgentState) -> str:
-            if state["current_step"] == 'calculator':
-                return 'calculator'
-            elif state["current_step"] == 'search':
+            if state["current_step"] == 'search':
                 return 'search'
             elif state["current_step"] == 'final_answer':
                 return 'final_answer'
@@ -79,7 +68,6 @@ class BasicAgent:
             "analyze",
             router,
             {
-                "calculator": "calculator",
                 "search": "search",
                 "final_answer": "final_answer"
             }
@@ -110,13 +98,8 @@ class BasicAgent:
         prompt = f"""Analyze this question and determine what needs to be done: {state['question']}
         Return your analysis in this format:
         {{
-            "needs_calculation": true/false,
             "needs_search": true/false,
-            "search_query": "query if needed",
-            "calculation": {{
-                "operation": "add/subtract/multiply/divide",
-                "numbers": [numbers if needed]
-            }}
+            "search_query": "query if needed"
         }}
         """
         
@@ -124,37 +107,13 @@ class BasicAgent:
         state["needs_more_info"] = analysis.get('needs_search', False)
         state["search_query"] = analysis.get('search_query', '')
         
-        if analysis.get('needs_calculation', False):
-            state["current_step"] = 'calculator'
-            state["tool_output"] = str(analysis['calculation'])
-        elif analysis.get('needs_search', False):
+        if analysis.get('needs_search', False):
             state["current_step"] = 'search'
         else:
             state["current_step"] = 'final_answer'
         
         return state
 
-    def _use_calculator(self, state: AgentState) -> AgentState:
-        """Use the calculator tool."""
-        try:
-            # Create calculator state with input from tool_output
-            calc_input = eval(state["tool_output"])
-            result = self.calculator.invoke({"input": calc_input})
-            
-            state["history"].append({
-                'step': 'calculator',
-                'input': state["tool_output"],
-                'output': str(result['output']['result'])
-            })
-            state["current_step"] = 'final_answer'
-        except Exception as e:
-            state["history"].append({
-                'step': 'calculator_error',
-                'error': str(e)
-            })
-            state["current_step"] = 'final_answer'
-        return state
-
     def _use_search(self, state: AgentState) -> AgentState:
         """Use the search tool."""
         try: