@opensumi/ide-ai-native 3.7.1-next-1737628160.0 → 3.7.1-next-1737703128.0

package/lib/common/tool-invocation-registry.js

@@ -0,0 +1,68 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolInvocationRegistryImpl = exports.ToolProvider = exports.ToolInvocationRegistry = exports.ToolRequest = exports.ToolParameterSchema = void 0;
+const tslib_1 = require("tslib");
+const di_1 = require("@opensumi/di");
+const zod_1 = require("zod");
+exports.ToolParameterSchema = zod_1.z.object({
+    type: zod_1.z.enum(['string', 'number', 'boolean', 'object', 'array']),
+    description: zod_1.z.string().optional(),
+    enum: zod_1.z.array(zod_1.z.any()).optional(),
+    items: zod_1.z.lazy(() => exports.ToolParameterSchema).optional(),
+    properties: zod_1.z.record(zod_1.z.lazy(() => exports.ToolParameterSchema)).optional(),
+    required: zod_1.z.array(zod_1.z.string()).optional(),
+});
+var ToolRequest;
+(function (ToolRequest) {
+    function isToolParameter(obj) {
+        return exports.ToolParameterSchema.safeParse(obj).success;
+    }
+    ToolRequest.isToolParameter = isToolParameter;
+})(ToolRequest || (exports.ToolRequest = ToolRequest = {}));
+exports.ToolInvocationRegistry = Symbol('ToolInvocationRegistry');
+exports.ToolProvider = Symbol('ToolProvider');
+let ToolInvocationRegistryImpl = class ToolInvocationRegistryImpl {
+    constructor() {
+        this.tools = new Map();
+    }
+    unregisterAllTools(providerName) {
+        const toolsToRemove = [];
+        for (const [id, tool] of this.tools.entries()) {
+            if (tool.providerName === providerName) {
+                toolsToRemove.push(id);
+            }
+        }
+        toolsToRemove.forEach(id => this.tools.delete(id));
+    }
+    getAllFunctions() {
+        return Array.from(this.tools.values());
+    }
+    registerTool(tool) {
+        if (this.tools.has(tool.id)) {
+            console.warn(`Function with id ${tool.id} is already registered.`);
+        }
+        else {
+            this.tools.set(tool.id, tool);
+        }
+    }
+    getFunction(toolId) {
+        return this.tools.get(toolId);
+    }
+    getFunctions(...toolIds) {
+        const tools = toolIds.map(toolId => {
+            const tool = this.tools.get(toolId);
+            if (tool) {
+                return tool;
+            }
+            else {
+                throw new Error(`Function with id ${toolId} does not exist.`);
+            }
+        });
+        return tools;
+    }
+};
+exports.ToolInvocationRegistryImpl = ToolInvocationRegistryImpl;
+exports.ToolInvocationRegistryImpl = ToolInvocationRegistryImpl = tslib_1.__decorate([
+    (0, di_1.Injectable)()
+], ToolInvocationRegistryImpl);
+//# sourceMappingURL=tool-invocation-registry.js.map

package/lib/common/tool-invocation-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-invocation-registry.js","sourceRoot":"","sources":["../../src/common/tool-invocation-registry.ts"],"names":[],"mappings":";;;;AAAA,qCAA0C;AAC1C,6BAAwB;AAGX,QAAA,mBAAmB,GAAG,OAAC,CAAC,MAAM,CAAC;IACxC,IAAI,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IAChE,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC,CAAC,QAAQ,EAAE;IACjC,KAAK,EAAE,OAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,2BAAmB,CAAC,CAAC,QAAQ,EAAE;IACnD,UAAU,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,2BAAmB,CAAC,CAAC,CAAC,QAAQ,EAAE;IAClE,QAAQ,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;CAC3C,CAAC,CAAC;AAaH,IAAiB,WAAW,CAI3B;AAJD,WAAiB,WAAW;IACxB,SAAgB,eAAe,CAAC,GAAY;QACxC,OAAO,2BAAmB,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC;IACtD,CAAC;IAFe,2BAAe,kBAE9B,CAAA;AACL,CAAC,EAJgB,WAAW,2BAAX,WAAW,QAI3B;AAEY,QAAA,sBAAsB,GAAG,MAAM,CAAC,wBAAwB,CAAC,CAAC;AA8C1D,QAAA,YAAY,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;AAM5C,IAAM,0BAA0B,GAAhC,MAAM,0BAA0B;IAAhC;QAEK,UAAK,GAA6B,IAAI,GAAG,EAAuB,CAAC;IAsC7E,CAAC;IApCG,kBAAkB,CAAC,YAAoB;QACnC,MAAM,aAAa,GAAa,EAAE,CAAC;QACnC,KAAK,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,YAAY,KAAK,YAAY,EAAE,CAAC;gBACrC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3B,CAAC;QACL,CAAC;QACD,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACvD,CAAC;IACD,eAAe;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,YAAY,CAAC,IAAiB;QAC1B,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YAC1B,OAAO,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,EAAE,yBAAyB,CAAC,CAAC;QACvE,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAClC,CAAC;IACL,CAAC;IAED,WAAW,CAAC,MAAc;QACtB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IAED,YAAY,CAAC,GAAG,OAAiB;QAC7B,MAAM,KAAK,GAAkB,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;YAC9C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACpC,IAAI,IAAI,EAAE,CAAC;gBACP,OAAO,IAAI,CAAC;YAChB,CAAC;iBAAM,CAAC;gBACJ,MAAM,IAAI,KAAK,CAAC,oBAAoB,MAAM,kBAAkB,CAAC,CAAC;YAClE,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,KAAK,CAAC;IACjB,CAAC;CACJ,CAAA;AAxCY,gEAA0B;qCAA1B,0BAA0B;IADtC,IAAA,eAAU,GAAE;GACA,0BAA0B,CAwCtC"}

