adding math tool - fixing prompts

manage_agents.py

@@ -17,6 +17,7 @@ from tools import (
     ExcelTool, 
     WebSearchTool, 
     ArvixSearchTool,
+    MathTool,
     PythonTool,
     ContentProcessingError
 )
@@ -45,7 +46,8 @@ class ContentTypeAgent:
             "excel": ExcelTool(),
             "web": WebSearchTool(),
             "arvix": ArvixSearchTool(),
-            "python": PythonTool()
+            "python": PythonTool(),
+            "math": MathTool()
         }
         
         self.type_identification_prompt = PromptTemplate(
@@ -61,6 +63,7 @@ class ContentTypeAgent:
             - excel: If the question refers to an Excel file or contains a task ID for Excel
             - web: If the question is about a research paper, academic work, or requires searching the broader web (including news, journals, or scholarly articles)
             - python: If the question refers to a Python file or contains a task ID for Python
+            - math: If the question involves an operation table, commutativity, associativity, or similar algebraic property
             
             Consider these special cases:
             1. If the question asks to search in Wikipedia, use "wiki"
@@ -143,6 +146,10 @@ class ContentTypeAgent:
         if content_type == "wiki" and 'wikipedia' not in question_lower:
             content_type = "web"
             
+        # Add detection for math/table/commutativity questions
+        if any(word in question_lower for word in ["commutative", "associative", "operation table", "table defining *", "counter-examples", "algebraic property"]):
+            content_type = "math"
+            
         return content_type, question, task_id
 
 class ProcessContentAgent:
@@ -302,6 +309,7 @@ class StateGraphAgent:
         self.excel_tool = ExcelTool()
         self.python_tool = PythonTool()
         self.arvix_tool = ArvixSearchTool()
+        self.math_tool = MathTool()
         
         # Create a dictionary of tools for easy access
         self.tools = {
@@ -312,7 +320,8 @@ class StateGraphAgent:
             "audio": self.audio_tool,
             "excel": self.excel_tool,
             "python": self.python_tool,
-            "arvix": self.arvix_tool
+            "arvix": self.arvix_tool,
+            "math": self.math_tool
         }
         
         # Tool usage tracking
@@ -479,7 +488,24 @@ class StateGraphAgent:
                         "required": ["task_id", "question"]
                     }
                 }
-            }
+            },
+            "math": {
+                "type": "function",
+                "function": {
+                    "name": "analyze_math",
+                    "description": "Analyze an operation table for algebraic properties (e.g., commutativity)",
+                    "parameters": {
+                        "type": "object",
+                        "properties": {
+                            "question": {
+                                "type": "string",
+                                "description": "The question containing the operation table and set S"
+                            }
+                        },
+                        "required": ["question"]
+                    }
+                }
+            },
         }
     
     def _identify_content_type(self, question, file_name, task_id):
@@ -590,6 +616,10 @@ class StateGraphAgent:
             question = args.get("question", "")
             self.last_used_tool = "python"
             result = self.python_tool._run(task_id, question=question)
+        elif tool_name == "analyze_math":
+            question = args.get("question", "")
+            self.last_used_tool = "math"
+            result = self.math_tool._run(question)
         else:
             result = f"Unknown tool: {tool_name}"
             
@@ -670,16 +700,31 @@ class StateGraphAgent:
         try:
             # Reset tool tracking
             self.last_used_tool = None
-            
-            # Try answering directly first
+
+            # Detect math/table/commutativity questions and always use MathTool
+            math_keywords = [
+                "commutative", "associative", "operation table", "table defining *", "counter-examples", "algebraic property"
+            ]
+            question_lower = question.lower()
+            if any(word in question_lower for word in math_keywords):
+                self.last_used_tool = "math"
+                return self.math_tool._run(question)
+
+            # Identify content type
+            content_type, content_parameter = self._identify_content_type(question, file_name, task_id)
+
+            # If it's a math/table question, use MathTool directly (redundant, but safe)
+            if content_type == "math":
+                self.last_used_tool = "math"
+                result = self.math_tool._run(question)
+                return result
+
+            # Otherwise, try direct answer first
             direct_answer = self._direct_answer_attempt(question)
             if direct_answer:
                 self.last_used_tool = "direct"
                 return direct_answer
             
