Example: Adding a Calculator to Our Agent › Extending the Calculator

import re

def calculate_enhanced(expression: str) -> dict:
    """
    Enhanced calculator with better error handling and formatting.
    
    Args:
        expression: A math expression
    
    Returns:
        Dictionary with result or error
    """
    try:
        # Clean the expression
        # Remove common words that might appear
        expression = expression.lower()
        expression = re.sub(r'\b(what is|calculate|compute|equals?)\b', '', expression)
        expression = expression.strip()
        
        # Replace common word operators
        expression = expression.replace('times', '*')
        expression = expression.replace('divided by', '/')
        expression = expression.replace('plus', '+')
        expression = expression.replace('minus', '-')
        expression = expression.replace('to the power of', '**')
        
        # Evaluate
        result = eval(expression)
        
        # Format the result nicely
        if isinstance(result, float):
            # Round to reasonable precision
            if result.is_integer():
                result = int(result)
            else:
                result = round(result, 6)
        
        return {
            "success": True,
            "result": result,
            "expression": expression
        }
    except Exception as e:
        return {
            "success": False,
            "error": f"Could not evaluate '{expression}': {str(e)}",
            "expression": expression
        }

tool_registry = {
    "calculate": calculate_enhanced
}