@cloudbase/agent-tools 0.0.2

package/dist/mcp/agkit-mcp-server.d.ts

@@ -0,0 +1,291 @@
+/**
+ * AG-Kit MCP Server
+ *
+ * A unified MCP server that directly exposes AG-Kit tools via the MCP protocol.
+ * For HTTP transports, provides transport instances for integration with any HTTP server.
+ */
+import { z } from "zod/v4";
+import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
+import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+import { BaseTool } from "../utils";
+import { MCPToolMetadata, MCPEventListener } from "./types";
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+export interface Transport {
+    start?(): Promise<void>;
+    close(): Promise<void>;
+    onMessage?(handler: (message: any) => void): void;
+    send?(message: any): Promise<void>;
+}
+export interface AGKitMCPServerConfig {
+    name: string;
+    version: string;
+    description?: string;
+    enableLogging?: boolean;
+    includeMetadata?: boolean;
+    errorHandling?: "throw" | "return_error";
+    transformInput?: (input: any) => any;
+    transformOutput?: (output: any) => any;
+}
+interface MCPToolConfig {
+    name?: string;
+    description?: string;
+    namePrefix?: string;
+}
+export type MCPTransportType = "stdio" | "memory" | "sse" | "streamableHttp";
+interface HTTPResponse {
+    writeHead(statusCode: number, headers?: Record<string, string>): void;
+    write(chunk: string): void;
+    end(data?: string): void;
+    flushHeaders?(): void;
+    on?(event: string, listener: (...args: any[]) => void): void;
+}
+/**
+ * Factory function for creating SSE transport instances
+ */
+export interface SSETransportFactory {
+    (endpoint: string, response: HTTPResponse, options?: {
+        enableDnsRebindingProtection?: boolean;
+        allowedHosts?: string[];
+        allowedOrigins?: string[];
+    }): Promise<SSEServerTransport>;
+}
+/**
+ * Factory function for creating StreamableHTTP transport instances
+ */
+export interface StreamableHTTPTransportFactory {
+    (options?: {
+        enableJsonResponse?: boolean;
+        enableDnsRebindingProtection?: boolean;
+        allowedHosts?: string[];
+        allowedOrigins?: string[];
+        sessionIdGenerator?: () => string;
+    }): Promise<StreamableHTTPServerTransport>;
+}
+/**
+ * Callback function for setting up SSE transport with user's HTTP server
+ */
+export interface SSETransportSetup {
+    (server: AGKitMCPServer, createTransport: SSETransportFactory): Promise<void> | void;
+}
+/**
+ * Callback function for setting up StreamableHTTP transport with user's HTTP server
+ */
+export interface StreamableHTTPTransportSetup {
+    (server: AGKitMCPServer, createTransport: StreamableHTTPTransportFactory): Promise<void> | void;
+}
+export interface MCPTransportConfig {
+    type: MCPTransportType;
+    memoryId?: string;
+    sseSetup?: SSETransportSetup;
+    streamableHttpSetup?: StreamableHTTPTransportSetup;
+}
+export declare class MemoryTransportRegistry {
+    private static instance;
+    private transports;
+    private servers;
+    static getInstance(): MemoryTransportRegistry;
+    registerServer(memoryId: string, server: AGKitMCPServer): void;
+    unregisterServer(memoryId: string): void;
+    getServer(memoryId: string): AGKitMCPServer | undefined;
+    /**
+     * Creates a pair of InMemory transports (server and client)
+     * Uses official InMemoryTransport.createLinkedPair()
+     */
+    createTransportPair(memoryId: string): {
+        server: any;
+        client: any;
+    };
+    /**
+     * Gets the client transport for client connections
+     */
+    getClientTransport(memoryId: string): any;
+    /**
+     * Gets the server-side transport
+     */
+    getServerTransport(memoryId: string): any;
+    callTool(memoryId: string, toolName: string, args: Record<string, any>): Promise<CallToolResult>;
+    listTools(memoryId: string): Promise<Array<{
+        name: string;
+        description: string;
+        inputSchema: any;
+    }>>;
+}
+/**
+ * Converts a zod schema to MCP JSON schema format
+ */
+export declare function zodSchemaToMCPSchema(zodSchema: z.ZodSchema<any>): any;
+/**
+ * Converts AG-Kit BaseTool to MCP tool metadata
+ */
+export declare function convertAGKitToolToMCPMetadata(tool: BaseTool, config?: AGKitMCPServerConfig, toolConfig?: MCPToolConfig): MCPToolMetadata;
+/**
+ * AG-Kit MCP Server
+ *
+ * A unified server that directly handles AG-Kit tools and MCP protocol
+ */
+export declare class AGKitMCPServer {
+    private server;
+    private config;
+    private tools;
+    private toolConfigs;
+    private eventListeners;
+    private isRunning;
+    private currentTransport;
+    private transportType;
+    private memoryId?;
+    constructor(config?: AGKitMCPServerConfig);
+    /**
+     * Setup MCP server request handlers
+     */
+    private setupServerHandlers;
+    /**
+     * Register an AG-Kit tool with the server
+     * @param tool - The AG-Kit tool instance to register
+     * @param toolConfig - Optional configuration for the tool (name, description, etc.)
+     * @returns The MCP metadata for the registered tool
+     */
+    registerTool(tool: BaseTool, toolConfig?: MCPToolConfig): MCPToolMetadata;
+    /**
+     * Register multiple AG-Kit tools
+     */
+    registerTools(tools: BaseTool[], toolConfig?: MCPToolConfig): MCPToolMetadata[];
+    /**
+     * Unregister a tool
+     */
+    unregisterTool(name: string): boolean;
+    /**
+     * Execute a tool via MCP call format
+     */
+    callTool(name: string, args: Record<string, any>): Promise<CallToolResult>;
+    /**
+     * Get all registered tool metadata
+     */
+    getToolsMetadata(): MCPToolMetadata[];
+    /**
+     * List all registered tools in MCP format
+     */
+    listTools(): Array<{
+        name: string;
+        description: string;
+        inputSchema: any;
+    }>;
+    /**
+     * Get specific tool metadata
+     */
+    getToolMetadata(name: string): MCPToolMetadata | null;
+    /**
+     * Check if a tool is registered
+     */
+    hasTool(name: string): boolean;
+    /**
+     * Get list of available tool names
+     */
+    getAvailableTools(): string[];
+    /**
+     * Get tool statistics
+     */
+    getStats(): {
+        totalTools: number;
+        toolNames: string[];
+        toolTypes: Record<string, number>;
+        isRunning: boolean;
+        serverInfo: {
+            name: string;
+            version: string;
+            description: string;
+        };
+    };
+    /**
+     * Start the MCP server with specified transport
+     */
+    run(transportConfig: MCPTransportConfig): Promise<void>;
+    /**
+     * Start the MCP server with stdio transport (backward compatibility)
+     */
+    runStdio(): Promise<void>;
+    /**
+     * Start the MCP server with memory transport
+     */
+    runMemory(memoryId?: string): Promise<void>;
+    /**
+     * Run with StreamableHTTP transport using user-provided setup callback
+     * @param setup - Callback function to integrate with user's HTTP server
+     * @param options - Optional transport configuration (deprecated - use factory function parameters)
+     */
+    runStreamableHttp(setup: StreamableHTTPTransportSetup, options?: {
+        enableJsonResponse?: boolean;
+        enableDnsRebindingProtection?: boolean;
+        allowedHosts?: string[];
+        allowedOrigins?: string[];
+        sessionIdGenerator?: () => string;
+    }): Promise<void>;
+    /**
+     * Run with SSE transport using user-provided setup callback
+     * @param setup - Callback function to integrate with user's HTTP server
+     * @param options - Optional transport configuration (deprecated - use factory function parameters)
+     */
+    runSSE(setup: SSETransportSetup, options?: {
+        endpoint?: string;
+        enableDnsRebindingProtection?: boolean;
+        allowedHosts?: string[];
+        allowedOrigins?: string[];
+    }): Promise<void>;
+    /**
+     * Stop the server
+     */
+    stop(): Promise<void>;
+    /**
+     * Check if server is running
+     */
+    isServerRunning(): boolean;
+    /**
+     * Get current transport type
+     */
+    getTransportType(): MCPTransportType | null;
+    /**
+     * Get memory ID (for memory transport)
+     */
+    getMemoryId(): string | undefined;
+    /**
+     * Get server connection info
+     */
+    getConnectionInfo(): {
+        isRunning: boolean;
+        transportType: MCPTransportType | null;
+        memoryId?: string;
+    };
+    /**
+     * Add event listener
+     */
+    addEventListener(listener: MCPEventListener): void;
+    /**
+     * Remove event listener
+     */
+    removeEventListener(listener: MCPEventListener): boolean;
+    /**
+     * Emit event to all listeners
+     */
+    private emit;
+    /**
+     * Update server configuration
+     */
+    updateConfig(newConfig: Partial<AGKitMCPServerConfig>): void;
+    /**
+     * Get current configuration
+     */
+    getConfig(): AGKitMCPServerConfig;
+    /**
+     * Clear all registered tools
+     */
+    clearTools(): void;
+}
+/**
+ * Utility function to create and start an AG-Kit MCP server
+ */
+export declare function createAGKitMCPServer(config: AGKitMCPServerConfig, tools?: BaseTool[]): Promise<AGKitMCPServer>;
+/**
+ * Global memory transport registry instance
+ */
+export declare const memoryTransportRegistry: MemoryTransportRegistry;
+export {};
+//# sourceMappingURL=agkit-mcp-server.d.ts.map

