drain-mcp 0.1.0

package/dist/tools/chat.js

@@ -0,0 +1,118 @@
+/**
+ * Chat Tool
+ *
+ * The main tool: Make AI chat requests with DRAIN payment.
+ */
+/**
+ * Make a chat completion request
+ */
+export async function chat(channelService, providerService, inferenceService, args) {
+    const channelId = args.channelId;
+    // Get channel to find provider
+    const channel = await channelService.getChannel(channelId);
+    if (channel.isExpired) {
+        throw new Error('Channel has expired. Open a new channel to continue.');
+    }
+    // Find provider by address
+    const providers = await providerService.getProviders();
+    const provider = providers.find(p => p.providerAddress.toLowerCase() === channel.provider.toLowerCase());
+    if (!provider) {
+        throw new Error(`Provider ${channel.provider} not found in directory. The channel may be for an unlisted provider.`);
+    }
+    // Check model is available
+    const modelInfo = provider.models.find(m => m.id === args.model);
+    if (!modelInfo) {
+        const availableModels = provider.models.map(m => m.id).join(', ');
+        throw new Error(`Model "${args.model}" not available from this provider. Available: ${availableModels}`);
+    }
+    // Make the request
+    const response = await inferenceService.chat(provider, channelId, {
+        model: args.model,
+        messages: args.messages,
+        max_tokens: args.maxTokens,
+        temperature: args.temperature,
+    });
+    // Format response
+    const assistantMessage = response.choices[0]?.message?.content ?? '';
+    const cost = response.drain?.cost ?? 'unknown';
+    const totalSpent = response.drain?.totalSpent ?? 'unknown';
+    return `${assistantMessage}
+
+---
+*Cost: $${formatCost(cost)} USDC | Total spent: $${formatCost(totalSpent)} USDC | Tokens: ${response.usage.total_tokens}*`;
+}
+/**
+ * Format cost for display
+ */
+function formatCost(cost) {
+    const num = parseFloat(cost) / 1_000_000; // Convert from wei to USDC
+    if (isNaN(num))
+        return cost;
+    if (num < 0.0001)
+        return num.toFixed(6);
+    if (num < 0.01)
+        return num.toFixed(4);
+    return num.toFixed(4);
+}
+// Tool definition for MCP
+export const chatTools = [
+    {
+        name: 'drain_chat',
+        description: `Send a chat completion request to an AI provider through DRAIN.
+
+This is the main tool for AI inference. It:
+1. Uses your open payment channel
+2. Automatically signs a payment voucher
+3. Sends the request to the provider
+4. Returns the AI response
+
+PREREQUISITES:
+- You must have an open channel (use drain_open_channel first)
+- The channel must have sufficient balance
+- The channel must not be expired
+
+The cost is automatically deducted from your channel balance.`,
+        inputSchema: {
+            type: 'object',
+            properties: {
+                channelId: {
+                    type: 'string',
+                    description: 'The payment channel ID to use (0x...)',
+                },
+                model: {
+                    type: 'string',
+                    description: 'Model ID to use (e.g., "gpt-4o", "gpt-4o-mini")',
+                },
+                messages: {
+                    type: 'array',
+                    description: 'Chat messages in OpenAI format',
+                    items: {
+                        type: 'object',
+                        properties: {
+                            role: {
+                                type: 'string',
+                                enum: ['system', 'user', 'assistant'],
+                                description: 'Message role',
+                            },
+                            content: {
+                                type: 'string',
+                                description: 'Message content',
+                            },
+                        },
+                        required: ['role', 'content'],
+                    },
+                },
+                maxTokens: {
+                    type: 'number',
+                    description: 'Maximum tokens to generate (optional)',
+                },
+                temperature: {
+                    type: 'number',
+                    description: 'Sampling temperature 0-2 (optional)',
+                },
+            },
+            required: ['channelId', 'model', 'messages'],
+        },
+    },
+];
+//# sourceMappingURL=chat.js.map

