mongo.do 0.1.0

package/dist/mcp/adapters/vercel-adapter.js

@@ -0,0 +1,487 @@
+/**
+ * Vercel AI SDK Adapter for MCP
+ *
+ * Adapts MCP server tools to Vercel AI SDK's tool format.
+ * Compatible with the Vercel AI SDK's `tool` and `generateText` APIs.
+ *
+ * Features:
+ * - Converts MCP tool definitions to Vercel AI SDK format
+ * - Supports both `tool()` and `CoreTool` interfaces
+ * - Handles streaming via Vercel's streaming responses
+ * - Integrates with `ai` package for Next.js compatibility
+ *
+ * @see https://sdk.vercel.ai/docs/ai-sdk-core/tools-and-tool-calling
+ */
+import { BaseAdapter } from './base-adapter';
+import { McpError } from './errors';
+import { streamTextContent, } from './streaming';
+import { z } from 'zod';
+// =============================================================================
+// Constants
+// =============================================================================
+const DEFAULT_STREAMING_THRESHOLD = 50 * 1024; // 50KB
+// =============================================================================
+// Vercel Adapter Class
+// =============================================================================
+/**
+ * Vercel AI SDK Adapter
+ *
+ * Converts MCP tools to Vercel AI SDK's CoreTool format and handles
+ * tool execution with streaming support.
+ *
+ * @example
+ * ```typescript
+ * import { generateText, tool } from 'ai'
+ *
+ * const mcpServer = createMcpServer({ dbAccess })
+ * const adapter = new VercelAdapter({ server: mcpServer })
+ *
+ * await adapter.initialize()
+ *
+ * // Get tools for Vercel AI SDK
+ * const tools = await adapter.getTools()
+ *
+ * // Use with generateText
+ * const result = await generateText({
+ *   model: openai('gpt-4'),
+ *   tools,
+ *   prompt: 'Search for documents about AI',
+ * })
+ * ```
+ */
+export class VercelAdapter extends BaseAdapter {
+    streamingThreshold;
+    defaultStreaming;
+    formatResult;
+    toolCache = new Map();
+    mcpToolNames = new Map(); // Vercel name -> MCP name
+    initialized = false;
+    constructor(config) {
+        super(config);
+        this.streamingThreshold = config.streamingThreshold ?? DEFAULT_STREAMING_THRESHOLD;
+        this.defaultStreaming = config.defaultStreaming ?? true;
+        this.formatResult = config.formatResult ?? this.defaultResultFormatter.bind(this);
+    }
+    /**
+     * Adapter name for logging
+     */
+    get adapterName() {
+        return 'vercel';
+    }
+    // ===========================================================================
+    // Lifecycle
+    // ===========================================================================
+    /**
+     * Initialize the adapter
+     *
+     * Fetches all available tools from the MCP server and converts
+     * them to Vercel AI SDK format.
+     */
+    async initialize() {
+        if (this.initialized) {
+            return;
+        }
+        this.log('info', 'Initializing Vercel AI SDK adapter');
+        try {
+            const mcpTools = await this.server.listTools();
+            for (const tool of mcpTools) {
+                const vercelTool = this.convertTool(tool);
+                this.toolCache.set(tool.name, vercelTool);
+                this.mcpToolNames.set(tool.name, tool.name);
+                this.log('debug', `Registered tool: ${tool.name}`);
+            }
+            this.initialized = true;
+            this.log('info', `Vercel adapter initialized with ${this.toolCache.size} tools`);
+        }
+        catch (error) {
+            this.log('error', 'Failed to initialize Vercel adapter', { error });
+            throw McpError.fromError(error instanceof Error ? error : new Error(String(error)));
+        }
+    }
+    /**
+     * Cleanup adapter resources
+     */
+    async cleanup() {
+        this.toolCache.clear();
+        this.mcpToolNames.clear();
+        this.initialized = false;
+        this.log('info', 'Vercel adapter cleaned up');
+    }
+    // ===========================================================================
+    // Tool Conversion
+    // ===========================================================================
+    /**
+     * Convert MCP tool definition to Vercel CoreTool format
+     */
+    convertTool(mcpTool) {
+        const parameters = this.convertToZodSchema(mcpTool.inputSchema);
+        const toolName = mcpTool.name;
+        return {
+            description: mcpTool.description,
+            parameters,
+            execute: async (params) => {
+                const response = await this.callTool(toolName, params);
+                return this.formatResult(response);
+            },
+        };
+    }
+    /**
+     * Convert JSON Schema to Zod schema
+     *
+     * Creates a Zod schema that validates the same structure as the
+     * original JSON Schema from the MCP tool definition.
+     */
+    convertToZodSchema(schema) {
+        const shape = {};
+        if (schema.properties) {
+            for (const [key, prop] of Object.entries(schema.properties)) {
+                shape[key] = this.convertPropertyToZod(prop, schema.required?.includes(key) ?? false);
+            }
+        }
+        return z.object(shape);
+    }
+    /**
+     * Convert a single JSON Schema property to Zod
+     */
+    convertPropertyToZod(prop, required) {
+        let zodType;
+        switch (prop.type) {
+            case 'string':
+                zodType = z.string();
+                if (prop.description) {
+                    zodType = zodType.describe(prop.description);
+                }
+                if ('enum' in prop && prop.enum) {
+                    zodType = z.enum(prop.enum);
+                }
+                break;
+            case 'number':
+            case 'integer':
+                zodType = z.number();
+                if (prop.description) {
+                    zodType = zodType.describe(prop.description);
+                }
+                break;
+            case 'boolean':
+                zodType = z.boolean();
+                if (prop.description) {
+                    zodType = zodType.describe(prop.description);
+                }
+                break;
+            case 'array':
+                if ('items' in prop && prop.items) {
+                    zodType = z.array(this.convertPropertyToZod(prop.items, true));
+                }
+                else {
+                    zodType = z.array(z.unknown());
+                }
+                if (prop.description) {
+                    zodType = zodType.describe(prop.description);
+                }
+                break;
+            case 'object':
+                if ('properties' in prop && prop.properties) {
+                    const nestedShape = {};
+                    for (const [key, value] of Object.entries(prop.properties)) {
+                        const isRequired = 'required' in prop && prop.required?.includes(key);
+                        nestedShape[key] = this.convertPropertyToZod(value, isRequired ?? false);
+                    }
+                    zodType = z.object(nestedShape);
+                }
+                else {
+                    zodType = z.record(z.string(), z.unknown());
+                }
+                if (prop.description) {
+                    zodType = zodType.describe(prop.description);
+                }
+                break;
+            default:
+                zodType = z.unknown();
+        }
+        return required ? zodType : zodType.optional();
+    }
+    /**
+     * Default result formatter
+     */
+    defaultResultFormatter(response) {
+        if (response.isError) {
+            return JSON.stringify({ error: this.extractTextContent(response) });
+        }
+        return this.extractTextContent(response);
+    }
+    /**
+     * Extract text content from MCP response
+     */
+    extractTextContent(response) {
+        return response.content
+            .filter((c) => c.type === 'text')
+            .map((c) => c.text)
+            .join('\n');
+    }
+    // ===========================================================================
+    // Public API
+    // ===========================================================================
+    /**
+     * Get all tools in Vercel AI SDK format
+     *
+     * Returns a record of tool definitions suitable for use with
+     * the Vercel AI SDK's `generateText` or `streamText` functions.
+     *
+     * @returns Record of tool name to CoreTool
+     *
+     * @example
+     * ```typescript
+     * import { generateText } from 'ai'
+     *
+     * const tools = await adapter.getTools()
+     *
+     * const result = await generateText({
+     *   model: openai('gpt-4'),
+     *   tools,
+     *   prompt: 'Search for users',
+     * })
+     * ```
+     */
+    async getTools() {
+        if (!this.initialized) {
+            await this.initialize();
+        }
+        return Object.fromEntries(this.toolCache);
+    }
+    /**
+     * Get a single tool by name
+     *
+     * @param name - Tool name
+     * @returns CoreTool or undefined
+     */
+    async getTool(name) {
+        if (!this.initialized) {
+            await this.initialize();
+        }
+        return this.toolCache.get(name);
+    }
+    /**
+     * Handle a tool call from model response
+     *
+     * Executes the requested tool and returns a tool result.
+     *
+     * @param toolCall - Tool call from model
+     * @returns Tool result
+     *
+     * @example
+     * ```typescript
+     * for (const toolCall of result.toolCalls) {
+     *   const toolResult = await adapter.handleToolCall(toolCall)
+     *   results.push(toolResult)
+     * }
+     * ```
+     */
+    async handleToolCall(toolCall) {
+        const startTime = Date.now();
+        this.log('debug', `Handling tool call: ${toolCall.toolName}`, {
+            id: toolCall.toolCallId,
+        });
+        try {
+            const response = await this.callTool(toolCall.toolName, toolCall.args);
+            return {
+                toolCallId: toolCall.toolCallId,
+                toolName: toolCall.toolName,
+                result: this.formatResult(response),
+            };
+        }
+        catch (error) {
+            const mcpError = McpError.fromError(error instanceof Error ? error : new Error(String(error)));
+            this.log('error', `Tool call failed: ${toolCall.toolName}`, {
+                id: toolCall.toolCallId,
+                error: mcpError.message,
+                duration: Date.now() - startTime,
+            });
+            return {
+                toolCallId: toolCall.toolCallId,
+                toolName: toolCall.toolName,
+                result: JSON.stringify({
+                    error: mcpError.message,
+                    code: mcpError.code,
+                }),
+            };
+        }
+    }
+    /**
+     * Handle multiple tool calls in parallel
+     *
+     * @param toolCalls - Array of tool calls
+     * @param concurrency - Maximum concurrent executions (default: 5)
+     * @returns Array of tool results
+     */
+    async handleToolCalls(toolCalls, concurrency = 5) {
+        const results = [];
+        const executing = [];
+        for (const toolCall of toolCalls) {
+            const promise = this.handleToolCall(toolCall).then((result) => {
+                results.push(result);
+            });
+            executing.push(promise);
+            if (executing.length >= concurrency) {
+                await Promise.race(executing);
+            }
+        }
+        await Promise.all(executing);
+        return results;
+    }
+    /**
+     * Create a streaming tool executor
+     *
+     * Returns a function that executes a tool and streams the response.
+     * Compatible with Vercel AI SDK's streaming patterns.
+     *
+     * @param toolName - Name of the tool
+     * @returns Streaming executor function
+     *
+     * @example
+     * ```typescript
+     * const searchStream = adapter.createStreamingExecutor('search')
+     *
+     * const stream = searchStream({ query: 'AI documents' })
+     *
+     * for await (const chunk of stream) {
+     *   process.stdout.write(chunk)
+     * }
+     * ```
+     */
+    createStreamingExecutor(toolName) {
+        return (args, options) => {
+            return this.executeWithStreaming(toolName, args, options);
+        };
+    }
+    /**
+     * Execute a tool with streaming response
+     *
+     * @param toolName - Tool name
+     * @param args - Tool arguments
+     * @param options - Stream options
+     * @yields String chunks of the response
+     */
+    async *executeWithStreaming(toolName, args, options) {
+        const response = await this.callTool(toolName, args);
+        const content = this.extractTextContent(response);
+        const shouldStream = options?.stream ?? this.defaultStreaming;
+        const chunkSize = options?.chunkSize ?? 1024;
+        if (!shouldStream || content.length <= this.streamingThreshold) {
+            yield content;
+            return;
+        }
+        for await (const chunk of streamTextContent(content, {
+            chunkSize,
+            ...(options?.signal && { signal: options.signal }),
+        })) {
+            yield chunk.content;
+        }
+    }
+    /**
+     * Create a ReadableStream for a tool execution
+     *
+     * Returns a Web Streams API ReadableStream that can be used
+     * with Response objects for HTTP streaming.
+     *
+     * @param toolName - Tool name
+     * @param args - Tool arguments
+     * @returns ReadableStream of response chunks
+     *
+     * @example
+     * ```typescript
+     * // In a Next.js API route
+     * export async function POST(req: Request) {
+     *   const { toolName, args } = await req.json()
+     *   const stream = adapter.createReadableStream(toolName, args)
+     *   return new Response(stream)
+     * }
+     * ```
+     */
+    createReadableStream(toolName, args) {
+        const encoder = new TextEncoder();
+        const adapter = this;
+        return new ReadableStream({
+            async start(controller) {
+                try {
+                    for await (const chunk of adapter.executeWithStreaming(toolName, args)) {
+                        controller.enqueue(encoder.encode(chunk));
+                    }
+                    controller.close();
+                }
+                catch (error) {
+                    controller.error(error);
+                }
+            },
+        });
+    }
+    /**
+     * Create a tool wrapper compatible with Vercel AI SDK's `tool()` helper
+     *
+     * This creates a tool definition that can be used directly with the
+     * `tool()` helper from the `ai` package.
+     *
+     * @param mcpToolName - MCP tool name
+     * @returns Tool configuration object
+     *
+     * @example
+     * ```typescript
+     * import { tool } from 'ai'
+     *
+     * const searchTool = adapter.createToolWrapper('search')
+     *
+     * // Use with generateText
+     * const result = await generateText({
+     *   model: openai('gpt-4'),
+     *   tools: { search: searchTool },
+     *   prompt: 'Find documents',
+     * })
+     * ```
+     */
+    createToolWrapper(mcpToolName) {
+        return this.toolCache.get(mcpToolName);
+    }
+}
+// =============================================================================
+// Factory Function
+// =============================================================================
+/**
+ * Create a Vercel AI SDK adapter for an MCP server
+ *
+ * @param config - Adapter configuration
+ * @returns Configured VercelAdapter instance
+ *
+ * @example
+ * ```typescript
+ * const adapter = createVercelAdapter({
+ *   server: mcpServer,
+ *   retry: { maxRetries: 3 },
+ *   streamingThreshold: 10000,
+ * })
+ *
+ * await adapter.initialize()
+ * const tools = await adapter.getTools()
+ * ```
+ */
+export function createVercelAdapter(config) {
+    return new VercelAdapter(config);
+}
+// =============================================================================
+// Utility Functions
+// =============================================================================
+/**
+ * Convert Vercel tool results to continuation messages
+ *
+ * Formats tool results for inclusion in the next model request.
+ *
+ * @param results - Array of tool results
+ * @returns Formatted tool results for API
+ */
+export function formatToolResults(results) {
+    return results.map((result) => ({
+        type: 'tool-result',
+        toolCallId: result.toolCallId,
+        toolName: result.toolName,
+        result: result.result,
+    }));
+}
+//# sourceMappingURL=vercel-adapter.js.map