package/lib/common/types.d.ts

@@ -17,4 +17,18 @@ export interface INearestCodeBlock {
     offset: number;
     type?: NearestCodeBlockType;
 }
+export interface IMCPServerProxyService {
+    $callMCPTool(name: string, args: any): Promise<{
+        content: {
+            type: string;
+            text: string;
+        }[];
+        isError?: boolean;
+    }>;
+    $getMCPTools(): Promise<{
+        name: string;
+        description: string;
+        inputSchema: any;
+    }[]>;
+}
 //# sourceMappingURL=types.d.ts.map

package/lib/common/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/common/types.ts"],"names":[],"mappings":"AAAA,oBAAY,oBAAoB;IAC9B,KAAK,UAAU;IACf,IAAI,SAAS;CACd;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE;QACL,KAAK,EAAE;YACL,IAAI,EAAE,MAAM,CAAC;YACb,SAAS,EAAE,MAAM,CAAC;SACnB,CAAC;QACF,GAAG,EAAE;YACH,IAAI,EAAE,MAAM,CAAC;YACb,SAAS,EAAE,MAAM,CAAC;SACnB,CAAC;KACH,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,oBAAoB,CAAC;CAC7B"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/common/types.ts"],"names":[],"mappings":"AAAA,oBAAY,oBAAoB;IAC9B,KAAK,UAAU;IACf,IAAI,SAAS;CACd;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE;QACL,KAAK,EAAE;YACL,IAAI,EAAE,MAAM,CAAC;YACb,SAAS,EAAE,MAAM,CAAC;SACnB,CAAC;QACF,GAAG,EAAE;YACH,IAAI,EAAE,MAAM,CAAC;YACb,SAAS,EAAE,MAAM,CAAC;SACnB,CAAC;KACH,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,oBAAoB,CAAC;CAC7B;AAGD,MAAM,WAAW,sBAAsB;IACrC,YAAY,CACV,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,GAAG,GACR,OAAO,CAAC;QACT,OAAO,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAA;SAAE,EAAE,CAAC;QAC1C,OAAO,CAAC,EAAE,OAAO,CAAC;KACnB,CAAC,CAAC;IACH,YAAY,IAAI,OAAO,CACrB;QACE,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,EAAE,GAAG,CAAC;KAClB,EAAE,CACJ,CAAC;CACH"}

package/lib/node/anthropic/anthropic-language-model.d.ts

@@ -0,0 +1,13 @@
+import { AnthropicProvider } from '@ai-sdk/anthropic';
+import { ChatReadableStream } from '@opensumi/ide-core-node';
+import { CancellationToken } from '@opensumi/ide-utils';
+import { ToolRequest } from '../../common/tool-invocation-registry';
+export declare const AnthropicModelIdentifier: unique symbol;
+export declare class AnthropicModel {
+    private readonly toolInvocationRegistry;
+    protected initializeAnthropicProvider(): AnthropicProvider;
+    request(request: string, chatReadableStream: ChatReadableStream, cancellationToken?: CancellationToken): Promise<any>;
+    private convertToolRequestToAITool;
+    protected handleStreamingRequest(anthropic: AnthropicProvider, request: string, tools: ToolRequest[], chatReadableStream: ChatReadableStream, cancellationToken?: CancellationToken): Promise<any>;
+}
+//# sourceMappingURL=anthropic-language-model.d.ts.map

package/lib/node/anthropic/anthropic-language-model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic-language-model.d.ts","sourceRoot":"","sources":["../../../src/node/anthropic/anthropic-language-model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAmB,MAAM,mBAAmB,CAAC;AAIvE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAExD,OAAO,EAAsD,WAAW,EAAE,MAAM,uCAAuC,CAAC;AAGxH,eAAO,MAAM,wBAAwB,eAAqC,CAAC;AAI3E,qBACa,cAAc;IAEzB,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAA6B;IAEpE,SAAS,CAAC,2BAA2B;IAU/B,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,iBAAiB,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC;IAM3H,OAAO,CAAC,0BAA0B;cASlB,sBAAsB,CACpC,SAAS,EAAE,iBAAiB,EAC5B,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,WAAW,EAAE,EACpB,kBAAkB,EAAE,kBAAkB,EACtC,iBAAiB,CAAC,EAAE,iBAAiB,GACpC,OAAO,CAAC,GAAG,CAAC;CA6ChB"}

package/lib/node/anthropic/anthropic-language-model.js

