MCP STDIO Integration

from praisonaiagents import Agent, MCP

calculator_agent = Agent(
    instructions="""You are a calculator agent that can perform basic arithmetic operations.""",
    llm="gpt-4o-mini",
    tools=MCP("python calculator_server.py")
)

calculator_agent.start("What is 25 * 16?")

# calculator_server.py
from mcp.server.fastmcp import FastMCP
import logging
import sys

# Set up logging
logging.basicConfig(level=logging.INFO, filename="calculator_server.log")
logger = logging.getLogger("calculator-server")

# Initialize FastMCP server for simple tools
mcp = FastMCP("calculator-tools")

@mcp.tool()
def add(a: float, b: float) -> float:
    """Add two numbers.

    Args:
        a: First number
        b: Second number
    
    Returns:
        The sum of a and b
    """
    logger.info(f"Adding {a} + {b}")
    return a + b

@mcp.tool()
def subtract(a: float, b: float) -> float:
    """Subtract b from a.

    Args:
        a: First number
        b: Second number
    
    Returns:
        The result of a - b
    """
    logger.info(f"Subtracting {b} from {a}")
    return a - b

@mcp.tool()
def multiply(a: float, b: float) -> float:
    """Multiply two numbers.

    Args:
        a: First number
        b: Second number
    
    Returns:
        The product of a and b
    """
    logger.info(f"Multiplying {a} * {b}")
    return a * b

@mcp.tool()
def divide(a: float, b: float) -> float:
    """Divide a by b.

    Args:
        a: First number (numerator)
        b: Second number (denominator)
    
    Returns:
        The result of a / b
    """
    if b == 0:
        raise ValueError("Cannot divide by zero")
    logger.info(f"Dividing {a} / {b}")
    return a / b

if __name__ == "__main__":
    # Run the MCP server using STDIO transport
    mcp.run()

pip install "praisonaiagents[llm]" mcp

export OPENAI_API_KEY="your_api_key"

python calculator_client.py