package/dist/mcp/adapters/vercel-adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vercel-adapter.js","sourceRoot":"","sources":["../../../src/mcp/adapters/vercel-adapter.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,EAAE,WAAW,EAA0B,MAAM,gBAAgB,CAAA;AAEpE,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAA;AACnC,OAAO,EACL,iBAAiB,GAClB,MAAM,aAAa,CAAA;AACpB,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAqFvB,gFAAgF;AAChF,YAAY;AACZ,gFAAgF;AAEhF,MAAM,2BAA2B,GAAG,EAAE,GAAG,IAAI,CAAA,CAAC,OAAO;AAErD,gFAAgF;AAChF,uBAAuB;AACvB,gFAAgF;AAEhF;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAM,OAAO,aAAc,SAAQ,WAAW;IAC3B,kBAAkB,CAAQ;IAC1B,gBAAgB,CAAS;IACzB,YAAY,CAAuC;IAC5D,SAAS,GAAgC,IAAI,GAAG,EAAE,CAAA;IAClD,YAAY,GAAwB,IAAI,GAAG,EAAE,CAAA,CAAC,0BAA0B;IACxE,WAAW,GAAG,KAAK,CAAA;IAE3B,YAAY,MAA2B;QACrC,KAAK,CAAC,MAAM,CAAC,CAAA;QACb,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,kBAAkB,IAAI,2BAA2B,CAAA;QAClF,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAA;QACvD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACnF,CAAC;IAED;;OAEG;IACH,IAAI,WAAW;QACb,OAAO,QAAQ,CAAA;IACjB,CAAC;IAED,8EAA8E;IAC9E,YAAY;IACZ,8EAA8E;IAE9E;;;;;OAKG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAM;QACR,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,oCAAoC,CAAC,CAAA;QAEtD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAA;YAE9C,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;gBAC5B,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAA;gBACzC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,CAAA;gBACzC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAA;gBAC3C,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAA;YACpD,CAAC;YAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAA;YACvB,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,mCAAmC,IAAI,CAAC,SAAS,CAAC,IAAI,QAAQ,CAAC,CAAA;QAClF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,qCAAqC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAA;YACnE,MAAM,QAAQ,CAAC,SAAS,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;QACrF,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAA;QACtB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;QACzB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAA;QACxB,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,2BAA2B,CAAC,CAAA;IAC/C,CAAC;IAED,8EAA8E;IAC9E,kBAAkB;IAClB,8EAA8E;IAE9E;;OAEG;IACK,WAAW,CAAC,OAA0B;QAC5C,MAAM,UAAU,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,WAAW,CAAC,CAAA;QAC/D,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAA;QAE7B,OAAO;YACL,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,UAAU;YACV,OAAO,EAAE,KAAK,EAAE,MAA+B,EAAE,EAAE;gBACjD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;gBACtD,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAA;YACpC,CAAC;SACF,CAAA;IACH,CAAC;IAED;;;;;OAKG;IACK,kBAAkB,CAAC,MAAkB;QAC3C,MAAM,KAAK,GAAiC,EAAE,CAAA;QAE9C,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YACtB,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC5D,KAAK,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,CAAA;YACvF,CAAC;QACH,CAAC;QAED,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IACxB,CAAC;IAED;;OAEG;IACK,oBAAoB,CAC1B,IAAyD,EACzD,QAAiB;QAEjB,IAAI,OAAqB,CAAA;QAEzB,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,QAAQ;gBACX,OAAO,GAAG,CAAC,CAAC,MAAM,EAAE,CAAA;gBACpB,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;oBACrB,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;gBAC9C,CAAC;gBACD,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBAChC,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAA6B,CAAC,CAAA;gBACtD,CAAC;gBACD,MAAK;YAEP,KAAK,QAAQ,CAAC;YACd,KAAK,SAAS;gBACZ,OAAO,GAAG,CAAC,CAAC,MAAM,EAAE,CAAA;gBACpB,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;oBACrB,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;gBAC9C,CAAC;gBACD,MAAK;YAEP,KAAK,SAAS;gBACZ,OAAO,GAAG,CAAC,CAAC,OAAO,EAAE,CAAA;gBACrB,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;oBACrB,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;gBAC9C,CAAC;gBACD,MAAK;YAEP,KAAK,OAAO;gBACV,IAAI,OAAO,IAAI,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBAClC,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAmB,EAAE,IAAI,CAAC,CAAC,CAAA;gBAC9E,CAAC;qBAAM,CAAC;oBACN,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAA;gBAChC,CAAC;gBACD,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;oBACrB,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;gBAC9C,CAAC;gBACD,MAAK;YAEP,KAAK,QAAQ;gBACX,IAAI,YAAY,IAAI,IAAI,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBAC5C,MAAM,WAAW,GAAiC,EAAE,CAAA;oBACpD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;wBAC3D,MAAM,UAAU,GAAG,UAAU,IAAI,IAAI,IAAK,IAAI,CAAC,QAAqB,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAA;wBACnF,WAAW,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,UAAU,IAAI,KAAK,CAAC,CAAA;oBAC1E,CAAC;oBACD,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAA;gBACjC,CAAC;qBAAM,CAAC;oBACN,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAA;gBAC7C,CAAC;gBACD,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;oBACrB,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;gBAC9C,CAAC;gBACD,MAAK;YAEP;gBACE,OAAO,GAAG,CAAC,CAAC,OAAO,EAAE,CAAA;QACzB,CAAC;QAED,OAAO,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAA;IAChD,CAAC;IAED;;OAEG;IACK,sBAAsB,CAAC,QAAyB;QACtD,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrB,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAA;QACrE,CAAC;QACD,OAAO,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAA;IAC1C,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,QAAyB;QAClD,OAAO,QAAQ,CAAC,OAAO;aACpB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;aAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aAClB,IAAI,CAAC,IAAI,CAAC,CAAA;IACf,CAAC;IAED,8EAA8E;IAC9E,aAAa;IACb,8EAA8E;IAE9E;;;;;;;;;;;;;;;;;;;;OAoBG;IACH,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAA;QACzB,CAAC;QAED,OAAO,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;IAC3C,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,OAAO,CAAC,IAAY;QACxB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAA;QACzB,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;IACjC,CAAC;IAED;;;;;;;;;;;;;;;OAeG;IACH,KAAK,CAAC,cAAc,CAAC,QAAwB;QAC3C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAE5B,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,uBAAuB,QAAQ,CAAC,QAAQ,EAAE,EAAE;YAC5D,EAAE,EAAE,QAAQ,CAAC,UAAU;SACxB,CAAC,CAAA;QAEF,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAA;YAEtE,OAAO;gBACL,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,QAAQ,EAAE,QAAQ,CAAC,QAAQ;gBAC3B,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC;aACpC,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,QAAQ,CAAC,SAAS,CACjC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAC1D,CAAA;YAED,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,qBAAqB,QAAQ,CAAC,QAAQ,EAAE,EAAE;gBAC1D,EAAE,EAAE,QAAQ,CAAC,UAAU;gBACvB,KAAK,EAAE,QAAQ,CAAC,OAAO;gBACvB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACjC,CAAC,CAAA;YAEF,OAAO;gBACL,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,QAAQ,EAAE,QAAQ,CAAC,QAAQ;gBAC3B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC;oBACrB,KAAK,EAAE,QAAQ,CAAC,OAAO;oBACvB,IAAI,EAAE,QAAQ,CAAC,IAAI;iBACpB,CAAC;aACH,CAAA;QACH,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,eAAe,CACnB,SAA2B,EAC3B,WAAW,GAAG,CAAC;QAEf,MAAM,OAAO,GAAuB,EAAE,CAAA;QACtC,MAAM,SAAS,GAAoB,EAAE,CAAA;QAErC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;gBAC5D,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YACtB,CAAC,CAAC,CAAA;YAEF,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAEvB,IAAI,SAAS,CAAC,MAAM,IAAI,WAAW,EAAE,CAAC;gBACpC,MAAM,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;YAC/B,CAAC;QACH,CAAC;QAED,MAAM,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;QAC5B,OAAO,OAAO,CAAA;IAChB,CAAC;IAED;;;;;;;;;;;;;;;;;;;OAmBG;IACH,uBAAuB,CACrB,QAAgB;QAEhB,OAAO,CAAC,IAA6B,EAAE,OAA6B,EAAE,EAAE;YACtE,OAAO,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,IAAI,EAAE,OAAO,CAAC,CAAA;QAC3D,CAAC,CAAA;IACH,CAAC;IAED;;;;;;;OAOG;IACK,KAAK,CAAC,CAAC,oBAAoB,CACjC,QAAgB,EAChB,IAA6B,EAC7B,OAA6B;QAE7B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;QACpD,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAA;QAEjD,MAAM,YAAY,GAAG,OAAO,EAAE,MAAM,IAAI,IAAI,CAAC,gBAAgB,CAAA;QAC7D,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,IAAI,CAAA;QAE5C,IAAI,CAAC,YAAY,IAAI,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC/D,MAAM,OAAO,CAAA;YACb,OAAM;QACR,CAAC;QAED,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,iBAAiB,CAAC,OAAO,EAAE;YACnD,SAAS;YACT,GAAG,CAAC,OAAO,EAAE,MAAM,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;SACnD,CAAC,EAAE,CAAC;YACH,MAAM,KAAK,CAAC,OAAO,CAAA;QACrB,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;;;;;;;;OAmBG;IACH,oBAAoB,CAClB,QAAgB,EAChB,IAA6B;QAE7B,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;QACjC,MAAM,OAAO,GAAG,IAAI,CAAA;QAEpB,OAAO,IAAI,cAAc,CAAC;YACxB,KAAK,CAAC,KAAK,CAAC,UAAU;gBACpB,IAAI,CAAC;oBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,OAAO,CAAC,oBAAoB,CAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,CAAC;wBACvE,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAA;oBAC3C,CAAC;oBACD,UAAU,CAAC,KAAK,EAAE,CAAA;gBACpB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACzB,CAAC;YACH,CAAC;SACF,CAAC,CAAA;IACJ,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,iBAAiB,CAAC,WAAmB;QACnC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC,CAAA;IACxC,CAAC;CACF;AAED,gFAAgF;AAChF,mBAAmB;AACnB,gFAAgF;AAEhF;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,UAAU,mBAAmB,CAAC,MAA2B;IAC7D,OAAO,IAAI,aAAa,CAAC,MAAM,CAAC,CAAA;AAClC,CAAC;AAED,gFAAgF;AAChF,oBAAoB;AACpB,gFAAgF;AAEhF;;;;;;;GAOG;AACH,MAAM,UAAU,iBAAiB,CAAC,OAA2B;IAM3D,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAC9B,IAAI,EAAE,aAAsB;QAC5B,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC,CAAC,CAAA;AACL,CAAC"}

