init

tools/calculator.py

@@ -0,0 +1,27 @@
+from typing import Dict, Any
+from langchain_core.tools import BaseTool
+
+
+class CalculatorTool(BaseTool):
+    """A simple calculator tool for mathematical operations"""
+    
+    name: str = "calculator"
+    description: str = "Perform mathematical calculations. Input should be a mathematical expression like '2 + 2' or '10 * (3 + 5)'"
+    
+    def _run(self, expression: str) -> str:
+        """Evaluate a mathematical expression"""
+        try:
+            # Security: Only allow safe mathematical operations
+            allowed_chars = set("0123456789+-*/(). ")
+            if not all(c in allowed_chars for c in expression):
+                return "Error: Expression contains invalid characters"
+            
+            # Evaluate the expression
+            result = eval(expression)
+            return f"Result: {result}"
+        except Exception as e:
+            return f"Error calculating expression: {str(e)}"
+    
+    async def _arun(self, expression: str) -> str:
+        """Async version of the tool"""
+        return self._run(expression)