@@ -0,0 +1,79 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AnthropicModel = exports.AnthropicModelIdentifier = void 0;
+const tslib_1 = require("tslib");
+const anthropic_1 = require("@ai-sdk/anthropic");
+const ai_1 = require("ai");
+const di_1 = require("@opensumi/di");
+const tool_invocation_registry_1 = require("../../common/tool-invocation-registry");
+exports.AnthropicModelIdentifier = Symbol('AnthropicModelIdentifier');
+const apiKey = '';
+let AnthropicModel = class AnthropicModel {
+    initializeAnthropicProvider() {
+        if (!apiKey) {
+            throw new Error('Please provide ANTHROPIC_API_KEY in preferences or via environment variable');
+        }
+        const anthropic = (0, anthropic_1.createAnthropic)({ apiKey });
+        return anthropic;
+    }
+    async request(request, chatReadableStream, cancellationToken) {
+        const anthropic = this.initializeAnthropicProvider();
+        const allFunctions = this.toolInvocationRegistry.getAllFunctions();
+        return this.handleStreamingRequest(anthropic, request, allFunctions, chatReadableStream, cancellationToken);
+    }
+    convertToolRequestToAITool(toolRequest) {
+        return (0, ai_1.tool)({
+            description: toolRequest.description || '',
+            // TODO 这里应该是 z.object 而不是 JSON Schema
+            parameters: (0, ai_1.jsonSchema)(toolRequest.parameters),
+            execute: async (args) => await toolRequest.handler(JSON.stringify(args)),
+        });
+    }
+    async handleStreamingRequest(anthropic, request, tools, chatReadableStream, cancellationToken) {
+        try {
+            const aiTools = Object.fromEntries(tools.map((tool) => [tool.name, this.convertToolRequestToAITool(tool)]));
+            const abortController = new AbortController();
+            if (cancellationToken) {
+                cancellationToken.onCancellationRequested(() => {
+                    abortController.abort();
+                });
+            }
+            const stream = await (0, ai_1.streamText)({
+                model: anthropic('claude-3-5-sonnet-20241022'),
+                maxTokens: 4096,
+                tools: aiTools,
+                messages: [{ role: 'user', content: request }],
+                abortSignal: abortController.signal,
+                maxSteps: 5,
+            });
+            for await (const chunk of stream.fullStream) {
+                console.log(chunk);
+                if (chunk.type === 'text-delta') {
+                    chatReadableStream.emitData({ kind: 'content', content: chunk.textDelta });
+                }
+                else if (chunk.type === 'tool-call') {
+                    chatReadableStream.emitData({ kind: 'toolCall', content: {
+                            id: chunk.toolCallId || Date.now().toString(),
+                            type: 'function',
+                            function: { name: chunk.toolName, arguments: JSON.stringify(chunk.args) },
+                        } });
+                }
+            }
+            chatReadableStream.end();
+        }
+        catch (error) {
+            console.error('Error during streaming:', error);
+            chatReadableStream.emitError(error);
+        }
+        return chatReadableStream;
+    }
+};
+exports.AnthropicModel = AnthropicModel;
+tslib_1.__decorate([
+    (0, di_1.Autowired)(tool_invocation_registry_1.ToolInvocationRegistry),
+    tslib_1.__metadata("design:type", tool_invocation_registry_1.ToolInvocationRegistryImpl)
+], AnthropicModel.prototype, "toolInvocationRegistry", void 0);
+exports.AnthropicModel = AnthropicModel = tslib_1.__decorate([
+    (0, di_1.Injectable)()
+], AnthropicModel);
+//# sourceMappingURL=anthropic-language-model.js.map

package/lib/node/anthropic/anthropic-language-model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic-language-model.js","sourceRoot":"","sources":["../../../src/node/anthropic/anthropic-language-model.ts"],"names":[],"mappings":";;;;AAAA,iDAAuE;AACvE,2BAAkD;AAElD,qCAAqD;AAIrD,oFAAwH;AAG3G,QAAA,wBAAwB,GAAG,MAAM,CAAC,0BAA0B,CAAC,CAAC;AAE3E,MAAM,MAAM,GAAG,EAAE,CAAC;AAGX,IAAM,cAAc,GAApB,MAAM,cAAc;IAIf,2BAA2B;QACnC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,6EAA6E,CAAC,CAAC;QACjG,CAAC;QAED,MAAM,SAAS,GAAG,IAAA,2BAAe,EAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAE9C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAe,EAAE,kBAAsC,EAAE,iBAAqC;QAC1G,MAAM,SAAS,GAAG,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACrD,MAAM,YAAY,GAAG,IAAI,CAAC,sBAAsB,CAAC,eAAe,EAAE,CAAC;QACnE,OAAO,IAAI,CAAC,sBAAsB,CAAC,SAAS,EAAE,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,iBAAiB,CAAC,CAAC;IAC9G,CAAC;IAEO,0BAA0B,CAAC,WAAwB;QACzD,OAAO,IAAA,SAAI,EAAC;YACV,WAAW,EAAE,WAAW,CAAC,WAAW,IAAI,EAAE;YAC1C,sCAAsC;YACtC,UAAU,EAAE,IAAA,eAAU,EAAC,WAAW,CAAC,UAAU,CAAC;YAC9C,OAAO,EAAE,KAAK,EAAE,IAAS,EAAE,EAAE,CAAC,MAAM,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;SAC9E,CAAC,CAAC;IACL,CAAC;IAES,KAAK,CAAC,sBAAsB,CACpC,SAA4B,EAC5B,OAAe,EACf,KAAoB,EACpB,kBAAsC,EACtC,iBAAqC;QAGrC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAChC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,CAAC,CAAC,CACxE,CAAC;YAEF,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;YAC9C,IAAI,iBAAiB,EAAE,CAAC;gBACtB,iBAAiB,CAAC,uBAAuB,CAAC,GAAG,EAAE;oBAC7C,eAAe,CAAC,KAAK,EAAE,CAAC;gBAC1B,CAAC,CAAC,CAAC;YACL,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAA,eAAU,EAAC;gBAC9B,KAAK,EAAE,SAAS,CAAC,4BAA4B,CAAC;gBAC9C,SAAS,EAAE,IAAI;gBACf,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;gBAC9C,WAAW,EAAE,eAAe,CAAC,MAAM;gBACnC,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;gBAC5C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBACnB,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAChC,kBAAkB,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC;gBAC7E,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACtC,kBAAkB,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE;4BACvD,EAAE,EAAE,KAAK,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;4BAC7C,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;yBAC1E,EAAC,CAAC,CAAC;gBACN,CAAC;YACH,CAAC;YAED,kBAAkB,CAAC,GAAG,EAAE,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;YAChD,kBAAkB,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACtC,CAAC;QAED,OAAO,kBAAkB,CAAC;IAC5B,CAAC;CAEF,CAAA;AAhFY,wCAAc;AAER;IADhB,IAAA,cAAS,EAAC,iDAAsB,CAAC;sCACO,qDAA0B;8DAAC;yBAFzD,cAAc;IAD1B,IAAA,eAAU,GAAE;GACA,cAAc,CAgF1B"}

