from fastmcp import FastMCP
import numpy as np

mcp = FastMCP()

@mcp.tool()
def numpy_add(a: list[float], b: list[float]) -> list[float]:
    """Adds two vectors element-wise."""
    return np.add(a, b).tolist()

@mcp.tool()
def numpy_multiply(a: list[float], b: list[float]) -> list[float]:
    """Multiplies two vectors element-wise."""
    return np.multiply(a, b).tolist()

if __name__ == "__main__":
    mcp.run()