package/dist/tools/chat.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat.js","sourceRoot":"","sources":["../../src/tools/chat.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,IAAI,CACxB,cAA8B,EAC9B,eAAgC,EAChC,gBAAkC,EAClC,IAMC;IAED,MAAM,SAAS,GAAG,IAAI,CAAC,SAAiB,CAAC;IAEzC,+BAA+B;IAC/B,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;IAE3D,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;IAC1E,CAAC;IAED,2BAA2B;IAC3B,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC,YAAY,EAAE,CAAC;IACvD,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAClC,CAAC,CAAC,eAAe,CAAC,WAAW,EAAE,KAAK,OAAO,CAAC,QAAQ,CAAC,WAAW,EAAE,CACnE,CAAC;IAEF,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,YAAY,OAAO,CAAC,QAAQ,uEAAuE,CAAC,CAAC;IACvH,CAAC;IAED,2BAA2B;IAC3B,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC;IACjE,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,IAAI,KAAK,CAAC,UAAU,IAAI,CAAC,KAAK,kDAAkD,eAAe,EAAE,CAAC,CAAC;IAC3G,CAAC;IAED,mBAAmB;IACnB,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE;QAChE,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,UAAU,EAAE,IAAI,CAAC,SAAS;QAC1B,WAAW,EAAE,IAAI,CAAC,WAAW;KAC9B,CAAC,CAAC;IAEH,kBAAkB;IAClB,MAAM,gBAAgB,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;IACrE,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,EAAE,IAAI,IAAI,SAAS,CAAC;IAC/C,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,EAAE,UAAU,IAAI,SAAS,CAAC;IAE3D,OAAO,GAAG,gBAAgB;;;UAGlB,UAAU,CAAC,IAAI,CAAC,yBAAyB,UAAU,CAAC,UAAU,CAAC,mBAAmB,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC;AAC3H,CAAC;AAED;;GAEG;AACH,SAAS,UAAU,CAAC,IAAY;IAC9B,MAAM,GAAG,GAAG,UAAU,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC,2BAA2B;IACrE,IAAI,KAAK,CAAC,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IAC5B,IAAI,GAAG,GAAG,MAAM;QAAE,OAAO,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACxC,IAAI,GAAG,GAAG,IAAI;QAAE,OAAO,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACtC,OAAO,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AACxB,CAAC;AAED,0BAA0B;AAC1B,MAAM,CAAC,MAAM,SAAS,GAAG;IACvB;QACE,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE;;;;;;;;;;;;;8DAa6C;QAC1D,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,SAAS,EAAE;oBACT,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,uCAAuC;iBACrD;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iDAAiD;iBAC/D;gBACD,QAAQ,EAAE;oBACR,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,gCAAgC;oBAC7C,KAAK,EAAE;wBACL,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,IAAI,EAAE;gCACJ,IAAI,EAAE,QAAQ;gCACd,IAAI,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,WAAW,CAAC;gCACrC,WAAW,EAAE,cAAc;6BAC5B;4BACD,OAAO,EAAE;gCACP,IAAI,EAAE,QAAQ;gCACd,WAAW,EAAE,iBAAiB;6BAC/B;yBACF;wBACD,QAAQ,EAAE,CAAC,MAAM,EAAE,SAAS,CAAC;qBAC9B;iBACF;gBACD,SAAS,EAAE;oBACT,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,uCAAuC;iBACrD;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qCAAqC;iBACnD;aACF;YACD,QAAQ,EAAE,CAAC,WAAW,EAAE,OAAO,EAAE,UAAU,CAAC;SAC7C;KACF;CACF,CAAC"}

package/dist/tools/providers.d.ts

@@ -0,0 +1,54 @@
+/**
+ * Provider Discovery Tools
+ *
+ * Tools for discovering and inspecting DRAIN AI providers.
+ */
+import type { ProviderService } from '../services/provider.js';
+/**
+ * List all available providers
+ */
+export declare function listProviders(providerService: ProviderService, args: {
+    onlineOnly?: boolean;
+    model?: string;
+}): Promise<string>;
+/**
+ * Get details for a specific provider
+ */
+export declare function getProvider(providerService: ProviderService, args: {
+    providerId: string;
+}): Promise<string>;
+export declare const providerTools: ({
+    name: string;
+    description: string;
+    inputSchema: {
+        type: string;
+        properties: {
+            onlineOnly: {
+                type: string;
+                description: string;
+            };
+            model: {
+                type: string;
+                description: string;
+            };
+            providerId?: undefined;
+        };
+        required?: undefined;
+    };
+} | {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: string;
+        properties: {
+            providerId: {
+                type: string;
+                description: string;
+            };
+            onlineOnly?: undefined;
+            model?: undefined;
+        };
+        required: string[];
+    };
+})[];
+//# sourceMappingURL=providers.d.ts.map

