@mcp-ts/sdk 1.3.6 → 1.3.9

package/src/adapters/mastra-adapter.ts

@@ -1,126 +1,126 @@
-import { MCPClient } from '../server/mcp/oauth-client';
-import { MultiSessionClient } from '../server/mcp/multi-session-client';
-import type { z } from 'zod';
-
-export interface MastraAdapterOptions {
-    /** 
-     * Prefix for tool names to avoid collision with other tools.
-     * Defaults to the client's serverId.
-     */
-    prefix?: string;
-}
-
-/**
- * Interface definition for a Mastra tool since we might not have the SDK installed.
- * Based on Mastra documentation.
- */
-export interface MastraTool {
-    id: string;
-    description: string;
-    inputSchema: z.ZodType<any>;
-    execute: (args: any) => Promise<any>;
-}
-
-/**
- * Adapter to use MCP tools within Mastra agents.
- */
-export class MastraAdapter {
-    private z: typeof z | undefined;
-
-    constructor(
-        private client: MCPClient | MultiSessionClient,
-        private options: MastraAdapterOptions = {}
-    ) { }
-
-    /**
-     * Lazy-loads Zod dependency
-     */
-    private async ensureZod() {
-        if (!this.z) {
-            try {
-                const zod = await import('zod');
-                this.z = zod.z;
-            } catch (error) {
-                throw new Error(
-                    'zod is not installed. Install with:\n' +
-                    '  npm install zod'
-                );
-            }
-        }
-    }
-
-
-
-    private async transformTools(client: MCPClient): Promise<Record<string, MastraTool>> {
-        if (!client.isConnected()) {
-            return {};
-        }
-
-        await this.ensureZod();
-
-        const result = await client.listTools();
-        const prefix = this.options.prefix ?? client.getServerId() ?? 'mcp';
-        const tools: Record<string, MastraTool> = {};
-
-        for (const tool of result.tools) {
-            const toolName = `${prefix}_${tool.name}`;
-
-            // In a real implementation, you would use a library like 'json-schema-to-zod'
-            const schema = this.jsonSchemaToZod(tool.inputSchema);
-
-            tools[toolName] = {
-                id: toolName,
-                description: tool.description || `Tool ${tool.name}`,
-                inputSchema: schema,
-                execute: async (args: any) => {
-                    return await client.callTool(tool.name, args);
-                },
-            };
-        }
-
-        return tools;
-    }
-
-    private jsonSchemaToZod(schema: any): z.ZodType<any> {
-        try {
-            const { parseSchema } = require('json-schema-to-zod');
-            const zodSchemaString = parseSchema(schema);
-            // eslint-disable-next-line
-            return new Function('z', 'return ' + zodSchemaString)(this.z);
-        } catch (error) {
-            // Fallback: Accept any object if conversion fails
-            console.warn('[MastraAdapter] Failed to convert JSON Schema to Zod, using fallback:', error);
-            return this.z!.record(this.z!.any()).optional().describe("Dynamic Input");
-        }
-    }
-
-    /**
-     * Fetches tools from the MCP server and converts them to Mastra tools.
-     */
-    async getTools(): Promise<Record<string, MastraTool>> {
-        // Use duck typing instead of instanceof to handle module bundling issues
-        const isMultiSession = typeof (this.client as any).getClients === 'function';
-        const clients = isMultiSession
-            ? (this.client as MultiSessionClient).getClients()
-            : [this.client as MCPClient];
-
-        const results = await Promise.all(
-            clients.map(async (client) => {
-                try {
-                    return await this.transformTools(client);
-                } catch (error) {
-                    console.error(`[MastraAdapter] Failed to fetch tools from ${client.getServerId()}:`, error);
-                    return {};
-                }
-            })
-        );
-        return results.reduce((acc, tools) => ({ ...acc, ...tools }), {});
-    }
-
-    /**
-     * Convenience static method to fetch tools in a single line.
-     */
-    static async getTools(client: MCPClient | MultiSessionClient, options: MastraAdapterOptions = {}): Promise<Record<string, MastraTool>> {
-        return new MastraAdapter(client, options).getTools();
-    }
-}
+import { MCPClient } from '../server/mcp/oauth-client';
+import { MultiSessionClient } from '../server/mcp/multi-session-client';
+import type { z } from 'zod';
+
+export interface MastraAdapterOptions {
+    /** 
+     * Prefix for tool names to avoid collision with other tools.
+     * Defaults to the client's serverId.
+     */
+    prefix?: string;
+}
+
+/**
+ * Interface definition for a Mastra tool since we might not have the SDK installed.
+ * Based on Mastra documentation.
+ */
+export interface MastraTool {
+    id: string;
+    description: string;
+    inputSchema: z.ZodType<any>;
+    execute: (args: any) => Promise<any>;
+}
+
+/**
+ * Adapter to use MCP tools within Mastra agents.
+ */
+export class MastraAdapter {
+    private z: typeof z | undefined;
+
+    constructor(
+        private client: MCPClient | MultiSessionClient,
+        private options: MastraAdapterOptions = {}
+    ) { }
+
+    /**
+     * Lazy-loads Zod dependency
+     */
+    private async ensureZod() {
+        if (!this.z) {
+            try {
+                const zod = await import('zod');
+                this.z = zod.z;
+            } catch (error) {
+                throw new Error(
+                    'zod is not installed. Install with:\n' +
+                    '  npm install zod'
+                );
+            }
+        }
+    }
+
+
+
+    private async transformTools(client: MCPClient): Promise<Record<string, MastraTool>> {
+        if (!client.isConnected()) {
+            return {};
+        }
+
+        await this.ensureZod();
+
+        const result = await client.listTools();
+        const prefix = this.options.prefix ?? client.getServerId()?.replace(/-/g, '').substring(0, 8) ?? 'mcp';
+        const tools: Record<string, MastraTool> = {};
+
+        for (const tool of result.tools) {
+            const toolName = `${prefix}_${tool.name}`;
+
+            // In a real implementation, you would use a library like 'json-schema-to-zod'
+            const schema = this.jsonSchemaToZod(tool.inputSchema);
+
+            tools[toolName] = {
+                id: toolName,
+                description: tool.description || `Tool ${tool.name}`,
+                inputSchema: schema,
+                execute: async (args: any) => {
+                    return await client.callTool(tool.name, args);
+                },
+            };
+        }
+
+        return tools;
+    }
+
+    private jsonSchemaToZod(schema: any): z.ZodType<any> {
+        try {
+            const { parseSchema } = require('json-schema-to-zod');
+            const zodSchemaString = parseSchema(schema);
+            // eslint-disable-next-line
+            return new Function('z', 'return ' + zodSchemaString)(this.z);
+        } catch (error) {
+            // Fallback: Accept any object if conversion fails
+            console.warn('[MastraAdapter] Failed to convert JSON Schema to Zod, using fallback:', error);
+            return this.z!.record(this.z!.any()).optional().describe("Dynamic Input");
+        }
+    }
+
+    /**
+     * Fetches tools from the MCP server and converts them to Mastra tools.
+     */
+    async getTools(): Promise<Record<string, MastraTool>> {
+        // Use duck typing instead of instanceof to handle module bundling issues
+        const isMultiSession = typeof (this.client as any).getClients === 'function';
+        const clients = isMultiSession
+            ? (this.client as MultiSessionClient).getClients()
+            : [this.client as MCPClient];
+
+        const results = await Promise.all(
+            clients.map(async (client) => {
+                try {
+                    return await this.transformTools(client);
+                } catch (error) {
+                    console.error(`[MastraAdapter] Failed to fetch tools from ${client.getServerId()}:`, error);
+                    return {};
+                }
+            })
+        );
+        return results.reduce((acc, tools) => ({ ...acc, ...tools }), {});
+    }
+
+    /**
+     * Convenience static method to fetch tools in a single line.
+     */
+    static async getTools(client: MCPClient | MultiSessionClient, options: MastraAdapterOptions = {}): Promise<Record<string, MastraTool>> {
+        return new MastraAdapter(client, options).getTools();
+    }
+}

