MCP Server Developers

pip install agentpay-sdk

from agentpay_sdk import AgentPayClient
import os

# Best practice: Store your Service Token as an environment variable
SERVICE_TOKEN = os.getenv("AGENTPAY_SERVICE_TOKEN")

if not SERVICE_TOKEN:
    # In a real application, you might want to log this error
    # and potentially disable paid features if the token is missing.
    raise ValueError("AGENTPAY_SERVICE_TOKEN environment variable not set.")

agentpay_client = AgentPayClient(service_token=SERVICE_TOKEN)

print("AgentPayClient initialized successfully.")

import uuid

# Your tool processing logic
async def process_tool_logic(tool_params: dict):
    # Replace this with your tool's actual operations
    print(f"Executing core logic for the tool with params: {tool_params}")
    # Simulate a successful operation
    return {"result": "Tool logic executed successfully", "params_received": tool_params}

# Your MCP Server tool handler
async def my_paid_tool(tool_params: dict):
    # 1. --- Execute your tool's core logic ---
    print(f"Processing tool with params: {tool_params}")
    tool_output = await process_tool_logic(tool_params)
    print(f"Tool logic completed. Output: {tool_output}")

    # 2. --- If core logic is successful, then attempt to charge ---
    COST_CENTS = 10 # Example: 10 cents for this tool usage
    usage_event_id = str(uuid.uuid4()) # Unique ID for idempotency

    print(f"Attempting to charge {COST_CENTS} cents for tool usage (ID: {usage_event_id})...")
    
    consumption_result = agentpay_client.consume(
        api_key=user_api_key, # Extracted previously from the request (see notes for more info)
        amount_cents=COST_CENTS,
        usage_event_id=usage_event_id
        # You can also pass optional metadata, description, etc.
    )

    if consumption_result.success:
        print(f"Successfully charged.")
        # Return the result of your tool's core logic
        return {"status": "success", "data": tool_output, "payment_info": "Charged successfully"}
    else:
        # If charging failed (e.g. insufficient funds, invalid API key post-tool execution)
        # AgentPay automatically handles complex payment failures, helping you easily handle different scenarios based on your business logic (see "Charging for Usage" guide for more details)
        print(f"CRITICAL: Tool executed but failed to charge: {consumption_result.error_message} (Code: {consumption_result.error_code})")
        return {"status": "error", "data": tool_output, "message": f"Tool executed, but payment failed: {consumption_result.error_message}"}