package/dist/tools/providers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"providers.d.ts","sourceRoot":"","sources":["../../src/tools/providers.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,eAAe,EAAY,MAAM,yBAAyB,CAAC;AA0BzE;;GAEG;AACH,wBAAsB,aAAa,CACjC,eAAe,EAAE,eAAe,EAChC,IAAI,EAAE;IAAE,UAAU,CAAC,EAAE,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,GAC7C,OAAO,CAAC,MAAM,CAAC,CAqBjB;AAED;;GAEG;AACH,wBAAsB,WAAW,CAC/B,eAAe,EAAE,eAAe,EAChC,IAAI,EAAE;IAAE,UAAU,EAAE,MAAM,CAAA;CAAE,GAC3B,OAAO,CAAC,MAAM,CAAC,CAQjB;AAGD,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAuCzB,CAAC"}

package/dist/tools/providers.js

@@ -0,0 +1,102 @@
+/**
+ * Provider Discovery Tools
+ *
+ * Tools for discovering and inspecting DRAIN AI providers.
+ */
+/**
+ * Format provider for display
+ */
+function formatProvider(p) {
+    const status = p.status.online ? '🟢 ONLINE' : '🔴 OFFLINE';
+    const latency = p.status.latencyMs ? `${p.status.latencyMs}ms` : 'N/A';
+    const models = p.models.map(m => `  - ${m.name} ($${m.pricing.inputPer1kTokens}/$${m.pricing.outputPer1kTokens} per 1k tokens)`).join('\n');
+    return `
+## ${p.name}
+- **ID:** ${p.id}
+- **Status:** ${status}
+- **Latency:** ${latency}
+- **Address:** ${p.providerAddress}
+- **API:** ${p.apiUrl}
+- **Chain:** ${p.chainId === 137 ? 'Polygon Mainnet' : 'Polygon Amoy'}
+
+**Description:** ${p.description}
+
+**Models:**
+${models}
+`.trim();
+}
+/**
+ * List all available providers
+ */
+export async function listProviders(providerService, args) {
+    let providers;
+    if (args.model) {
+        providers = await providerService.findByModel(args.model);
+    }
+    else if (args.onlineOnly) {
+        providers = await providerService.getOnlineProviders();
+    }
+    else {
+        providers = await providerService.getProviders();
+    }
+    if (providers.length === 0) {
+        if (args.model) {
+            return `No providers found supporting model "${args.model}".`;
+        }
+        return 'No providers available in the directory.';
+    }
+    const formatted = providers.map(formatProvider).join('\n\n---\n\n');
+    return `# DRAIN AI Providers\n\nFound ${providers.length} provider(s):\n\n${formatted}`;
+}
+/**
+ * Get details for a specific provider
+ */
+export async function getProvider(providerService, args) {
+    const provider = await providerService.getProvider(args.providerId);
+    if (!provider) {
+        return `Provider "${args.providerId}" not found.`;
+    }
+    return formatProvider(provider);
+}
+// Tool definitions for MCP
+export const providerTools = [
+    {
+        name: 'drain_providers',
+        description: `List available DRAIN AI providers. 
+    
+These are AI services that accept DRAIN micropayments for inference.
+Use this to discover which providers and models are available before opening a channel.
+
+Returns: List of providers with their models, pricing, and status.`,
+        inputSchema: {
+            type: 'object',
+            properties: {
+                onlineOnly: {
+                    type: 'boolean',
+                    description: 'If true, only return providers that are currently online',
+                },
+                model: {
+                    type: 'string',
+                    description: 'Filter providers by model name (e.g., "gpt-4o", "gpt-3.5-turbo")',
+                },
+            },
+        },
+    },
+    {
+        name: 'drain_provider_info',
+        description: `Get detailed information about a specific DRAIN provider.
+    
+Returns: Provider details including all available models and their pricing.`,
+        inputSchema: {
+            type: 'object',
+            properties: {
+                providerId: {
+                    type: 'string',
+                    description: 'The provider ID to look up',
+                },
+            },
+            required: ['providerId'],
+        },
+    },
+];
+//# sourceMappingURL=providers.js.map