-            # Identify content type
-            content_type, content_parameter = self._identify_content_type(question, file_name, task_id)
-            
             # For file-based questions, use the appropriate tool directly
             if file_name:
                 if content_type in self.file_tools:
@@ -900,6 +945,7 @@ class MainAgent:
         self.audio_tool = AudioTool()
         self.excel_tool = ExcelTool()
         self.python_tool = PythonTool()
+        self.math_tool = MathTool()
         
         # Create a dictionary of tools for easy access
         self.tools = {
@@ -911,6 +957,7 @@ class MainAgent:
             "audio": self.audio_tool,
             "excel": self.excel_tool,
             "python": self.python_tool,
+            "math": self.math_tool
         }
         
         # Tool usage tracking
@@ -1301,6 +1348,9 @@ class MainAgent:
                 task_id = args.get("task_id", "")
                 question = args.get("question", "")
                 result = self.python_tool._run(task_id, question=question)
+            elif tool_name == "analyze_math":
+                question = args.get("question", "")
+                result = self.math_tool._run(question)
             else:
                 result = f"Unknown tool: {tool_name}"
                 
@@ -1333,7 +1383,7 @@ class MainAgent:
                 
             question = state.messages[0].content
             query = f"""Analyze the question, understand the instructions, the context.
-            If you can answer this directly (like math, text reversal, etc), provide the answer.
+            If you can answer this directly without using any tools, provide the answer.
             Otherwise respond with 'TOOLS_REQUIRED'
 
             Question: {question}
@@ -1397,11 +1447,26 @@ class MainAgent:
             - If the answer is a list, output only the list as requested (e.g., comma-separated, one per line, etc.).
             - If the answer is: how many wheels does the car have?, output only the number, not a sentence.
 
+            Example:
+            What's the number of the racing car?
             Context:
-            {tool_output}
+            Racing car number: 1234567890
+            Answer:
+            1234567890
+            
+            How many birds are in the picture?
+            Context:
+            There are 10 birds in the picture.
+            Answer:
+            10
 
             Question:
             {question}
+            Context:
+            {tool_output}
+
+            If the question asks what was told in the context, answer it directly. You don't need the name of the person, just the answer.
+            The answer should be exactly what was said in the context.
             """
             response = self.llm.invoke(answer_prompt)
             answer = response.content if hasattr(response, 'content') else str(response)

tools.py

@@ -800,3 +800,42 @@ class WebSearchTool(BaseTool):
             return search_result[:10000]
         except Exception as e:
             return f"Error searching the web: {str(e)}"
+
+class MathTool(BaseTool):
+    """Tool for analyzing operation tables for algebraic properties (e.g., commutativity)."""
+    name: str = "math_tool"
+    description: str = "Analyze operation tables for algebraic properties (e.g., commutativity)."
+
+    def _run(self, question: str) -> str:
+        import re
+        # Extract the set S
+        set_match = re.search(r'\{([a-zA-Z0-9_,\s]+)\}', question)
+        if not set_match:
+            return "Could not find set."
+        S = [x.strip() for x in set_match.group(1).split(',')]
+        # Extract the markdown table (find all lines that start with | and have at least 2 |'s)
+        table_lines = [line for line in question.splitlines() if line.strip().startswith('|') and line.count('|') > 2]
+        if not table_lines:
+            return "Could not find operation table."
+        # Remove separator row (contains only dashes and pipes)
+        table_lines = [line for line in table_lines if not set(line.replace('|', '').strip()) <= set('-')]
+        if not table_lines:
+            return "Could not find operation table after removing separator."
+        # Parse header
+        header = [cell.strip() for cell in table_lines[0].strip('|').split('|')][1:]
+        table = {}
+        for line in table_lines[1:]:
+            row = [cell.strip() for cell in line.strip('|').split('|')]
+            row_label = row[0]
+            table[row_label] = {col: val for col, val in zip(header, row[1:])}
+        # Check commutativity
+        involved = set()
+        for x in S:
+            for y in S:
+                if x != y:
+                    xy = table[x][y]
+                    yx = table[y][x]
+                    if xy != yx:
+                        involved.update([x, y, xy, yx])
+        involved = sorted([z for z in involved if z in S])
+        return ', '.join(involved)