package/lib/node/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/node/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,QAAQ,EAAE,MAAM,cAAc,CAAC;AAEpD,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAGrD,qBACa,cAAe,SAAQ,UAAU;IAC5C,SAAS,EAAE,QAAQ,EAAE,CAKnB;IAEF,YAAY;;;QAKV;CACH"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/node/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,QAAQ,EAAE,MAAM,cAAc,CAAC;AAEpD,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAWrD,qBACa,cAAe,SAAQ,UAAU;IAC5C,SAAS,EAAE,QAAQ,EAAE,CAiBnB;IAEF,YAAY;;;QASV;CACH"}

package/lib/node/index.js

@@ -6,6 +6,11 @@ const di_1 = require("@opensumi/di");
 const ide_core_common_1 = require("@opensumi/ide-core-common");
 const ide_core_node_1 = require("@opensumi/ide-core-node");
 const base_back_service_1 = require("@opensumi/ide-core-node/lib/ai-native/base-back.service");
+const common_1 = require("../common");
+const mcp_server_manager_1 = require("../common/mcp-server-manager");
+const tool_invocation_registry_1 = require("../common/tool-invocation-registry");
+const sumi_mcp_server_1 = require("./mcp/sumi-mcp-server");
+const mcp_server_manager_impl_1 = require("./mcp-server-manager-impl");
 let AINativeModule = class AINativeModule extends ide_core_node_1.NodeModule {
     constructor() {
         super(...arguments);
@@ -14,12 +19,28 @@ let AINativeModule = class AINativeModule extends ide_core_node_1.NodeModule {
                 token: ide_core_common_1.AIBackSerivceToken,
                 useClass: base_back_service_1.BaseAIBackService,
             },
+            {
+                token: mcp_server_manager_1.MCPServerManager,
+                useClass: mcp_server_manager_impl_1.MCPServerManagerImpl,
+            },
+            {
+                token: tool_invocation_registry_1.ToolInvocationRegistry,
+                useClass: tool_invocation_registry_1.ToolInvocationRegistryImpl,
+            },
+            {
+                token: common_1.TokenMCPServerProxyService,
+                useClass: sumi_mcp_server_1.SumiMCPServerBackend,
+            },
         ];
         this.backServices = [
             {
                 servicePath: ide_core_common_1.AIBackSerivcePath,
                 token: ide_core_common_1.AIBackSerivceToken,
             },
+            {
+                servicePath: common_1.SumiMCPServerProxyServicePath,
+                token: common_1.TokenMCPServerProxyService,
+            },
         ];
     }
 };

package/lib/node/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/node/index.ts"],"names":[],"mappings":";;;;AAAA,qCAAoD;AACpD,+DAAkF;AAClF,2DAAqD;AACrD,+FAA4F;AAGrF,IAAM,cAAc,GAApB,MAAM,cAAe,SAAQ,0BAAU;IAAvC;;QACL,cAAS,GAAe;YACtB;gBACE,KAAK,EAAE,oCAAkB;gBACzB,QAAQ,EAAE,qCAAiB;aAC5B;SACF,CAAC;QAEF,iBAAY,GAAG;YACb;gBACE,WAAW,EAAE,mCAAiB;gBAC9B,KAAK,EAAE,oCAAkB;aAC1B;SACF,CAAC;IACJ,CAAC;CAAA,CAAA;AAdY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,eAAU,GAAE;GACA,cAAc,CAc1B"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/node/index.ts"],"names":[],"mappings":";;;;AAAA,qCAAoD;AACpD,+DAAkF;AAClF,2DAAqD;AACrD,+FAA4F;AAE5F,sCAAuF;AACvF,qEAAgE;AAChE,iFAAwG;AAExG,2DAA6D;AAC7D,uEAAiE;AAI1D,IAAM,cAAc,GAApB,MAAM,cAAe,SAAQ,0BAAU;IAAvC;;QACL,cAAS,GAAe;YACtB;gBACE,KAAK,EAAE,oCAAkB;gBACzB,QAAQ,EAAE,qCAAiB;aAC5B;YACD;gBACE,KAAK,EAAE,qCAAgB;gBACvB,QAAQ,EAAE,8CAAoB;aAC/B;YACD;gBACE,KAAK,EAAE,iDAAsB;gBAC7B,QAAQ,EAAE,qDAA0B;aACrC;YACD;gBACE,KAAK,EAAE,mCAA0B;gBACjC,QAAQ,EAAE,sCAAoB;aAC/B;SACF,CAAC;QAEF,iBAAY,GAAG;YACb;gBACE,WAAW,EAAE,mCAAiB;gBAC9B,KAAK,EAAE,oCAAkB;aAC1B;YACD;gBACE,WAAW,EAAE,sCAA6B;gBAC1C,KAAK,EAAE,mCAA0B;aAClC;SACF,CAAC;IACJ,CAAC;CAAA,CAAA;AA9BY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,eAAU,GAAE;GACA,cAAc,CA8B1B"}

package/lib/node/mcp/sumi-mcp-server.d.ts