package/dist/tools/providers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"providers.js","sourceRoot":"","sources":["../../src/tools/providers.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH;;GAEG;AACH,SAAS,cAAc,CAAC,CAAW;IACjC,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,CAAC;IAC5D,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;IACvE,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,OAAO,CAAC,gBAAgB,KAAK,CAAC,CAAC,OAAO,CAAC,iBAAiB,iBAAiB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAE5I,OAAO;KACJ,CAAC,CAAC,IAAI;YACC,CAAC,CAAC,EAAE;gBACA,MAAM;iBACL,OAAO;iBACP,CAAC,CAAC,eAAe;aACrB,CAAC,CAAC,MAAM;eACN,CAAC,CAAC,OAAO,KAAK,GAAG,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,cAAc;;mBAElD,CAAC,CAAC,WAAW;;;EAG9B,MAAM;CACP,CAAC,IAAI,EAAE,CAAC;AACT,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,eAAgC,EAChC,IAA8C;IAE9C,IAAI,SAAqB,CAAC;IAE1B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,SAAS,GAAG,MAAM,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5D,CAAC;SAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;QAC3B,SAAS,GAAG,MAAM,eAAe,CAAC,kBAAkB,EAAE,CAAC;IACzD,CAAC;SAAM,CAAC;QACN,SAAS,GAAG,MAAM,eAAe,CAAC,YAAY,EAAE,CAAC;IACnD,CAAC;IAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,OAAO,wCAAwC,IAAI,CAAC,KAAK,IAAI,CAAC;QAChE,CAAC;QACD,OAAO,0CAA0C,CAAC;IACpD,CAAC;IAED,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAEpE,OAAO,iCAAiC,SAAS,CAAC,MAAM,oBAAoB,SAAS,EAAE,CAAC;AAC1F,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,eAAgC,EAChC,IAA4B;IAE5B,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAEpE,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,OAAO,aAAa,IAAI,CAAC,UAAU,cAAc,CAAC;IACpD,CAAC;IAED,OAAO,cAAc,CAAC,QAAQ,CAAC,CAAC;AAClC,CAAC;AAED,2BAA2B;AAC3B,MAAM,CAAC,MAAM,aAAa,GAAG;IAC3B;QACE,IAAI,EAAE,iBAAiB;QACvB,WAAW,EAAE;;;;;mEAKkD;QAC/D,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,0DAA0D;iBACxE;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,kEAAkE;iBAChF;aACF;SACF;KACF;IACD;QACE,IAAI,EAAE,qBAAqB;QAC3B,WAAW,EAAE;;4EAE2D;QACxE,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,4BAA4B;iBAC1C;aACF;YACD,QAAQ,EAAE,CAAC,YAAY,CAAC;SACzB;KACF;CACF,CAAC"}

package/package.json

@@ -0,0 +1,43 @@
+{
+  "name": "drain-mcp",
+  "version": "0.1.0",
+  "description": "MCP Server for DRAIN Protocol - Autonomous AI payments",
+  "type": "module",
+  "main": "dist/index.js",
+  "bin": {
+    "drain-mcp": "./dist/index.js"
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "start": "node dist/index.js",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "mcp",
+    "drain",
+    "ai",
+    "payments",
+    "micropayments",
+    "polygon",
+    "usdc",
+    "agent"
+  ],
+  "author": "",
+  "license": "MIT",
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "viem": "^2.21.0"
+  },
+  "devDependencies": {
+    "@types/node": "^22.0.0",
+    "typescript": "^5.6.0"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ]
+}