package/dist/mcp/agkit-mcp-server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agkit-mcp-server.d.ts","sourceRoot":"","sources":["../../src/mcp/agkit-mcp-server.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAC;AAI3B,OAAO,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAC;AAC7E,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AAQnG,OAAO,EAAE,QAAQ,EAAc,MAAM,UAAU,CAAC;AAChD,OAAO,EAAE,eAAe,EAAY,gBAAgB,EAAE,MAAM,SAAS,CAAC;AACtE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AAGzE,MAAM,WAAW,SAAS;IACxB,KAAK,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACxB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,EAAE,GAAG,KAAK,IAAI,GAAG,IAAI,CAAC;IAClD,IAAI,CAAC,CAAC,OAAO,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,oBAAoB;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,OAAO,CAAC;IAExB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,aAAa,CAAC,EAAE,OAAO,GAAG,cAAc,CAAC;IACzC,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,GAAG,CAAC;IACrC,eAAe,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,GAAG,CAAC;CACxC;AAED,UAAU,aAAa;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAGD,MAAM,MAAM,gBAAgB,GAAG,OAAO,GAAG,QAAQ,GAAG,KAAK,GAAG,gBAAgB,CAAC;AAE7E,UAAU,YAAY;IACpB,SAAS,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC;IACtE,KAAK,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,YAAY,CAAC,IAAI,IAAI,CAAC;IACtB,EAAE,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,IAAI,GAAG,IAAI,CAAC;CAC9D;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,CACE,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,YAAY,EACtB,OAAO,CAAC,EAAE;QACR,4BAA4B,CAAC,EAAE,OAAO,CAAC;QACvC,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;KAC3B,GACA,OAAO,CAAC,kBAAkB,CAAC,CAAC;CAChC;AAED;;GAEG;AACH,MAAM,WAAW,8BAA8B;IAC7C,CAAC,OAAO,CAAC,EAAE;QACT,kBAAkB,CAAC,EAAE,OAAO,CAAC;QAC7B,4BAA4B,CAAC,EAAE,OAAO,CAAC;QACvC,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;QAC1B,kBAAkB,CAAC,EAAE,MAAM,MAAM,CAAC;KACnC,GAAG,OAAO,CAAC,6BAA6B,CAAC,CAAC;CAC5C;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,CACE,MAAM,EAAE,cAAc,EACtB,eAAe,EAAE,mBAAmB,GACnC,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,4BAA4B;IAC3C,CACE,MAAM,EAAE,cAAc,EACtB,eAAe,EAAE,8BAA8B,GAC9C,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CACzB;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,gBAAgB,CAAC;IAGvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAGlB,QAAQ,CAAC,EAAE,iBAAiB,CAAC;IAG7B,mBAAmB,CAAC,EAAE,4BAA4B,CAAC;CACpD;AAGD,qBAAa,uBAAuB;IAClC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAA0B;IACjD,OAAO,CAAC,UAAU,CAAmD;IACrE,OAAO,CAAC,OAAO,CAAqC;IAEpD,MAAM,CAAC,WAAW,IAAI,uBAAuB;IAO7C,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,GAAG,IAAI;IAI9D,gBAAgB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAKxC,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS;IAIvD;;;OAGG;IACH,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG;QAAE,MAAM,EAAE,GAAG,CAAC;QAAC,MAAM,EAAE,GAAG,CAAA;KAAE;IAenE;;OAEG;IACH,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,GAAG;IAQzC;;OAEG;IACH,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,GAAG;IASnC,QAAQ,CACZ,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GACxB,OAAO,CAAC,cAAc,CAAC;IAQpB,SAAS,CACb,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,GAAG,CAAA;KAAE,CAAC,CAAC;CAO3E;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,GAAG,CAgGrE;AAED;;GAEG;AACH,wBAAgB,6BAA6B,CAC3C,IAAI,EAAE,QAAQ,EACd,MAAM,GAAE,oBAGP,EACD,UAAU,CAAC,EAAE,aAAa,GACzB,eAAe,CA0BjB;AAED;;;;GAIG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,KAAK,CAA+B;IAC5C,OAAO,CAAC,WAAW,CAAoC;IACvD,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,gBAAgB,CAA0B;IAClD,OAAO,CAAC,aAAa,CAAiC;IACtD,OAAO,CAAC,QAAQ,CAAC,CAAS;gBAGxB,MAAM,GAAE,oBAGP;IAyBH;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAqG3B;;;;;OAKG;IACH,YAAY,CAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,CAAC,EAAE,aAAa,GAAG,eAAe;IAwBzE;;OAEG;IACH,aAAa,CACX,KAAK,EAAE,QAAQ,EAAE,EACjB,UAAU,CAAC,EAAE,aAAa,GACzB,eAAe,EAAE;IAYpB;;OAEG;IACH,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAWrC;;OAEG;IACG,QAAQ,CACZ,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GACxB,OAAO,CAAC,cAAc,CAAC;IAqG1B;;OAEG;IACH,gBAAgB,IAAI,eAAe,EAAE;IAOrC;;OAEG;IACH,SAAS,IAAI,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,GAAG,CAAA;KAAE,CAAC;IAwB3E;;OAEG;IACH,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,eAAe,GAAG,IAAI;IAQrD;;OAEG;IACH,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAI9B;;OAEG;IACH,iBAAiB,IAAI,MAAM,EAAE;IAI7B;;OAEG;IACH,QAAQ;;;;;;;;;;;IAuBR;;OAEG;IACG,GAAG,CAAC,eAAe,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAyK7D;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAI/B;;OAEG;IACG,SAAS,CAAC,QAAQ,GAAE,MAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAI5D;;;;OAIG;IACG,iBAAiB,CACrB,KAAK,EAAE,4BAA4B,EACnC,OAAO,CAAC,EAAE;QACR,kBAAkB,CAAC,EAAE,OAAO,CAAC;QAC7B,4BAA4B,CAAC,EAAE,OAAO,CAAC;QACvC,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;QAC1B,kBAAkB,CAAC,EAAE,MAAM,MAAM,CAAC;KACnC,GACA,OAAO,CAAC,IAAI,CAAC;IAOhB;;;;OAIG;IACG,MAAM,CACV,KAAK,EAAE,iBAAiB,EACxB,OAAO,CAAC,EAAE;QACR,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,4BAA4B,CAAC,EAAE,OAAO,CAAC;QACvC,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;KAC3B,GACA,OAAO,CAAC,IAAI,CAAC;IAOhB;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAuD3B;;OAEG;IACH,eAAe,IAAI,OAAO;IAI1B;;OAEG;IACH,gBAAgB,IAAI,gBAAgB,GAAG,IAAI;IAI3C;;OAEG;IACH,WAAW,IAAI,MAAM,GAAG,SAAS;IAIjC;;OAEG;IACH,iBAAiB,IAAI;QACnB,SAAS,EAAE,OAAO,CAAC;QACnB,aAAa,EAAE,gBAAgB,GAAG,IAAI,CAAC;QACvC,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB;IAQD;;OAEG;IACH,gBAAgB,CAAC,QAAQ,EAAE,gBAAgB,GAAG,IAAI;IAIlD;;OAEG;IACH,mBAAmB,CAAC,QAAQ,EAAE,gBAAgB,GAAG,OAAO;IAIxD;;OAEG;IACH,OAAO,CAAC,IAAI;IAUZ;;OAEG;IACH,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,oBAAoB,CAAC,GAAG,IAAI;IAI5D;;OAEG;IACH,SAAS,IAAI,oBAAoB;IAIjC;;OAEG;IACH,UAAU,IAAI,IAAI;CAQnB;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,oBAAoB,EAC5B,KAAK,GAAE,QAAQ,EAAO,GACrB,OAAO,CAAC,cAAc,CAAC,CAQzB;AAED;;GAEG;AACH,eAAO,MAAM,uBAAuB,yBAAwC,CAAC"}