@@ -0,0 +1,84 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { RPCService } from '@opensumi/ide-connection';
+import { IMCPServerProxyService } from '../../common/types';
+import { IMCPServer } from '../mcp-server';
+export declare class SumiMCPServerBackend extends RPCService<IMCPServerProxyService> {
+    private readonly mcpServerManager;
+    private server;
+    getMCPTools(): Promise<{
+        name: string;
+        description: string;
+        inputSchema: any;
+    }[]>;
+    callMCPTool(name: string, args: any): Promise<{
+        content: {
+            type: string;
+            text: string;
+        }[];
+        isError?: boolean;
+    }>;
+    getServer(): Server<{
+        method: string;
+        params?: {
+            [x: string]: unknown;
+            _meta?: {
+                [x: string]: unknown;
+                progressToken?: string | number | undefined;
+            } | undefined;
+        } | undefined;
+    }, {
+        method: string;
+        params?: {
+            [x: string]: unknown;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | undefined;
+    }, {
+        [x: string]: unknown;
+        _meta?: {
+            [x: string]: unknown;
+        } | undefined;
+    }> | undefined;
+    initBuiltinMCPServer(): void;
+    initExposedMCPServer(): Promise<Server<{
+        method: string;
+        params?: {
+            [x: string]: unknown;
+            _meta?: {
+                [x: string]: unknown;
+                progressToken?: string | number | undefined;
+            } | undefined;
+        } | undefined;
+    }, {
+        method: string;
+        params?: {
+            [x: string]: unknown;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | undefined;
+    }, {
+        [x: string]: unknown;
+        _meta?: {
+            [x: string]: unknown;
+        } | undefined;
+    }>>;
+}
+export declare const TokenBuiltinMCPServer: unique symbol;
+export declare class BuiltinMCPServer implements IMCPServer {
+    private readonly sumiMCPServer;
+    constructor(sumiMCPServer: SumiMCPServerBackend);
+    private started;
+    isStarted(): boolean;
+    getServerName(): string;
+    start(): Promise<void>;
+    callTool(toolName: string, arg_string: string): Promise<any>;
+    getTools(): ReturnType<Client['listTools']>;
+    update(_command: string, _args?: string[], _env?: {
+        [key: string]: string;
+    }): void;
+    stop(): void;
+}
+//# sourceMappingURL=sumi-mcp-server.d.ts.map

package/lib/node/mcp/sumi-mcp-server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sumi-mcp-server.d.ts","sourceRoot":"","sources":["../../../src/node/mcp/sumi-mcp-server.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAInE,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAGtD,OAAO,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAQ3C,qBACa,oBAAqB,SAAQ,UAAU,CAAC,sBAAsB,CAAC;IAK1E,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAuB;IAExD,OAAO,CAAC,MAAM,CAAqB;IAE7B,WAAW;;;;;IAUX,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG;;;;;;;IAOzC,SAAS;;;;;;;;;;;;;;;;;;;;;;;IAIT,oBAAoB;IAKd,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;CAoC3B;AAGD,eAAO,MAAM,qBAAqB,eAAkC,CAAC;AAErE,qBAAa,gBAAiB,YAAW,UAAU;IAG/C,OAAO,CAAC,QAAQ,CAAC,aAAa;gBAAb,aAAa,EAAE,oBAAoB;IAGtD,OAAO,CAAC,OAAO,CAAiB;IAEhC,SAAS,IAAI,OAAO;IAIpB,aAAa,IAAI,MAAM;IAIjB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAStB,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC;IAkB5D,QAAQ,IAAI,UAAU,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAQjD,MAAM,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAAA;KAAE,GAAG,IAAI;IAIlF,IAAI,IAAI,IAAI;CAOb"}

package/lib/node/mcp/sumi-mcp-server.js

@@ -0,0 +1,134 @@
+"use strict";
+// 想要通过 MCP 的方式暴露 Opensumi 的 IDE 能力，就需要 Node.js 层打通 MCP 的通信
+// 因为大部分 MCP 功能的实现在前端，因此需要再这里做前后端通信
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BuiltinMCPServer = exports.TokenBuiltinMCPServer = exports.SumiMCPServerBackend = void 0;
+const tslib_1 = require("tslib");
+const index_js_1 = require("@modelcontextprotocol/sdk/server/index.js");
+const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
+const di_1 = require("@opensumi/di");
+const ide_connection_1 = require("@opensumi/ide-connection");
+const mcp_server_manager_1 = require("../../common/mcp-server-manager");
+const mcp_server_manager_impl_1 = require("../mcp-server-manager-impl");
+// 每个 BrowserTab 都对应了一个 SumiMCPServerBackend 实例
+// SumiMCPServerBackend 需要做的事情：
+// 维护 Browser 端工具的注册和调用
+// 处理第三方 MCP Server 的注册和调用
+let SumiMCPServerBackend = class SumiMCPServerBackend extends ide_connection_1.RPCService {
+    async getMCPTools() {
+        if (!this.client) {
+            throw new Error('SUMI MCP RPC Client not initialized');
+        }
+        // 获取 MCP 工具
+        const tools = await this.client.$getMCPTools();
+        console.log('[Node backend] SUMI MCP tools', tools);
+        return tools;
+    }
+    async callMCPTool(name, args) {
+        if (!this.client) {
+            throw new Error('SUMI MCP RPC Client not initialized');
+        }
+        return await this.client.$callMCPTool(name, args);
+    }
+    getServer() {
+        return this.server;
+    }
+    initBuiltinMCPServer() {
+        const builtinMCPServer = new BuiltinMCPServer(this);
+        this.mcpServerManager.initBuiltinServer(builtinMCPServer);
+    }
+    async initExposedMCPServer() {
+        // 初始化 MCP Server
+        this.server = new index_js_1.Server({
+            name: 'sumi-ide-mcp-server',
+            version: '0.2.0',
+        }, {
+            capabilities: {
+                tools: {},
+            },
+        });
+        // 设置工具列表请求处理器
+        this.server.setRequestHandler(types_js_1.ListToolsRequestSchema, async () => {
+            const tools = await this.getMCPTools();
+            return { tools };
+        });
+        // 设置工具调用请求处理器
+        this.server.setRequestHandler(types_js_1.CallToolRequestSchema, async (request) => {
+            try {
+                const { name, arguments: args } = request.params;
+                return await this.callMCPTool(name, args);
+            }
+            catch (error) {
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                return {
+                    content: [{ type: 'text', text: `Error: ${errorMessage}` }],
+                    isError: true,
+                };
+            }
+        });
+        return this.server;
+    }
+};
+exports.SumiMCPServerBackend = SumiMCPServerBackend;
+tslib_1.__decorate([
+    (0, di_1.Autowired)(mcp_server_manager_1.MCPServerManager),
+    tslib_1.__metadata("design:type", mcp_server_manager_impl_1.MCPServerManagerImpl)
+], SumiMCPServerBackend.prototype, "mcpServerManager", void 0);
+exports.SumiMCPServerBackend = SumiMCPServerBackend = tslib_1.__decorate([
+    (0, di_1.Injectable)({ multiple: true })
+], SumiMCPServerBackend);
+exports.TokenBuiltinMCPServer = Symbol('TokenBuiltinMCPServer');
+class BuiltinMCPServer {
+    constructor(sumiMCPServer) {
+        this.sumiMCPServer = sumiMCPServer;
+        this.started = true;
+    }
+    isStarted() {
+        return this.started;
+    }
+    getServerName() {
+        return 'sumi-builtin';
+    }
+    async start() {
+        if (this.started) {
+            return;
+        }
+        // TODO 考虑 MCP Server 的对外暴露
+        // await this.sumiMCPServer.initMCPServer();
+        this.started = true;
+    }
+    async callTool(toolName, arg_string) {
+        if (!this.started) {
+            throw new Error('MCP Server not started');
+        }
+        let args;
+        try {
+            args = JSON.parse(arg_string);
+        }
+        catch (error) {
+            console.error(`Failed to parse arguments for calling tool "${toolName}" in Builtin MCP server.
+        Invalid JSON: ${arg_string}`, error);
+            throw error;
+        }
+        return this.sumiMCPServer.callMCPTool(toolName, args);
+    }
+    async getTools() {
+        if (!this.started) {
+            throw new Error('MCP Server not started');
+        }
+        const tools = await this.sumiMCPServer.getMCPTools();
+        return { tools };
+    }
+    update(_command, _args, _env) {
+        // No-op for builtin server as it doesn't need command/args/env updates
+    }
+    stop() {
+        if (!this.started) {
+            return;
+        }
+        // No explicit cleanup needed for in-memory server
+        this.started = false;
+    }
+}
+exports.BuiltinMCPServer = BuiltinMCPServer;
+//# sourceMappingURL=sumi-mcp-server.js.map

package/lib/node/mcp/sumi-mcp-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sumi-mcp-server.js","sourceRoot":"","sources":["../../../src/node/mcp/sumi-mcp-server.ts"],"names":[],"mappings":";AAAA,2DAA2D;AAC3D,mCAAmC;;;;AAGnC,wEAAmE;AACnE,iEAAmG;AAEnG,qCAAqD;AACrD,6DAAsD;AAEtD,wEAAmE;AAGnE,wEAAkE;AAElE,+CAA+C;AAC/C,+BAA+B;AAC/B,uBAAuB;AACvB,0BAA0B;AAGnB,IAAM,oBAAoB,GAA1B,MAAM,oBAAqB,SAAQ,2BAAkC;IAS1E,KAAK,CAAC,WAAW;QACf,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QACD,YAAY;QACZ,MAAM,KAAK,GAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;QAChD,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;QACpD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAY,EAAE,IAAS;QACvC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QACD,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IACpD,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,oBAAoB;QAClB,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACpD,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,oBAAoB;QACxB,iBAAiB;QACjB,IAAI,CAAC,MAAM,GAAG,IAAI,iBAAM,CACtB;YACE,IAAI,EAAE,qBAAqB;YAC3B,OAAO,EAAE,OAAO;SACjB,EACD;YACE,YAAY,EAAE;gBACZ,KAAK,EAAE,EAAE;aACV;SACF,CACF,CAAC;QAEF,cAAc;QACd,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,iCAAsB,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;YACvC,OAAO,EAAE,KAAK,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,cAAc;QACd,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,gCAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACrE,IAAI,CAAC;gBACH,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;gBACjD,OAAO,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC5C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC5E,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,YAAY,EAAE,EAAE,CAAC;oBAC3D,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;CACF,CAAA;AAvEY,oDAAoB;AAKd;IADhB,IAAA,cAAS,EAAC,qCAAgB,CAAC;sCACO,8CAAoB;8DAAC;+BAL7C,oBAAoB;IADhC,IAAA,eAAU,EAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;GAClB,oBAAoB,CAuEhC;AAGY,QAAA,qBAAqB,GAAG,MAAM,CAAC,uBAAuB,CAAC,CAAC;AAErE,MAAa,gBAAgB;IAE3B,YACmB,aAAmC;QAAnC,kBAAa,GAAb,aAAa,CAAsB;QAG9C,YAAO,GAAY,IAAI,CAAC;IAF7B,CAAC;IAIJ,SAAS;QACP,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,aAAa;QACX,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,2BAA2B;QAC3B,4CAA4C;QAC5C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,UAAkB;QACjD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,IAAI,CAAC;QACT,IAAI,CAAC;YACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,+CAA+C,QAAQ;wBACvC,UAAU,EAAE,EAC5B,KAAK,CACN,CAAC;YACF,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;QACrD,OAAO,EAAE,KAAK,EAAE,CAAC;IACnB,CAAC;IAED,MAAM,CAAC,QAAgB,EAAE,KAAgB,EAAE,IAAgC;QACzE,uEAAuE;IACzE,CAAC;IAED,IAAI;QACF,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QACD,kDAAkD;QAClD,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;IACvB,CAAC;CACF;AA9DD,4CA8DC"}

package/lib/node/mcp-server-manager-impl.d.ts

@@ -0,0 +1,20 @@
+import { MCPServerDescription, MCPServerManager } from '../common/mcp-server-manager';
+import { BuiltinMCPServer } from './mcp/sumi-mcp-server';
+import { IMCPServer } from './mcp-server';
+export declare class MCPServerManagerImpl implements MCPServerManager {
+    private readonly toolInvocationRegistry;
+    protected servers: Map<string, IMCPServer>;
+    stopServer(serverName: string): Promise<void>;
+    getStartedServers(): Promise<string[]>;
+    callTool(serverName: string, toolName: string, arg_string: string): ReturnType<IMCPServer['callTool']>;
+    startServer(serverName: string): Promise<void>;
+    getServerNames(): Promise<string[]>;
+    private convertToToolRequest;
+    collectTools(serverName: string): Promise<void>;
+    getTools(serverName: string): ReturnType<IMCPServer['getTools']>;
+    addOrUpdateServer(description: MCPServerDescription): void;
+    addOrUpdateServerDirectly(server: IMCPServer): void;
+    initBuiltinServer(builtinMCPServer: BuiltinMCPServer): void;
+    removeServer(name: string): void;
+}
+//# sourceMappingURL=mcp-server-manager-impl.d.ts.map

package/lib/node/mcp-server-manager-impl.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-server-manager-impl.d.ts","sourceRoot":"","sources":["../../src/node/mcp-server-manager-impl.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,oBAAoB,EAAE,gBAAgB,EAAW,MAAM,8BAA8B,CAAC;AAG/F,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,UAAU,EAAiB,MAAM,cAAc,CAAC;AAEzD,qBACa,oBAAqB,YAAW,gBAAgB;IAE3D,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAA6B;IAEpE,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAa;IAEjD,UAAU,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAS7C,iBAAiB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAU5C,QAAQ,CAAC,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,UAAU,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IAQhG,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ9C,cAAc,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAIzC,OAAO,CAAC,oBAAoB;IAuBf,YAAY,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAc/C,QAAQ,CAAC,UAAU,EAAE,MAAM,GAAG,UAAU,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IAQ7E,iBAAiB,CAAC,WAAW,EAAE,oBAAoB,GAAG,IAAI;IAY1D,yBAAyB,CAAC,MAAM,EAAE,UAAU,GAAG,IAAI;IAInD,iBAAiB,CAAC,gBAAgB,EAAE,gBAAgB,GAAG,IAAI;IAK3D,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;CASjC"}

package/lib/node/mcp-server-manager-impl.js

@@ -0,0 +1,123 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MCPServerManagerImpl = void 0;
+const tslib_1 = require("tslib");
+const di_1 = require("@opensumi/di");
+const tool_invocation_registry_1 = require("../common/tool-invocation-registry");
+const mcp_server_1 = require("./mcp-server");
+let MCPServerManagerImpl = class MCPServerManagerImpl {
+    constructor() {
+        this.servers = new Map();
+    }
+    async stopServer(serverName) {
+        const server = this.servers.get(serverName);
+        if (!server) {
+            throw new Error(`MCP server "${serverName}" not found.`);
+        }
+        server.stop();
+        console.log(`MCP server "${serverName}" stopped.`);
+    }
+    async getStartedServers() {
+        const startedServers = [];
+        for (const [name, server] of this.servers.entries()) {
+            if (server.isStarted()) {
+                startedServers.push(name);
+            }
+        }
+        return startedServers;
+    }
+    callTool(serverName, toolName, arg_string) {
+        const server = this.servers.get(serverName);
+        if (!server) {
+            throw new Error(`MCP server "${toolName}" not found.`);
+        }
+        return server.callTool(toolName, arg_string);
+    }
+    async startServer(serverName) {
+        const server = this.servers.get(serverName);
+        if (!server) {
+            throw new Error(`MCP server "${serverName}" not found.`);
+        }
+        await server.start();
+    }
+    async getServerNames() {
+        return Array.from(this.servers.keys());
+    }
+    convertToToolRequest(tool, serverName) {
+        const id = `mcp_${serverName}_${tool.name}`;
+        return {
+            id,
+            name: id,
+            providerName: `mcp_${serverName}`,
+            parameters: tool.inputSchema,
+            description: tool.description,
+            handler: async (arg_string) => {
+                try {
+                    const res = await this.callTool(serverName, tool.name, arg_string);
+                    console.log(`[MCP: ${serverName}] ${tool.name} called with ${arg_string}`);
+                    console.log(res);
+                    return JSON.stringify(res);
+                }
+                catch (error) {
+                    console.error(`Error in tool handler for ${tool.name} on MCP server ${serverName}:`, error);
+                    throw error;
+                }
+            },
+        };
+    }
+    async collectTools(serverName) {
+        const server = this.servers.get(serverName);
+        if (!server) {
+            throw new Error(`MCP server "${serverName}" not found.`);
+        }
+        const { tools } = await server.getTools();
+        const toolRequests = tools.map((tool) => this.convertToToolRequest(tool, serverName));
+        for (const toolRequest of toolRequests) {
+            this.toolInvocationRegistry.registerTool(toolRequest);
+        }
+    }
+    async getTools(serverName) {
+        const server = this.servers.get(serverName);
+        if (!server) {
+            throw new Error(`MCP server "${serverName}" not found.`);
+        }
+        return server.getTools();
+    }
+    addOrUpdateServer(description) {
+        const { name, command, args, env } = description;
+        const existingServer = this.servers.get(name);
+        if (existingServer) {
+            existingServer.update(command, args, env);
+        }
+        else {
+            const newServer = new mcp_server_1.MCPServerImpl(name, command, args, env);
+            this.servers.set(name, newServer);
+        }
+    }
+    addOrUpdateServerDirectly(server) {
+        this.servers.set(server.getServerName(), server);
+    }
+    initBuiltinServer(builtinMCPServer) {
+        this.addOrUpdateServerDirectly(builtinMCPServer);
+        this.collectTools(builtinMCPServer.getServerName());
+    }
+    removeServer(name) {
+        const server = this.servers.get(name);
+        if (server) {
+            server.stop();
+            this.servers.delete(name);
+        }
+        else {
+            console.warn(`MCP server "${name}" not found.`);
+        }
+    }
+};
+exports.MCPServerManagerImpl = MCPServerManagerImpl;
+tslib_1.__decorate([
+    (0, di_1.Autowired)(tool_invocation_registry_1.ToolInvocationRegistry),
+    tslib_1.__metadata("design:type", tool_invocation_registry_1.ToolInvocationRegistryImpl)
+], MCPServerManagerImpl.prototype, "toolInvocationRegistry", void 0);
+exports.MCPServerManagerImpl = MCPServerManagerImpl = tslib_1.__decorate([
+    (0, di_1.Injectable)()
+], MCPServerManagerImpl);
+//# sourceMappingURL=mcp-server-manager-impl.js.map

package/lib/node/mcp-server-manager-impl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-server-manager-impl.js","sourceRoot":"","sources":["../../src/node/mcp-server-manager-impl.ts"],"names":[],"mappings":";;;;AAAA,qCAAqD;AAGrD,iFAAqH;AAGrH,6CAAyD;AAGlD,IAAM,oBAAoB,GAA1B,MAAM,oBAAoB;IAA1B;QAIK,YAAO,GAA4B,IAAI,GAAG,EAAE,CAAC;IAoHzD,CAAC;IAlHC,KAAK,CAAC,UAAU,CAAC,UAAkB;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,eAAe,UAAU,cAAc,CAAC,CAAC;QAC3D,CAAC;QACD,MAAM,CAAC,IAAI,EAAE,CAAC;QACd,OAAO,CAAC,GAAG,CAAC,eAAe,UAAU,YAAY,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,iBAAiB;QACrB,MAAM,cAAc,GAAa,EAAE,CAAC;QACpC,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACpD,IAAI,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC;gBACvB,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QACD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,QAAQ,CAAC,UAAkB,EAAE,QAAgB,EAAE,UAAkB;QAC/D,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,eAAe,QAAQ,cAAc,CAAC,CAAC;QACzD,CAAC;QACD,OAAO,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,UAAkB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,eAAe,UAAU,cAAc,CAAC,CAAC;QAC3D,CAAC;QACD,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACzC,CAAC;IAEO,oBAAoB,CAAC,IAAa,EAAE,UAAkB;QAC5D,MAAM,EAAE,GAAG,OAAO,UAAU,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QAE5C,OAAO;YACL,EAAE;YACF,IAAI,EAAE,EAAE;YACR,YAAY,EAAE,OAAO,UAAU,EAAE;YACjC,UAAU,EAAE,IAAI,CAAC,WAAW;YAC5B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,OAAO,EAAE,KAAK,EAAE,UAAkB,EAAE,EAAE;gBACpC,IAAI,CAAC;oBACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;oBACnE,OAAO,CAAC,GAAG,CAAC,SAAS,UAAU,KAAK,IAAI,CAAC,IAAI,gBAAgB,UAAU,EAAE,CAAC,CAAC;oBAC3E,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;oBACjB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBAC7B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,IAAI,kBAAkB,UAAU,GAAG,EAAE,KAAK,CAAC,CAAC;oBAC5F,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;SACF,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,UAAkB;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,eAAe,UAAU,cAAc,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,QAAQ,EAAE,CAAC;QAC1C,MAAM,YAAY,GAAkB,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC;QAErG,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;YACvC,IAAI,CAAC,sBAAsB,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,QAAQ,CAAC,UAAkB;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,eAAe,UAAU,cAAc,CAAC,CAAC;QAC3D,CAAC;QACD,OAAO,MAAM,CAAC,QAAQ,EAAE,CAAC;IAC3B,CAAC;IAED,iBAAiB,CAAC,WAAiC;QACjD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,WAAW,CAAC;QACjD,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAE9C,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;QAC5C,CAAC;aAAM,CAAC;YACN,MAAM,SAAS,GAAG,IAAI,0BAAa,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YAC9D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,yBAAyB,CAAC,MAAkB;QAC1C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,MAAM,CAAC,CAAC;IACnD,CAAC;IAED,iBAAiB,CAAC,gBAAkC;QAClD,IAAI,CAAC,yBAAyB,CAAC,gBAAgB,CAAC,CAAC;QACjD,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC;IACtD,CAAC;IAED,YAAY,CAAC,IAAY;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,IAAI,EAAE,CAAC;YACd,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,CAAC,eAAe,IAAI,cAAc,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;CACF,CAAA;AAxHY,oDAAoB;AAEd;IADhB,IAAA,cAAS,EAAC,iDAAsB,CAAC;sCACO,qDAA0B;oEAAC;+BAFzD,oBAAoB;IADhC,IAAA,eAAU,GAAE;GACA,oBAAoB,CAwHhC"}