Resolve a Mathematical equation

[tinmarino]

AI is pretty bad at Math but Python is good at it. So let's make it a Function Calling.

Code

import ast  # To parse the expresion to an AST
import math  # To embed in execution context of the equation
from instructor import OpenAISchema  # Required for SGPT Function Calling definition
from pydantic import Field  # Required for SGPT Function Callings argument definition

class Function(OpenAISchema):
    """
    Resolve a Mathematical expresion with Python. Only functions in the math module are allowed. Do not use other module.
    """
    expression: str = Field(
        ...,
        example="10 + 0x2c * sqrt(25) - 8",
        description="Matematical equation to solve",
    )
    format_output: str = Field(
        None,
        example=".4e",
        description="Optional Python integer format for the output (e.g., '.4e' for scientific notation)"
    )
    class Config:
        title = "solve_mathematical_equation"

    @classmethod
    def execute(cls, expression: str, format_output: str = None) -> str:
        try:
            parsed = ast.parse(expression, mode='eval')
            for node in ast.walk(parsed):
                if isinstance(node, ast.Call):
                    if not isinstance(node.func, ast.Name) or node.func.id not in dir(math):
                        return f"Error: Failed to solve equation {expression}: Invalid function used {node.func}"
            code = compile(parsed, filename='', mode='eval')
            result = eval(
                code,
                {'__builtins__': math},  # Globals
                None  # Locals
            )
            if format_output:
                result = format(result, format_output)
            return str(result)
        except Exception as e:
            return str(result)
        except Exception as e:
            return f"Error: Failled to solve equation {expression}: {e}"

Example

sgpt "10 + 0x2c * sqrt(25) - 8"
@FunctionCall solve_mathematical_equation(expression="10 + 0x2c * sqrt(25) - 8")
222.0
The result of the expression (10 + 0x2c \times \sqrt{25} - 8) is (222.0).