Metadata-Version: 2.4
Name: mathemagic-ai
Version: 0.1.0
Summary: AI Calculator Agent
Requires-Python: >=3.10
Description-Content-Type: text/markdown
Requires-Dist: anthropic>=0.49.0
Requires-Dist: asyncio>=3.4.3
Requires-Dist: fastmcp>=0.4.1
Requires-Dist: mcp[cli]>=1.5.0
Requires-Dist: pint>=0.24.4
Requires-Dist: pylint>=3.3.6
Requires-Dist: pytest>=8.3.5
Requires-Dist: python-dotenv>=1.1.0
Requires-Dist: typer>=0.15.2

# Mathemagic
An AI calculator agent that interprets and solves science, technology, mathematics, and engineering problems in Python.

<img src="./resources/mathemagic.png" alt="Mathemagic" width="420"/>

## Features
- Converts natural language math problems to Python code using LLMs (Claude is currently the only supported model)
- Handles units using the [Pint](https://github.com/hgrecco/pint/tree/master) library
- Provides access through MCP and CLI interfaces
- Supports complex mathematical operations with proper unit handling

## Examples
- "Convert 100 kilometers per hour to miles per hour"
- "What is the volume of a cylinder with radius 5cm and height 10cm?"
- "If I have 3 apples and 4 oranges, how many pieces of fruit do I have?"
- "Calculate the kinetic energy of a 2kg object moving at 10 meters per second"

## Basic Usage
Enter command to start CLI
```bash
mathemagic
```
CLI initializes, and you type in your math problem
```code
Mathemagic Calculator (Press Ctrl+C to exit)
Enter your math problem:
: Two hundred miles times 634 feet in meters squared.
Processing: Two hundred miles times 634 feet in meters squared.

Result:
62198956.89216 meter ** 2
```

## Installation

```bash
# Clone the repository
git clone https://github.com/yourusername/mathemagic.git
cd mathemagic

# Install dependencies using UV
uv pip install -e .

```
Set up environment variables. Create a `.env` file with your Anthropic API key in the project root directory.
```bash
echo "ANTHROPIC_API_KEY=your_api_key_here" >> .env
```
## Usage
### Using the CLI
```bash
# Run in interactive mode
mathemagic

# Run the CLI with a math problem
mathemagic "100 miles times 234 acres in liters"

# Show the generated Python code along with the result
mathemagic "100 miles times 234 acres in liters" --output-python

# help menu
mathemagic --help
```

#### Output the Executed Python Code
Using the -p or --output-python flags
```bash
mathemagic -p
```
````code
Mathemagic Calculator (Press Ctrl+C to exit)
Enter your math problem:
: Two hundred miles times 634 feet in meters squared.
Processing: Two hundred miles times 634 feet in meters squared.

Generated Python code:
```python
from pint import UnitRegistry

ureg = UnitRegistry()
Q_ = ureg.Quantity

# Define the variables with their respective units
miles = Q_(200, 'mile')  # distance in miles
feet = Q_(634, 'foot')  # distance in feet

# Perform the multiplication
result = miles * feet

# Convert the result to meters squared
result_m2 = result.to('meter**2')

print(result_m2)
```
Result:
62198956.89216 meter ** 2
````

#### Handles Arbitrary Units
mathemagic uses Pint to handle units.
```code
Enter another problem:
: 100 bananas times 66 guavas
Processing: 100 bananas times 66 guavas

Result:
6600 banana * guava
```
### Using the MCP Server
#### Starting the Server
```bash
# Start the MCP server
uv run src/mathemagic/server.py
```
### Add the MCP Server to Claude Code
Assuming you have claude code installed, type this command, then follow the wizard. You'll enter the command above when prompted.
```bash
claude mcp add
```

## Architecture
The application consists of:
- `mathemagic.py`: Core functionality for converting natural language to Python code and executing it
- `server.py`: MCP server that provides API endpoints for the calculator
- `cli.py`: Command-line interface for interacting with the calculator

# TODO
1. Clean up python output through server (currently outputs in a dict and not pretty)
1. Improve CLI (make it pretty)
1. Add symbolic math support
1. Add linear algebra support
1. Update to use openrouter
1. Add tests
1. Astropy support?
1. Auth as alternative to .env for remote use?