package/dist/mcp/agent.d.ts

@@ -0,0 +1,192 @@
+/**
+ * MonDoAgent - Main Entry Point for AI Agents
+ *
+ * The MonDoAgent orchestrates:
+ * - MCP Server for tool handling (search, fetch, do)
+ * - AgentFS for file system operations
+ * - KV Store for key-value storage
+ * - Tool Call Auditing for logging
+ * - Code execution via Worker Loader or Miniflare fallback
+ */
+import { type CodeLoader, type McpServerConfig } from './server';
+import { type AgentFSDatabase } from '../agentfs/vfs';
+import { type KVStorageBackend } from '../agentfs/kv-store';
+import { type AuditBackend } from '../agentfs/toolcalls';
+import type { DatabaseAccess, McpToolResponse } from './types';
+/**
+ * Execution result from tool or code execution
+ */
+export interface ExecutionResult<T = unknown> {
+    success: boolean;
+    value?: T;
+    error?: string;
+    logs?: string[];
+    duration?: number;
+}
+/**
+ * Search options for the search method
+ */
+export interface SearchOptions {
+    collection?: string;
+    database?: string;
+    limit?: number;
+}
+/**
+ * Search result in OpenAI Deep Research format
+ */
+export interface SearchResult {
+    id: string;
+    title: string;
+    url: string;
+    text: string;
+}
+/**
+ * Fetch result with document metadata
+ */
+export interface FetchResult {
+    id: string;
+    title: string;
+    url: string;
+    text: string;
+    metadata: {
+        database: string;
+        collection: string;
+        _id: string;
+    };
+}
+/**
+ * MCP server interface exposed by the agent
+ */
+export interface McpServerLike {
+    listTools(): Promise<{
+        name: string;
+        description: string;
+    }[]>;
+    callTool(name: string, args: Record<string, unknown>): Promise<McpToolResponse>;
+}
+/**
+ * File system interface exposed by the agent
+ */
+export interface FileSystemLike {
+    readFile(path: string): Promise<string>;
+    writeFile(path: string, content: string): Promise<void>;
+    deleteFile(path: string): Promise<boolean>;
+    exists(path: string): Promise<boolean>;
+    stat(path: string): Promise<{
+        type: string;
+        size: number;
+    } | null>;
+    readdir(path: string): Promise<string[]>;
+    mkdir(path: string): Promise<void>;
+    glob(pattern: string): Promise<string[]>;
+}
+/**
+ * Key-Value store interface exposed by the agent
+ */
+export interface KVStoreLike {
+    get<T>(key: string): Promise<T | null>;
+    set<T>(key: string, value: T): Promise<void>;
+    delete(key: string): Promise<boolean>;
+    has(key: string): Promise<boolean>;
+    keys(prefix?: string): Promise<string[]>;
+}
+/**
+ * Audit log interface exposed by the agent
+ */
+export interface AuditLogLike {
+    record(tool: string, inputs: unknown, outputs: unknown, durationMs?: number): Promise<string>;
+    list(options?: {
+        limit?: number;
+        offset?: number;
+        tool?: string;
+    }): Promise<{
+        id: string;
+        tool: string;
+        timestamp: Date;
+    }[]>;
+    count(): Promise<number>;
+}
+/**
+ * Configuration for creating a MonDoAgent
+ */
+export interface MonDoAgentConfig {
+    /** Unique agent identifier */
+    id?: string;
+    /** Database access for MCP tools */
+    dbAccess: DatabaseAccess;
+    /** Optional code loader for 'do' tool */
+    codeLoader?: CodeLoader;
+    /** Optional custom filesystem database */
+    fsDatabase?: AgentFSDatabase;
+    /** Optional custom KV storage backend */
+    kvBackend?: KVStorageBackend;
+    /** Optional custom audit storage backend */
+    auditBackend?: AuditBackend;
+}
+/**
+ * MonDoAgent - AI Agent interface for mongo.do
+ */
+export interface MonDoAgent {
+    /** Agent identifier */
+    readonly id: string;
+    /** Access to MCP server for tool handling */
+    readonly mcp: McpServerLike;
+    /** Access to virtual filesystem */
+    readonly fs: FileSystemLike;
+    /** Access to key-value store */
+    readonly kv: KVStoreLike;
+    /** Access to tool call audit log */
+    readonly audit: AuditLogLike;
+    /**
+     * Execute a tool by name with given arguments
+     */
+    execute<T = unknown>(toolName: string, args: Record<string, unknown>): Promise<ExecutionResult<T>>;
+    /**
+     * Execute code in the secure sandbox
+     */
+    do(code: string, description?: string): Promise<ExecutionResult<unknown>>;
+    /**
+     * Search for documents
+     */
+    search(query: string, options?: SearchOptions): Promise<SearchResult[]>;
+    /**
+     * Fetch a document by ID
+     */
+    fetch(id: string): Promise<FetchResult | null>;
+    /**
+     * Dispose agent resources
+     */
+    dispose(): Promise<void>;
+}
+/**
+ * Create a new MonDoAgent instance
+ *
+ * @param config - Agent configuration
+ * @returns MonDoAgent instance
+ *
+ * @example
+ * ```typescript
+ * const agent = createMonDoAgent({
+ *   dbAccess: myDatabaseAccess,
+ *   codeLoader: myCodeLoader,
+ * });
+ *
+ * // Execute code
+ * const result = await agent.do('return await db.collection("users").count()');
+ *
+ * // Search documents
+ * const results = await agent.search('users with admin role');
+ *
+ * // Use filesystem
+ * await agent.fs.writeFile('/data/output.json', JSON.stringify(data));
+ *
+ * // Use KV store
+ * await agent.kv.set('session-state', { step: 1 });
+ *
+ * // Clean up
+ * await agent.dispose();
+ * ```
+ */
+export declare function createMonDoAgent(config: MonDoAgentConfig): MonDoAgent;
+export type { DatabaseAccess, CodeLoader, McpServerConfig };
+//# sourceMappingURL=agent.d.ts.map