MCP STDIO Integration

Guide for integrating Standard Input/Output (STDIO) with PraisonAI agents using MCP

Add STDIO Tool to AI Agent

Quick Start

1. Create a client file

   Create a new file calculator_client.py with the following code:

   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?")
   
2. Set Up STDIO MCP Server

   Create a file calculator_server.py with the following code:

   # 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()
   
3. Install Dependencies

   Make sure you have the required packages installed:

   pip install "praisonaiagents[llm]" mcp
   
4. Export API Key

   export OPENAI_API_KEY="your_api_key"
   
5. Run the Agent

   Run the agent which will automatically start the calculator server:

   python calculator_client.py
   

Requirements

• Python 3.10 or higher
• MCP package

Alternative LLM Integrations

Using Groq with STDIO

from praisonaiagents import Agent, MCP

calculator_agent = Agent(
    instructions="""You are a calculator agent that can perform basic arithmetic operations.""",
    llm="groq/llama-3.2-90b-vision-preview",
    tools=MCP("python calculator_server.py")
)

calculator_agent.start("What is 144 divided by 12?")

Using Ollama with STDIO

from praisonaiagents import Agent, MCP

calculator_agent = Agent(
    instructions="""You are a calculator agent that can perform basic arithmetic operations.""",
    llm="ollama/llama3.2",
    tools=MCP("python calculator_server.py")
)

calculator_agent.start("What is 15 + 27? Use the add tool with parameters a and b.")

Gradio UI Integration

Create a Gradio UI for your calculator service:

from praisonaiagents import Agent, MCP
import gradio as gr

def calculate(query):
    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")
    )

    result = calculator_agent.start(query)
    return f"## Calculation Result\n\n{result}"

demo = gr.Interface(
    fn=calculate,
    inputs=gr.Textbox(placeholder="What is 25 * 16?"),
    outputs=gr.Markdown(),
    title="Calculator MCP Agent",
    description="Ask any arithmetic question:"
)

if __name__ == "__main__":
    demo.launch()

Features

Simple Integration

Use standard input/output for easy integration with any tool or service.

Cross-Platform

Works on any operating system that supports Python.

Multiple LLM Options

Use with OpenAI, Groq, Ollama, or other supported LLMs.

Gradio UI

Create user-friendly interfaces for your STDIO integrations.