package/src/bin/mcp-ts.ts

@@ -1,102 +1,102 @@
-#!/usr/bin/env node
-import * as fs from 'fs';
-import * as path from 'path';
-
-/**
- * MCP-TS CLI Utility
- * 
- * Provides helper commands for users of the @mcp-ts/sdk library.
- */
-async function run() {
-    const args = process.argv.slice(2);
-    const command = args[0];
-
-    if (command === 'supabase-init') {
-        await initSupabase();
-    } else {
-        showHelp();
-    }
-}
-
-function showHelp() {
-    console.log(`
-🚀 MCP-TS CLI Utility
-Usage: npx mcp-ts <command>
-
-Commands:
-  supabase-init    Initialize Supabase migrations in your project
-    `);
-}
-
-async function initSupabase() {
-    console.log('🚀 Initializing Supabase storage for MCP-TS...');
-
-    // When running from dist/bin/mcp-ts.js (compiled), __dirname is dist/bin.
-    // The supabase/ migrations are at the root of the package.
-    // We need to look up two levels to find 'supabase' folder in the package.
-    const pkgRoot = path.resolve(__dirname, '../..');
-    const sourceDir = path.join(pkgRoot, 'supabase', 'migrations');
-    
-    if (!fs.existsSync(sourceDir)) {
-        console.error(`❌ Error: Could not find migration files in package at: ${sourceDir}`);
-        console.log('Please ensure you are running this from a project where @mcp-ts/sdk is installed.');
-        process.exit(1);
-    }
-
-    const targetDir = path.join(process.cwd(), 'supabase', 'migrations');
-
-    try {
-        if (!fs.existsSync(targetDir)) {
-            fs.mkdirSync(targetDir, { recursive: true });
-            console.log(`📁 Created directory: ${targetDir}`);
-        }
-
-        const files = fs.readdirSync(sourceDir);
-        let copiedCount = 0;
-
-        for (const file of files) {
-            if (file.endsWith('.sql')) {
-                const srcPath = path.join(sourceDir, file);
-                const destPath = path.join(targetDir, file);
-                
-                if (fs.existsSync(destPath)) {
-                    console.log(`⏭️  Skipping existing migration: ${file}`);
-                    continue;
-                }
-
-                fs.copyFileSync(srcPath, destPath);
-                console.log(`✅ Copied: ${file}`);
-                copiedCount++;
-            }
-        }
-
-        if (copiedCount > 0) {
-            console.log('\n✨ Database migrations successfully initialized!');
-            console.log('\nNext steps:');
-            console.log('1. Link your Supabase project:');
-            console.log('   npx supabase link --project-ref <your-project-id>');
-            console.log('\n2. Push the migrations to your remote database:');
-            console.log('   npx supabase db push');
-            console.log('\n3. Add your Supabase credentials to .env:');
-            console.log('   SUPABASE_URL=https://<your-project-id>.supabase.co');
-            console.log('   SUPABASE_SERVICE_ROLE_KEY=<your-service-role-key>');
-            console.log('\n⚠️  Important: Use the service_role key (not the anon key) for server-side storage.');
-            console.log('   The service_role key bypasses RLS policies and is required for mcp-ts to work correctly.');
-            console.log('   Find it in: Supabase Dashboard -> Project Settings -> API -> service_role');
-        } else if (files.length > 0) {
-            console.log('\n👍 All migration files are already present in your project.');
-            console.log('   Ensure SUPABASE_SERVICE_ROLE_KEY (not SUPABASE_ANON_KEY) is set in your .env');
-        } else {
-            console.log('⚠️  No migration files found to copy.');
-        }
-
-    } catch (error: any) {
-        console.error(`❌ Error initializing Supabase: ${error.message}`);
-        process.exit(1);
-    }
-}
-
-run().catch(err => {
-    console.error(err);
-    process.exit(1);
-});
+#!/usr/bin/env node
+import * as fs from 'fs';
+import * as path from 'path';
+
+/**
+ * MCP-TS CLI Utility
+ * 
+ * Provides helper commands for users of the @mcp-ts/sdk library.
+ */
+async function run() {
+    const args = process.argv.slice(2);
+    const command = args[0];
+
+    if (command === 'supabase-init') {
+        await initSupabase();
+    } else {
+        showHelp();
+    }
+}
+
+function showHelp() {
+    console.log(`
+🚀 MCP-TS CLI Utility
+Usage: npx mcp-ts <command>
+
+Commands:
+  supabase-init    Initialize Supabase migrations in your project
+    `);
+}
+
+async function initSupabase() {
+    console.log('🚀 Initializing Supabase storage for MCP-TS...');
+
+    // When running from dist/bin/mcp-ts.js (compiled), __dirname is dist/bin.
+    // The supabase/ migrations are at the root of the package.
+    // We need to look up two levels to find 'supabase' folder in the package.
+    const pkgRoot = path.resolve(__dirname, '../..');
+    const sourceDir = path.join(pkgRoot, 'supabase', 'migrations');
+    
+    if (!fs.existsSync(sourceDir)) {
+        console.error(`❌ Error: Could not find migration files in package at: ${sourceDir}`);
+        console.log('Please ensure you are running this from a project where @mcp-ts/sdk is installed.');
+        process.exit(1);
+    }
+
+    const targetDir = path.join(process.cwd(), 'supabase', 'migrations');
+
+    try {
+        if (!fs.existsSync(targetDir)) {
+            fs.mkdirSync(targetDir, { recursive: true });
+            console.log(`📁 Created directory: ${targetDir}`);
+        }
+
+        const files = fs.readdirSync(sourceDir);
+        let copiedCount = 0;
+
+        for (const file of files) {
+            if (file.endsWith('.sql')) {
+                const srcPath = path.join(sourceDir, file);
+                const destPath = path.join(targetDir, file);
+                
+                if (fs.existsSync(destPath)) {
+                    console.log(`⏭️  Skipping existing migration: ${file}`);
+                    continue;
+                }
+
+                fs.copyFileSync(srcPath, destPath);
+                console.log(`✅ Copied: ${file}`);
+                copiedCount++;
+            }
+        }
+
+        if (copiedCount > 0) {
+            console.log('\n✨ Database migrations successfully initialized!');
+            console.log('\nNext steps:');
+            console.log('1. Link your Supabase project:');
+            console.log('   npx supabase link --project-ref <your-project-id>');
+            console.log('\n2. Push the migrations to your remote database:');
+            console.log('   npx supabase db push');
+            console.log('\n3. Add your Supabase credentials to .env:');
+            console.log('   SUPABASE_URL=https://<your-project-id>.supabase.co');
+            console.log('   SUPABASE_SERVICE_ROLE_KEY=<your-service-role-key>');
+            console.log('\n⚠️  Important: Use the service_role key (not the anon key) for server-side storage.');
+            console.log('   The service_role key bypasses RLS policies and is required for mcp-ts to work correctly.');
+            console.log('   Find it in: Supabase Dashboard -> Project Settings -> API -> service_role');
+        } else if (files.length > 0) {
+            console.log('\n👍 All migration files are already present in your project.');
+            console.log('   Ensure SUPABASE_SERVICE_ROLE_KEY (not SUPABASE_ANON_KEY) is set in your .env');
+        } else {
+            console.log('⚠️  No migration files found to copy.');
+        }
+
+    } catch (error: any) {
+        console.error(`❌ Error initializing Supabase: ${error.message}`);
+        process.exit(1);
+    }
+}
+
+run().catch(err => {
+    console.error(err);
+    process.exit(1);
+});