MCP Tools

from server import mcp

@mcp.tool()
def multiply(first_val: float, second_val: float) -> float:
    '''
    Calculate the product of two numbers

    Args:
        first_val: the first value to be multiplied
        second_val: the second value to be multiplied
    '''
    return round(first_val * second_val, 4)

def my_tool(input_text: str, count: int) -> list[str]:
    # Implementation
    pass

@mcp.tool()
def search_database(query: str, limit: int = 10) -> list[dict]:
    """
    Search the database for matching records

    Args:
        query: Search query string
        limit: Maximum number of results to return (default: 10)

    Returns:
        List of matching records as dictionaries
    """
    # Implementation
    pass

from .math import multiply
from .search import search_database

__all__ = ['multiply', 'search_database']

from fastmcp import FastMCP

mcp = FastMCP("OsmosisTools")

@mcp.get("/health")
async def health():
    return {"status": "healthy"}

import argparse
from server.mcp_server import mcp
from tools import *  # Import all tools

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--host", default="0.0.0.0")
    parser.add_argument("--port", type=int, default=8080)
    args = parser.parse_args()

    mcp.run(transport="http", host=args.host, port=args.port)

from server import mcp
from typing import Optional
import json

@mcp.tool()
def fetch_user_profile(
    user_id: str,
    include_history: bool = False,
    max_items: Optional[int] = None
) -> dict:
    """
    Fetch a user's profile information from the database

    This tool retrieves user profile data and optionally includes
    their activity history. Use this when you need to look up
    information about a specific user.

    Args:
        user_id: Unique identifier for the user
        include_history: Whether to include activity history (default: False)
        max_items: Maximum number of history items to return (optional)

    Returns:
        Dictionary containing user profile data with keys:
        - id: User ID
        - name: User's full name
        - email: User's email address
        - history: Activity history (if include_history=True)

    Raises:
        ValueError: If user_id is not found
    """
    # Your implementation here
    profile = {
        "id": user_id,
        "name": "John Doe",
        "email": "john@example.com"
    }

    if include_history:
        history = get_user_history(user_id, max_items)
        profile["history"] = history

    return profile

python mcp/main.py

python mcp/test/test.py

# tests/test_math.py
from mcp.tools.math import multiply

def test_multiply():
    assert multiply(2.0, 3.0) == 6.0
    assert multiply(0.5, 0.5) == 0.25
    assert multiply(-2.0, 3.0) == -6.0

# Good - focused functionality
@mcp.tool()
def calculate_average(numbers: list[float]) -> float:
    """Calculate the arithmetic mean of a list of numbers"""
    return sum(numbers) / len(numbers)

# Avoid - doing too much
@mcp.tool()
def analyze_data(data: list[float]) -> dict:
    """Calculate average, median, mode, stddev, and generate charts"""
    # Too many responsibilities
    pass

# Good
@mcp.tool()
def convert_currency(amount: float, from_code: str, to_code: str) -> float:
    pass

# Avoid
@mcp.tool()
def currency(a: float, f: str, t: str) -> float:
    pass

@mcp.tool()
def divide(numerator: float, denominator: float) -> float:
    """
    Divide two numbers

    Args:
        numerator: The number to be divided
        denominator: The number to divide by

    Returns:
        The quotient

    Raises:
        ValueError: If denominator is zero
    """
    if denominator == 0:
        raise ValueError("Cannot divide by zero")
    return numerator / denominator

from typing import Optional, Union, Literal

@mcp.tool()
def format_text(
    text: str,
    style: Literal["uppercase", "lowercase", "title"] = "title",
    max_length: Optional[int] = None
) -> str:
    """Format text according to specified style"""
    # Implementation
    pass