@opensumi/ide-ai-native 3.7.2-next-1739859371.0 → 3.7.2-next-1740013940.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/lib/browser/ai-core.contribution.d.ts +3 -0
- package/lib/browser/ai-core.contribution.d.ts.map +1 -1
- package/lib/browser/ai-core.contribution.js +68 -2
- package/lib/browser/ai-core.contribution.js.map +1 -1
- package/lib/browser/chat/chat-model.d.ts +2 -2
- package/lib/browser/chat/chat-model.d.ts.map +1 -1
- package/lib/browser/chat/chat-model.js +16 -5
- package/lib/browser/chat/chat-model.js.map +1 -1
- package/lib/browser/chat/chat-proxy.service.d.ts +4 -0
- package/lib/browser/chat/chat-proxy.service.d.ts.map +1 -1
- package/lib/browser/chat/chat-proxy.service.js +43 -0
- package/lib/browser/chat/chat-proxy.service.js.map +1 -1
- package/lib/browser/chat/chat.view.d.ts.map +1 -1
- package/lib/browser/chat/chat.view.js +29 -2
- package/lib/browser/chat/chat.view.js.map +1 -1
- package/lib/browser/components/ChatContext/ContextSelector.d.ts +12 -0
- package/lib/browser/components/ChatContext/ContextSelector.d.ts.map +1 -0
- package/lib/browser/components/ChatContext/ContextSelector.js +113 -0
- package/lib/browser/components/ChatContext/ContextSelector.js.map +1 -0
- package/lib/browser/components/ChatContext/index.d.ts +4 -0
- package/lib/browser/components/ChatContext/index.d.ts.map +1 -0
- package/lib/browser/components/ChatContext/index.js +84 -0
- package/lib/browser/components/ChatContext/index.js.map +1 -0
- package/lib/browser/components/ChatContext/style.module.less +189 -0
- package/lib/browser/components/ChatInput.d.ts.map +1 -1
- package/lib/browser/components/ChatInput.js.map +1 -1
- package/lib/browser/components/ChatReply.d.ts.map +1 -1
- package/lib/browser/components/ChatReply.js +25 -0
- package/lib/browser/components/ChatReply.js.map +1 -1
- package/lib/browser/components/ChatToolRender.d.ts +6 -0
- package/lib/browser/components/ChatToolRender.d.ts.map +1 -0
- package/lib/browser/components/ChatToolRender.js +53 -0
- package/lib/browser/components/ChatToolRender.js.map +1 -0
- package/lib/browser/components/ChatToolRender.module.less +86 -0
- package/lib/browser/components/components.module.less +32 -31
- package/lib/browser/components/utils.d.ts +2 -2
- package/lib/browser/context/llm-context.contribution.d.ts +7 -0
- package/lib/browser/context/llm-context.contribution.d.ts.map +1 -0
- package/lib/browser/context/llm-context.contribution.js +21 -0
- package/lib/browser/context/llm-context.contribution.js.map +1 -0
- package/lib/browser/context/llm-context.service.d.ts +24 -0
- package/lib/browser/context/llm-context.service.d.ts.map +1 -0
- package/lib/browser/context/llm-context.service.js +136 -0
- package/lib/browser/context/llm-context.service.js.map +1 -0
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.controller.d.ts +2 -2
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.controller.d.ts.map +1 -1
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.controller.js.map +1 -1
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.feature.registry.d.ts +4 -0
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.feature.registry.d.ts.map +1 -1
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.feature.registry.js +7 -0
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.feature.registry.js.map +1 -1
- package/lib/browser/index.d.ts +11 -3
- package/lib/browser/index.d.ts.map +1 -1
- package/lib/browser/index.js +56 -3
- package/lib/browser/index.js.map +1 -1
- package/lib/browser/mcp/mcp-server-proxy.service.d.ts +25 -0
- package/lib/browser/mcp/mcp-server-proxy.service.d.ts.map +1 -0
- package/lib/browser/mcp/mcp-server-proxy.service.js +56 -0
- package/lib/browser/mcp/mcp-server-proxy.service.js.map +1 -0
- package/lib/browser/mcp/mcp-server.feature.registry.d.ts +16 -0
- package/lib/browser/mcp/mcp-server.feature.registry.d.ts.map +1 -0
- package/lib/browser/mcp/mcp-server.feature.registry.js +53 -0
- package/lib/browser/mcp/mcp-server.feature.registry.js.map +1 -0
- package/lib/browser/mcp/mcp-tools-dialog.module.less +44 -0
- package/lib/browser/mcp/mcp-tools-dialog.view.d.ts +8 -0
- package/lib/browser/mcp/mcp-tools-dialog.view.d.ts.map +1 -0
- package/lib/browser/mcp/mcp-tools-dialog.view.js +16 -0
- package/lib/browser/mcp/mcp-tools-dialog.view.js.map +1 -0
- package/lib/browser/mcp/tools/createNewFileWithText.d.ts +9 -0
- package/lib/browser/mcp/tools/createNewFileWithText.d.ts.map +1 -0
- package/lib/browser/mcp/tools/createNewFileWithText.js +83 -0
- package/lib/browser/mcp/tools/createNewFileWithText.js.map +1 -0
- package/lib/browser/mcp/tools/findFilesByNameSubstring.d.ts +9 -0
- package/lib/browser/mcp/tools/findFilesByNameSubstring.d.ts.map +1 -0
- package/lib/browser/mcp/tools/findFilesByNameSubstring.js +92 -0
- package/lib/browser/mcp/tools/findFilesByNameSubstring.js.map +1 -0
- package/lib/browser/mcp/tools/getCurrentFilePath.d.ts +8 -0
- package/lib/browser/mcp/tools/getCurrentFilePath.d.ts.map +1 -0
- package/lib/browser/mcp/tools/getCurrentFilePath.js +49 -0
- package/lib/browser/mcp/tools/getCurrentFilePath.js.map +1 -0
- package/lib/browser/mcp/tools/getDiagnosticsByPath.d.ts +10 -0
- package/lib/browser/mcp/tools/getDiagnosticsByPath.d.ts.map +1 -0
- package/lib/browser/mcp/tools/getDiagnosticsByPath.js +119 -0
- package/lib/browser/mcp/tools/getDiagnosticsByPath.js.map +1 -0
- package/lib/browser/mcp/tools/getFileTextByPath.d.ts +9 -0
- package/lib/browser/mcp/tools/getFileTextByPath.d.ts.map +1 -0
- package/lib/browser/mcp/tools/getFileTextByPath.js +97 -0
- package/lib/browser/mcp/tools/getFileTextByPath.js.map +1 -0
- package/lib/browser/mcp/tools/getOpenEditorFileDiagnostics.d.ts +11 -0
- package/lib/browser/mcp/tools/getOpenEditorFileDiagnostics.d.ts.map +1 -0
- package/lib/browser/mcp/tools/getOpenEditorFileDiagnostics.js +119 -0
- package/lib/browser/mcp/tools/getOpenEditorFileDiagnostics.js.map +1 -0
- package/lib/browser/mcp/tools/getOpenEditorFileText.d.ts +8 -0
- package/lib/browser/mcp/tools/getOpenEditorFileText.d.ts.map +1 -0
- package/lib/browser/mcp/tools/getOpenEditorFileText.js +50 -0
- package/lib/browser/mcp/tools/getOpenEditorFileText.js.map +1 -0
- package/lib/browser/mcp/tools/getSelectedText.d.ts +8 -0
- package/lib/browser/mcp/tools/getSelectedText.d.ts.map +1 -0
- package/lib/browser/mcp/tools/getSelectedText.js +57 -0
- package/lib/browser/mcp/tools/getSelectedText.js.map +1 -0
- package/lib/browser/mcp/tools/handlers/ListDir.d.ts +21 -0
- package/lib/browser/mcp/tools/handlers/ListDir.d.ts.map +1 -0
- package/lib/browser/mcp/tools/handlers/ListDir.js +112 -0
- package/lib/browser/mcp/tools/handlers/ListDir.js.map +1 -0
- package/lib/browser/mcp/tools/handlers/ReadFile.d.ts +47 -0
- package/lib/browser/mcp/tools/handlers/ReadFile.d.ts.map +1 -0
- package/lib/browser/mcp/tools/handlers/ReadFile.js +147 -0
- package/lib/browser/mcp/tools/handlers/ReadFile.js.map +1 -0
- package/lib/browser/mcp/tools/listDir.d.ts +8 -0
- package/lib/browser/mcp/tools/listDir.d.ts.map +1 -0
- package/lib/browser/mcp/tools/listDir.js +65 -0
- package/lib/browser/mcp/tools/listDir.js.map +1 -0
- package/lib/browser/mcp/tools/readFile.d.ts +8 -0
- package/lib/browser/mcp/tools/readFile.d.ts.map +1 -0
- package/lib/browser/mcp/tools/readFile.js +82 -0
- package/lib/browser/mcp/tools/readFile.js.map +1 -0
- package/lib/browser/mcp/tools/replaceOpenEditorFile.d.ts +8 -0
- package/lib/browser/mcp/tools/replaceOpenEditorFile.d.ts.map +1 -0
- package/lib/browser/mcp/tools/replaceOpenEditorFile.js +79 -0
- package/lib/browser/mcp/tools/replaceOpenEditorFile.js.map +1 -0
- package/lib/browser/mcp/tools/replaceOpenEditorFileByDiffPreviewer.d.ts +8 -0
- package/lib/browser/mcp/tools/replaceOpenEditorFileByDiffPreviewer.d.ts.map +1 -0
- package/lib/browser/mcp/tools/replaceOpenEditorFileByDiffPreviewer.js +84 -0
- package/lib/browser/mcp/tools/replaceOpenEditorFileByDiffPreviewer.js.map +1 -0
- package/lib/browser/mcp/tools/runTerminalCmd.d.ts +18 -0
- package/lib/browser/mcp/tools/runTerminalCmd.d.ts.map +1 -0
- package/lib/browser/mcp/tools/runTerminalCmd.js +96 -0
- package/lib/browser/mcp/tools/runTerminalCmd.js.map +1 -0
- package/lib/browser/preferences/schema.d.ts.map +1 -1
- package/lib/browser/preferences/schema.js +60 -0
- package/lib/browser/preferences/schema.js.map +1 -1
- package/lib/browser/types.d.ts +64 -3
- package/lib/browser/types.d.ts.map +1 -1
- package/lib/browser/types.js +5 -1
- package/lib/browser/types.js.map +1 -1
- package/lib/browser/widget/inline-chat/inline-chat-editor.controller.js +1 -1
- package/lib/browser/widget/inline-chat/inline-chat-editor.controller.js.map +1 -1
- package/lib/browser/widget/inline-diff/inline-diff.controller.d.ts.map +1 -1
- package/lib/browser/widget/inline-input/inline-input.controller.d.ts.map +1 -1
- package/lib/common/index.d.ts +9 -0
- package/lib/common/index.d.ts.map +1 -1
- package/lib/common/index.js +4 -1
- package/lib/common/index.js.map +1 -1
- package/lib/common/llm-context.d.ts +37 -0
- package/lib/common/llm-context.d.ts.map +1 -0
- package/lib/common/llm-context.js +5 -0
- package/lib/common/llm-context.js.map +1 -0
- package/lib/common/mcp-server-manager.d.ts +40 -0
- package/lib/common/mcp-server-manager.d.ts.map +1 -0
- package/lib/common/mcp-server-manager.js +6 -0
- package/lib/common/mcp-server-manager.js.map +1 -0
- package/lib/common/tool-invocation-registry.d.ts +91 -0
- package/lib/common/tool-invocation-registry.d.ts.map +1 -0
- package/lib/common/tool-invocation-registry.js +90 -0
- package/lib/common/tool-invocation-registry.js.map +1 -0
- package/lib/common/types.d.ts +17 -0
- package/lib/common/types.d.ts.map +1 -1
- package/lib/node/anthropic/anthropic-language-model.d.ts +9 -0
- package/lib/node/anthropic/anthropic-language-model.d.ts.map +1 -0
- package/lib/node/anthropic/anthropic-language-model.js +26 -0
- package/lib/node/anthropic/anthropic-language-model.js.map +1 -0
- package/lib/node/base-language-model.d.ts +14 -0
- package/lib/node/base-language-model.d.ts.map +1 -0
- package/lib/node/base-language-model.js +136 -0
- package/lib/node/base-language-model.js.map +1 -0
- package/lib/node/deepseek/deepseek-language-model.d.ts +9 -0
- package/lib/node/deepseek/deepseek-language-model.d.ts.map +1 -0
- package/lib/node/deepseek/deepseek-language-model.js +26 -0
- package/lib/node/deepseek/deepseek-language-model.js.map +1 -0
- package/lib/node/index.d.ts.map +1 -1
- package/lib/node/index.js +19 -0
- package/lib/node/index.js.map +1 -1
- package/lib/node/mcp/sumi-mcp-server.d.ts +91 -0
- package/lib/node/mcp/sumi-mcp-server.d.ts.map +1 -0
- package/lib/node/mcp/sumi-mcp-server.js +172 -0
- package/lib/node/mcp/sumi-mcp-server.js.map +1 -0
- package/lib/node/mcp-server-manager-impl.d.ts +27 -0
- package/lib/node/mcp-server-manager-impl.d.ts.map +1 -0
- package/lib/node/mcp-server-manager-impl.js +127 -0
- package/lib/node/mcp-server-manager-impl.js.map +1 -0
- package/lib/node/mcp-server.d.ts +207 -0
- package/lib/node/mcp-server.d.ts.map +1 -0
- package/lib/node/mcp-server.js +91 -0
- package/lib/node/mcp-server.js.map +1 -0
- package/lib/node/openai/openai-language-model.d.ts +9 -0
- package/lib/node/openai/openai-language-model.d.ts.map +1 -0
- package/lib/node/openai/openai-language-model.js +29 -0
- package/lib/node/openai/openai-language-model.js.map +1 -0
- package/package.json +34 -22
- package/src/browser/ai-core.contribution.ts +77 -1
- package/src/browser/chat/chat-model.ts +24 -6
- package/src/browser/chat/chat-proxy.service.ts +42 -0
- package/src/browser/chat/chat.view.tsx +59 -6
- package/src/browser/components/ChatContext/ContextSelector.tsx +177 -0
- package/src/browser/components/ChatContext/index.tsx +135 -0
- package/src/browser/components/ChatContext/style.module.less +189 -0
- package/src/browser/components/ChatInput.tsx +1 -0
- package/src/browser/components/ChatReply.tsx +32 -0
- package/src/browser/components/ChatToolRender.module.less +86 -0
- package/src/browser/components/ChatToolRender.tsx +77 -0
- package/src/browser/components/components.module.less +32 -31
- package/src/browser/context/llm-context.contribution.ts +14 -0
- package/src/browser/context/llm-context.service.ts +156 -0
- package/src/browser/contrib/intelligent-completions/intelligent-completions.controller.ts +2 -3
- package/src/browser/contrib/intelligent-completions/intelligent-completions.feature.registry.ts +11 -0
- package/src/browser/index.ts +68 -4
- package/src/browser/mcp/mcp-server-proxy.service.ts +53 -0
- package/src/browser/mcp/mcp-server.feature.registry.ts +54 -0
- package/src/browser/mcp/mcp-tools-dialog.module.less +44 -0
- package/src/browser/mcp/mcp-tools-dialog.view.tsx +24 -0
- package/src/browser/mcp/tools/createNewFileWithText.ts +83 -0
- package/src/browser/mcp/tools/findFilesByNameSubstring.ts +93 -0
- package/src/browser/mcp/tools/getCurrentFilePath.ts +49 -0
- package/src/browser/mcp/tools/getDiagnosticsByPath.ts +123 -0
- package/src/browser/mcp/tools/getFileTextByPath.ts +97 -0
- package/src/browser/mcp/tools/getOpenEditorFileDiagnostics.ts +121 -0
- package/src/browser/mcp/tools/getOpenEditorFileText.ts +50 -0
- package/src/browser/mcp/tools/getSelectedText.ts +57 -0
- package/src/browser/mcp/tools/handlers/ListDir.ts +117 -0
- package/src/browser/mcp/tools/handlers/ReadFile.ts +174 -0
- package/src/browser/mcp/tools/listDir.ts +66 -0
- package/src/browser/mcp/tools/readFile.ts +82 -0
- package/src/browser/mcp/tools/replaceOpenEditorFile.ts +80 -0
- package/src/browser/mcp/tools/replaceOpenEditorFileByDiffPreviewer.ts +91 -0
- package/src/browser/mcp/tools/runTerminalCmd.ts +107 -0
- package/src/browser/preferences/schema.ts +60 -0
- package/src/browser/types.ts +92 -3
- package/src/browser/widget/inline-chat/inline-chat-editor.controller.ts +1 -1
- package/src/browser/widget/inline-diff/inline-diff.controller.ts +1 -1
- package/src/browser/widget/inline-input/inline-input.controller.ts +1 -1
- package/src/common/index.ts +14 -0
- package/src/common/llm-context.ts +41 -0
- package/src/common/mcp-server-manager.ts +46 -0
- package/src/common/tool-invocation-registry.ts +170 -0
- package/src/common/types.ts +22 -0
- package/src/node/anthropic/anthropic-language-model.ts +25 -0
- package/src/node/base-language-model.ts +163 -0
- package/src/node/deepseek/deepseek-language-model.ts +25 -0
- package/src/node/index.ts +21 -0
- package/src/node/mcp/sumi-mcp-server.ts +197 -0
- package/src/node/mcp-server-manager-impl.ts +148 -0
- package/src/node/mcp-server.ts +126 -0
- package/src/node/openai/openai-language-model.ts +25 -0
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.AnthropicModel = exports.AnthropicModelIdentifier = void 0;
|
|
4
|
+
const tslib_1 = require("tslib");
|
|
5
|
+
const anthropic_1 = require("@ai-sdk/anthropic");
|
|
6
|
+
const di_1 = require("@opensumi/di");
|
|
7
|
+
const ai_native_1 = require("@opensumi/ide-core-common/lib/settings/ai-native");
|
|
8
|
+
const base_language_model_1 = require("../base-language-model");
|
|
9
|
+
exports.AnthropicModelIdentifier = Symbol('AnthropicModelIdentifier');
|
|
10
|
+
let AnthropicModel = class AnthropicModel extends base_language_model_1.BaseLanguageModel {
|
|
11
|
+
initializeProvider(options) {
|
|
12
|
+
const apiKey = options.apiKey;
|
|
13
|
+
if (!apiKey) {
|
|
14
|
+
throw new Error(`Please provide Anthropic API Key in preferences (${ai_native_1.AINativeSettingSectionsId.AnthropicApiKey})`);
|
|
15
|
+
}
|
|
16
|
+
return (0, anthropic_1.createAnthropic)({ apiKey });
|
|
17
|
+
}
|
|
18
|
+
getModelIdentifier(provider) {
|
|
19
|
+
return provider('claude-3-5-sonnet-20241022');
|
|
20
|
+
}
|
|
21
|
+
};
|
|
22
|
+
exports.AnthropicModel = AnthropicModel;
|
|
23
|
+
exports.AnthropicModel = AnthropicModel = tslib_1.__decorate([
|
|
24
|
+
(0, di_1.Injectable)()
|
|
25
|
+
], AnthropicModel);
|
|
26
|
+
//# sourceMappingURL=anthropic-language-model.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic-language-model.js","sourceRoot":"","sources":["../../../src/node/anthropic/anthropic-language-model.ts"],"names":[],"mappings":";;;;AAAA,iDAAuE;AAEvE,qCAA0C;AAE1C,gFAA6F;AAE7F,gEAA2D;AAE9C,QAAA,wBAAwB,GAAG,MAAM,CAAC,0BAA0B,CAAC,CAAC;AAGpE,IAAM,cAAc,GAApB,MAAM,cAAe,SAAQ,uCAAiB;IACzC,kBAAkB,CAAC,OAA6B;QACxD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC9B,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oDAAoD,qCAAyB,CAAC,eAAe,GAAG,CAAC,CAAC;QACpH,CAAC;QAED,OAAO,IAAA,2BAAe,EAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IACrC,CAAC;IAES,kBAAkB,CAAC,QAA2B;QACtD,OAAO,QAAQ,CAAC,4BAA4B,CAAC,CAAC;IAChD,CAAC;CACF,CAAA;AAbY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,eAAU,GAAE;GACA,cAAc,CAa1B"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import { IAIBackServiceOption, IChatMessage } from '@opensumi/ide-core-common';
|
|
2
|
+
import { ChatReadableStream } from '@opensumi/ide-core-node';
|
|
3
|
+
import { CancellationToken } from '@opensumi/ide-utils';
|
|
4
|
+
import { IToolInvocationRegistryManager, ToolRequest } from '../common/tool-invocation-registry';
|
|
5
|
+
export declare abstract class BaseLanguageModel {
|
|
6
|
+
protected readonly toolInvocationRegistryManager: IToolInvocationRegistryManager;
|
|
7
|
+
protected abstract initializeProvider(options: IAIBackServiceOption): any;
|
|
8
|
+
private convertChatMessageRole;
|
|
9
|
+
request(request: string, chatReadableStream: ChatReadableStream, options: IAIBackServiceOption, cancellationToken?: CancellationToken): Promise<any>;
|
|
10
|
+
private convertToolRequestToAITool;
|
|
11
|
+
protected abstract getModelIdentifier(provider: any): any;
|
|
12
|
+
protected handleStreamingRequest(provider: any, request: string, tools: ToolRequest[], chatReadableStream: ChatReadableStream, history?: IChatMessage[], cancellationToken?: CancellationToken): Promise<any>;
|
|
13
|
+
}
|
|
14
|
+
//# sourceMappingURL=base-language-model.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base-language-model.d.ts","sourceRoot":"","sources":["../../src/node/base-language-model.ts"],"names":[],"mappings":"AAGA,OAAO,EAAmB,oBAAoB,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAChG,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAExD,OAAO,EACL,8BAA8B,EAE9B,WAAW,EACZ,MAAM,oCAAoC,CAAC;AAE5C,8BACsB,iBAAiB;IAErC,SAAS,CAAC,QAAQ,CAAC,6BAA6B,EAAE,8BAA8B,CAAC;IAEjF,SAAS,CAAC,QAAQ,CAAC,kBAAkB,CAAC,OAAO,EAAE,oBAAoB,GAAG,GAAG;IAEzE,OAAO,CAAC,sBAAsB;IAexB,OAAO,CACX,OAAO,EAAE,MAAM,EACf,kBAAkB,EAAE,kBAAkB,EACtC,OAAO,EAAE,oBAAoB,EAC7B,iBAAiB,CAAC,EAAE,iBAAiB,GACpC,OAAO,CAAC,GAAG,CAAC;IAkBf,OAAO,CAAC,0BAA0B;IASlC,SAAS,CAAC,QAAQ,CAAC,kBAAkB,CAAC,QAAQ,EAAE,GAAG,GAAG,GAAG;cAEzC,sBAAsB,CACpC,QAAQ,EAAE,GAAG,EACb,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,WAAW,EAAE,EACpB,kBAAkB,EAAE,kBAAkB,EACtC,OAAO,GAAE,YAAY,EAAO,EAC5B,iBAAiB,CAAC,EAAE,iBAAiB,GACpC,OAAO,CAAC,GAAG,CAAC;CAsFhB"}
|
|
@@ -0,0 +1,136 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.BaseLanguageModel = void 0;
|
|
4
|
+
const tslib_1 = require("tslib");
|
|
5
|
+
const ai_1 = require("ai");
|
|
6
|
+
const di_1 = require("@opensumi/di");
|
|
7
|
+
const tool_invocation_registry_1 = require("../common/tool-invocation-registry");
|
|
8
|
+
let BaseLanguageModel = class BaseLanguageModel {
|
|
9
|
+
convertChatMessageRole(role) {
|
|
10
|
+
switch (role) {
|
|
11
|
+
case 0 /* ChatMessageRole.System */:
|
|
12
|
+
return 'system';
|
|
13
|
+
case 1 /* ChatMessageRole.User */:
|
|
14
|
+
return 'user';
|
|
15
|
+
case 2 /* ChatMessageRole.Assistant */:
|
|
16
|
+
return 'assistant';
|
|
17
|
+
case 3 /* ChatMessageRole.Function */:
|
|
18
|
+
return 'tool';
|
|
19
|
+
default:
|
|
20
|
+
return 'user';
|
|
21
|
+
}
|
|
22
|
+
}
|
|
23
|
+
async request(request, chatReadableStream, options, cancellationToken) {
|
|
24
|
+
const provider = this.initializeProvider(options);
|
|
25
|
+
const clientId = options.clientId;
|
|
26
|
+
if (!clientId) {
|
|
27
|
+
throw new Error('clientId is required');
|
|
28
|
+
}
|
|
29
|
+
const registry = this.toolInvocationRegistryManager.getRegistry(clientId);
|
|
30
|
+
const allFunctions = registry.getAllFunctions();
|
|
31
|
+
return this.handleStreamingRequest(provider, request, allFunctions, chatReadableStream, options.history || [], cancellationToken);
|
|
32
|
+
}
|
|
33
|
+
convertToolRequestToAITool(toolRequest) {
|
|
34
|
+
return (0, ai_1.tool)({
|
|
35
|
+
description: toolRequest.description || '',
|
|
36
|
+
// TODO 这里应该是 z.object 而不是 JSON Schema
|
|
37
|
+
parameters: (0, ai_1.jsonSchema)(toolRequest.parameters),
|
|
38
|
+
execute: async (args) => await toolRequest.handler(JSON.stringify(args)),
|
|
39
|
+
});
|
|
40
|
+
}
|
|
41
|
+
async handleStreamingRequest(provider, request, tools, chatReadableStream, history = [], cancellationToken) {
|
|
42
|
+
try {
|
|
43
|
+
const aiTools = Object.fromEntries(tools.map((tool) => [tool.name, this.convertToolRequestToAITool(tool)]));
|
|
44
|
+
const abortController = new AbortController();
|
|
45
|
+
if (cancellationToken) {
|
|
46
|
+
cancellationToken.onCancellationRequested(() => {
|
|
47
|
+
abortController.abort();
|
|
48
|
+
});
|
|
49
|
+
}
|
|
50
|
+
const messages = [
|
|
51
|
+
...history.map((msg) => ({
|
|
52
|
+
role: this.convertChatMessageRole(msg.role), // 这个 SDK 包里的类型不太好完全对应,
|
|
53
|
+
content: msg.content,
|
|
54
|
+
})),
|
|
55
|
+
{ role: 'user', content: request },
|
|
56
|
+
];
|
|
57
|
+
const stream = await (0, ai_1.streamText)({
|
|
58
|
+
model: this.getModelIdentifier(provider),
|
|
59
|
+
maxTokens: 4096,
|
|
60
|
+
tools: aiTools,
|
|
61
|
+
messages,
|
|
62
|
+
abortSignal: abortController.signal,
|
|
63
|
+
experimental_toolCallStreaming: true,
|
|
64
|
+
maxSteps: 12,
|
|
65
|
+
});
|
|
66
|
+
for await (const chunk of stream.fullStream) {
|
|
67
|
+
if (chunk.type === 'text-delta') {
|
|
68
|
+
chatReadableStream.emitData({ kind: 'content', content: chunk.textDelta });
|
|
69
|
+
}
|
|
70
|
+
else if (chunk.type === 'tool-call') {
|
|
71
|
+
chatReadableStream.emitData({
|
|
72
|
+
kind: 'toolCall',
|
|
73
|
+
content: {
|
|
74
|
+
id: chunk.toolCallId || Date.now().toString(),
|
|
75
|
+
type: 'function',
|
|
76
|
+
function: { name: chunk.toolName, arguments: JSON.stringify(chunk.args) },
|
|
77
|
+
state: 'complete',
|
|
78
|
+
},
|
|
79
|
+
});
|
|
80
|
+
}
|
|
81
|
+
else if (chunk.type === 'tool-call-streaming-start') {
|
|
82
|
+
chatReadableStream.emitData({
|
|
83
|
+
kind: 'toolCall',
|
|
84
|
+
content: {
|
|
85
|
+
id: chunk.toolCallId,
|
|
86
|
+
type: 'function',
|
|
87
|
+
function: { name: chunk.toolName },
|
|
88
|
+
state: 'streaming-start',
|
|
89
|
+
},
|
|
90
|
+
});
|
|
91
|
+
}
|
|
92
|
+
else if (chunk.type === 'tool-call-delta') {
|
|
93
|
+
chatReadableStream.emitData({
|
|
94
|
+
kind: 'toolCall',
|
|
95
|
+
content: {
|
|
96
|
+
id: chunk.toolCallId,
|
|
97
|
+
type: 'function',
|
|
98
|
+
function: { name: chunk.toolName, arguments: chunk.argsTextDelta },
|
|
99
|
+
state: 'streaming',
|
|
100
|
+
},
|
|
101
|
+
});
|
|
102
|
+
}
|
|
103
|
+
else if (chunk.type === 'tool-result') {
|
|
104
|
+
chatReadableStream.emitData({
|
|
105
|
+
kind: 'toolCall',
|
|
106
|
+
content: {
|
|
107
|
+
id: chunk.toolCallId,
|
|
108
|
+
type: 'function',
|
|
109
|
+
function: { name: chunk.toolName, arguments: JSON.stringify(chunk.args) },
|
|
110
|
+
result: chunk.result,
|
|
111
|
+
state: 'result',
|
|
112
|
+
},
|
|
113
|
+
});
|
|
114
|
+
}
|
|
115
|
+
else if (chunk.type === 'error') {
|
|
116
|
+
chatReadableStream.emitError(new Error(chunk.error));
|
|
117
|
+
}
|
|
118
|
+
}
|
|
119
|
+
chatReadableStream.end();
|
|
120
|
+
}
|
|
121
|
+
catch (error) {
|
|
122
|
+
// Use a logger service in production instead of console
|
|
123
|
+
chatReadableStream.emitError(error);
|
|
124
|
+
}
|
|
125
|
+
return chatReadableStream;
|
|
126
|
+
}
|
|
127
|
+
};
|
|
128
|
+
exports.BaseLanguageModel = BaseLanguageModel;
|
|
129
|
+
tslib_1.__decorate([
|
|
130
|
+
(0, di_1.Autowired)(tool_invocation_registry_1.ToolInvocationRegistryManager),
|
|
131
|
+
tslib_1.__metadata("design:type", Object)
|
|
132
|
+
], BaseLanguageModel.prototype, "toolInvocationRegistryManager", void 0);
|
|
133
|
+
exports.BaseLanguageModel = BaseLanguageModel = tslib_1.__decorate([
|
|
134
|
+
(0, di_1.Injectable)()
|
|
135
|
+
], BaseLanguageModel);
|
|
136
|
+
//# sourceMappingURL=base-language-model.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base-language-model.js","sourceRoot":"","sources":["../../src/node/base-language-model.ts"],"names":[],"mappings":";;;;AAAA,2BAA+D;AAE/D,qCAAqD;AAKrD,iFAI4C;AAGrC,IAAe,iBAAiB,GAAhC,MAAe,iBAAiB;IAM7B,sBAAsB,CAAC,IAAqB;QAClD,QAAQ,IAAI,EAAE,CAAC;YACb;gBACE,OAAO,QAAQ,CAAC;YAClB;gBACE,OAAO,MAAM,CAAC;YAChB;gBACE,OAAO,WAAW,CAAC;YACrB;gBACE,OAAO,MAAM,CAAC;YAChB;gBACE,OAAO,MAAM,CAAC;QAClB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CACX,OAAe,EACf,kBAAsC,EACtC,OAA6B,EAC7B,iBAAqC;QAErC,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAClC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,6BAA6B,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC1E,MAAM,YAAY,GAAG,QAAQ,CAAC,eAAe,EAAE,CAAC;QAChD,OAAO,IAAI,CAAC,sBAAsB,CAChC,QAAQ,EACR,OAAO,EACP,YAAY,EACZ,kBAAkB,EAClB,OAAO,CAAC,OAAO,IAAI,EAAE,EACrB,iBAAiB,CAClB,CAAC;IACJ,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;IAIS,KAAK,CAAC,sBAAsB,CACpC,QAAa,EACb,OAAe,EACf,KAAoB,EACpB,kBAAsC,EACtC,UAA0B,EAAE,EAC5B,iBAAqC;QAErC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAE5G,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,QAAQ,GAAkB;gBAC9B,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBACvB,IAAI,EAAE,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,IAAI,CAAQ,EAAE,uBAAuB;oBAC3E,OAAO,EAAE,GAAG,CAAC,OAAO;iBACrB,CAAC,CAAC;gBACH,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE;aACnC,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAA,eAAU,EAAC;gBAC9B,KAAK,EAAE,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC;gBACxC,SAAS,EAAE,IAAI;gBACf,KAAK,EAAE,OAAO;gBACd,QAAQ;gBACR,WAAW,EAAE,eAAe,CAAC,MAAM;gBACnC,8BAA8B,EAAE,IAAI;gBACpC,QAAQ,EAAE,EAAE;aACb,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;gBAC5C,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;wBAC1B,IAAI,EAAE,UAAU;wBAChB,OAAO,EAAE;4BACP,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;4BACzE,KAAK,EAAE,UAAU;yBAClB;qBACF,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,2BAA2B,EAAE,CAAC;oBACtD,kBAAkB,CAAC,QAAQ,CAAC;wBAC1B,IAAI,EAAE,UAAU;wBAChB,OAAO,EAAE;4BACP,EAAE,EAAE,KAAK,CAAC,UAAU;4BACpB,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE;4BAClC,KAAK,EAAE,iBAAiB;yBACzB;qBACF,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;oBAC5C,kBAAkB,CAAC,QAAQ,CAAC;wBAC1B,IAAI,EAAE,UAAU;wBAChB,OAAO,EAAE;4BACP,EAAE,EAAE,KAAK,CAAC,UAAU;4BACpB,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,SAAS,EAAE,KAAK,CAAC,aAAa,EAAE;4BAClE,KAAK,EAAE,WAAW;yBACnB;qBACF,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;oBACxC,kBAAkB,CAAC,QAAQ,CAAC;wBAC1B,IAAI,EAAE,UAAU;wBAChB,OAAO,EAAE;4BACP,EAAE,EAAE,KAAK,CAAC,UAAU;4BACpB,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;4BACzE,MAAM,EAAE,KAAK,CAAC,MAAM;4BACpB,KAAK,EAAE,QAAQ;yBAChB;qBACF,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBAClC,kBAAkB,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,KAAe,CAAC,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;YAED,kBAAkB,CAAC,GAAG,EAAE,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,wDAAwD;YACxD,kBAAkB,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACtC,CAAC;QAED,OAAO,kBAAkB,CAAC;IAC5B,CAAC;CACF,CAAA;AApJqB,8CAAiB;AAElB;IADlB,IAAA,cAAS,EAAC,wDAA6B,CAAC;;wEACwC;4BAF7D,iBAAiB;IADtC,IAAA,eAAU,GAAE;GACS,iBAAiB,CAoJtC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { DeepSeekProvider } from '@ai-sdk/deepseek';
|
|
2
|
+
import { IAIBackServiceOption } from '@opensumi/ide-core-common';
|
|
3
|
+
import { BaseLanguageModel } from '../base-language-model';
|
|
4
|
+
export declare const DeepSeekModelIdentifier: unique symbol;
|
|
5
|
+
export declare class DeepSeekModel extends BaseLanguageModel {
|
|
6
|
+
protected initializeProvider(options: IAIBackServiceOption): DeepSeekProvider;
|
|
7
|
+
protected getModelIdentifier(provider: DeepSeekProvider): import("ai").LanguageModelV1;
|
|
8
|
+
}
|
|
9
|
+
//# sourceMappingURL=deepseek-language-model.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"deepseek-language-model.d.ts","sourceRoot":"","sources":["../../../src/node/deepseek/deepseek-language-model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAkB,MAAM,kBAAkB,CAAC;AAGpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAGjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAE3D,eAAO,MAAM,uBAAuB,eAAoC,CAAC;AAEzE,qBACa,aAAc,SAAQ,iBAAiB;IAClD,SAAS,CAAC,kBAAkB,CAAC,OAAO,EAAE,oBAAoB,GAAG,gBAAgB;IAS7E,SAAS,CAAC,kBAAkB,CAAC,QAAQ,EAAE,gBAAgB;CAGxD"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.DeepSeekModel = exports.DeepSeekModelIdentifier = void 0;
|
|
4
|
+
const tslib_1 = require("tslib");
|
|
5
|
+
const deepseek_1 = require("@ai-sdk/deepseek");
|
|
6
|
+
const di_1 = require("@opensumi/di");
|
|
7
|
+
const ai_native_1 = require("@opensumi/ide-core-common/lib/settings/ai-native");
|
|
8
|
+
const base_language_model_1 = require("../base-language-model");
|
|
9
|
+
exports.DeepSeekModelIdentifier = Symbol('DeepSeekModelIdentifier');
|
|
10
|
+
let DeepSeekModel = class DeepSeekModel extends base_language_model_1.BaseLanguageModel {
|
|
11
|
+
initializeProvider(options) {
|
|
12
|
+
const apiKey = options.apiKey;
|
|
13
|
+
if (!apiKey) {
|
|
14
|
+
throw new Error(`Please provide Deepseek API Key in preferences (${ai_native_1.AINativeSettingSectionsId.DeepseekApiKey})`);
|
|
15
|
+
}
|
|
16
|
+
return (0, deepseek_1.createDeepSeek)({ apiKey });
|
|
17
|
+
}
|
|
18
|
+
getModelIdentifier(provider) {
|
|
19
|
+
return provider('deepseek-chat');
|
|
20
|
+
}
|
|
21
|
+
};
|
|
22
|
+
exports.DeepSeekModel = DeepSeekModel;
|
|
23
|
+
exports.DeepSeekModel = DeepSeekModel = tslib_1.__decorate([
|
|
24
|
+
(0, di_1.Injectable)()
|
|
25
|
+
], DeepSeekModel);
|
|
26
|
+
//# sourceMappingURL=deepseek-language-model.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"deepseek-language-model.js","sourceRoot":"","sources":["../../../src/node/deepseek/deepseek-language-model.ts"],"names":[],"mappings":";;;;AAAA,+CAAoE;AAEpE,qCAA0C;AAE1C,gFAA6F;AAE7F,gEAA2D;AAE9C,QAAA,uBAAuB,GAAG,MAAM,CAAC,yBAAyB,CAAC,CAAC;AAGlE,IAAM,aAAa,GAAnB,MAAM,aAAc,SAAQ,uCAAiB;IACxC,kBAAkB,CAAC,OAA6B;QACxD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC9B,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,mDAAmD,qCAAyB,CAAC,cAAc,GAAG,CAAC,CAAC;QAClH,CAAC;QAED,OAAO,IAAA,yBAAc,EAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IACpC,CAAC;IAES,kBAAkB,CAAC,QAA0B;QACrD,OAAO,QAAQ,CAAC,eAAe,CAAC,CAAC;IACnC,CAAC;CACF,CAAA;AAbY,sCAAa;wBAAb,aAAa;IADzB,IAAA,eAAU,GAAE;GACA,aAAa,CAazB"}
|
package/lib/node/index.d.ts.map
CHANGED
|
@@ -1 +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;
|
|
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;AAQrD,qBACa,cAAe,SAAQ,UAAU;IAC5C,SAAS,EAAE,QAAQ,EAAE,CAanB;IAEF,YAAY;;;QAaV;CACH"}
|
package/lib/node/index.js
CHANGED
|
@@ -6,6 +6,9 @@ const di_1 = require("@opensumi/di");
|
|
|
6
6
|
const ide_core_common_1 = require("@opensumi/ide-core-common");
|
|
7
7
|
const ide_core_node_1 = require("@opensumi/ide-core-node");
|
|
8
8
|
const base_back_service_1 = require("@opensumi/ide-core-node/lib/ai-native/base-back.service");
|
|
9
|
+
const common_1 = require("../common");
|
|
10
|
+
const tool_invocation_registry_1 = require("../common/tool-invocation-registry");
|
|
11
|
+
const sumi_mcp_server_1 = require("./mcp/sumi-mcp-server");
|
|
9
12
|
let AINativeModule = class AINativeModule extends ide_core_node_1.NodeModule {
|
|
10
13
|
constructor() {
|
|
11
14
|
super(...arguments);
|
|
@@ -14,12 +17,28 @@ let AINativeModule = class AINativeModule extends ide_core_node_1.NodeModule {
|
|
|
14
17
|
token: ide_core_common_1.AIBackSerivceToken,
|
|
15
18
|
useClass: base_back_service_1.BaseAIBackService,
|
|
16
19
|
},
|
|
20
|
+
{
|
|
21
|
+
token: tool_invocation_registry_1.ToolInvocationRegistryManager,
|
|
22
|
+
useClass: tool_invocation_registry_1.ToolInvocationRegistryManagerImpl,
|
|
23
|
+
},
|
|
24
|
+
{
|
|
25
|
+
token: common_1.TokenMCPServerProxyService,
|
|
26
|
+
useClass: sumi_mcp_server_1.SumiMCPServerBackend,
|
|
27
|
+
},
|
|
17
28
|
];
|
|
18
29
|
this.backServices = [
|
|
19
30
|
{
|
|
20
31
|
servicePath: ide_core_common_1.AIBackSerivcePath,
|
|
21
32
|
token: ide_core_common_1.AIBackSerivceToken,
|
|
22
33
|
},
|
|
34
|
+
// {
|
|
35
|
+
// servicePath: MCPServerManagerPath,
|
|
36
|
+
// token: MCPServerManager,
|
|
37
|
+
// },
|
|
38
|
+
{
|
|
39
|
+
servicePath: common_1.SumiMCPServerProxyServicePath,
|
|
40
|
+
token: common_1.TokenMCPServerProxyService,
|
|
41
|
+
},
|
|
23
42
|
];
|
|
24
43
|
}
|
|
25
44
|
};
|
package/lib/node/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/node/index.ts"],"names":[],"mappings":";;;;AAAA,qCAAoD;AACpD,+DAAkF;AAClF,2DAAqD;AACrD,+FAA4F;
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/node/index.ts"],"names":[],"mappings":";;;;AAAA,qCAAoD;AACpD,+DAAkF;AAClF,2DAAqD;AACrD,+FAA4F;AAE5F,sCAAsF;AACtF,iFAAsH;AAEtH,2DAA6D;AAGtD,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,wDAA6B;gBACpC,QAAQ,EAAE,4DAAiC;aAC5C;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,IAAI;YACJ,uCAAuC;YACvC,6BAA6B;YAC7B,KAAK;YACL;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"}
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
|
|
2
|
+
import { Server } from '@modelcontextprotocol/sdk/server/index.js';
|
|
3
|
+
import { RPCService } from '@opensumi/ide-connection';
|
|
4
|
+
import { ILogger } from '@opensumi/ide-core-common';
|
|
5
|
+
import { ISumiMCPServerBackend } from '../../common';
|
|
6
|
+
import { MCPServerDescription } from '../../common/mcp-server-manager';
|
|
7
|
+
import { IMCPServerProxyService, MCPTool } from '../../common/types';
|
|
8
|
+
import { IMCPServer } from '../mcp-server';
|
|
9
|
+
export declare class SumiMCPServerBackend extends RPCService<IMCPServerProxyService> implements ISumiMCPServerBackend {
|
|
10
|
+
private readonly mcpServerManager;
|
|
11
|
+
private readonly toolInvocationRegistryManager;
|
|
12
|
+
private readonly logger;
|
|
13
|
+
private server;
|
|
14
|
+
private clientId;
|
|
15
|
+
constructor();
|
|
16
|
+
setConnectionClientId(clientId: string): void;
|
|
17
|
+
getMCPTools(): Promise<MCPTool[]>;
|
|
18
|
+
callMCPTool(name: string, args: any): Promise<{
|
|
19
|
+
content: {
|
|
20
|
+
type: string;
|
|
21
|
+
text: string;
|
|
22
|
+
}[];
|
|
23
|
+
isError?: boolean;
|
|
24
|
+
}>;
|
|
25
|
+
getServer(): Server<{
|
|
26
|
+
method: string;
|
|
27
|
+
params?: {
|
|
28
|
+
[x: string]: unknown;
|
|
29
|
+
_meta?: {
|
|
30
|
+
[x: string]: unknown;
|
|
31
|
+
progressToken?: string | number | undefined;
|
|
32
|
+
} | undefined;
|
|
33
|
+
} | undefined;
|
|
34
|
+
}, {
|
|
35
|
+
method: string;
|
|
36
|
+
params?: {
|
|
37
|
+
[x: string]: unknown;
|
|
38
|
+
_meta?: {
|
|
39
|
+
[x: string]: unknown;
|
|
40
|
+
} | undefined;
|
|
41
|
+
} | undefined;
|
|
42
|
+
}, {
|
|
43
|
+
[x: string]: unknown;
|
|
44
|
+
_meta?: {
|
|
45
|
+
[x: string]: unknown;
|
|
46
|
+
} | undefined;
|
|
47
|
+
}> | undefined;
|
|
48
|
+
getAllMCPTools(): Promise<MCPTool[]>;
|
|
49
|
+
initBuiltinMCPServer(): Promise<void>;
|
|
50
|
+
initExternalMCPServers(servers: MCPServerDescription[]): Promise<void>;
|
|
51
|
+
initExposedMCPServer(): Promise<Server<{
|
|
52
|
+
method: string;
|
|
53
|
+
params?: {
|
|
54
|
+
[x: string]: unknown;
|
|
55
|
+
_meta?: {
|
|
56
|
+
[x: string]: unknown;
|
|
57
|
+
progressToken?: string | number | undefined;
|
|
58
|
+
} | undefined;
|
|
59
|
+
} | undefined;
|
|
60
|
+
}, {
|
|
61
|
+
method: string;
|
|
62
|
+
params?: {
|
|
63
|
+
[x: string]: unknown;
|
|
64
|
+
_meta?: {
|
|
65
|
+
[x: string]: unknown;
|
|
66
|
+
} | undefined;
|
|
67
|
+
} | undefined;
|
|
68
|
+
}, {
|
|
69
|
+
[x: string]: unknown;
|
|
70
|
+
_meta?: {
|
|
71
|
+
[x: string]: unknown;
|
|
72
|
+
} | undefined;
|
|
73
|
+
}>>;
|
|
74
|
+
}
|
|
75
|
+
export declare const TokenBuiltinMCPServer: unique symbol;
|
|
76
|
+
export declare class BuiltinMCPServer implements IMCPServer {
|
|
77
|
+
private readonly sumiMCPServer;
|
|
78
|
+
private readonly logger;
|
|
79
|
+
private started;
|
|
80
|
+
constructor(sumiMCPServer: SumiMCPServerBackend, logger: ILogger);
|
|
81
|
+
isStarted(): boolean;
|
|
82
|
+
getServerName(): string;
|
|
83
|
+
start(): Promise<void>;
|
|
84
|
+
callTool(toolName: string, arg_string: string): Promise<any>;
|
|
85
|
+
getTools(): ReturnType<Client['listTools']>;
|
|
86
|
+
update(_command: string, _args?: string[], _env?: {
|
|
87
|
+
[key: string]: string;
|
|
88
|
+
}): void;
|
|
89
|
+
stop(): void;
|
|
90
|
+
}
|
|
91
|
+
//# sourceMappingURL=sumi-mcp-server.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
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;AACtD,OAAO,EAAE,OAAO,EAAE,MAAM,2BAA2B,CAAC;AAGpD,OAAO,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAC;AACrD,OAAO,EAAE,oBAAoB,EAAoB,MAAM,iCAAiC,CAAC;AAEzF,OAAO,EAAE,sBAAsB,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AACrE,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAQ3C,qBACa,oBAAqB,SAAQ,UAAU,CAAC,sBAAsB,CAAE,YAAW,qBAAqB;IAG3G,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAuB;IAGxD,OAAO,CAAC,QAAQ,CAAC,6BAA6B,CAAiC;IAG/E,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAU;IAEjC,OAAO,CAAC,MAAM,CAAqB;IAGnC,OAAO,CAAC,QAAQ,CAAc;;IAOvB,qBAAqB,CAAC,QAAQ,EAAE,MAAM;IAKvC,WAAW;IAUX,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG;;;;;;;IAOzC,SAAS;;;;;;;;;;;;;;;;;;;;;;;IAKH,cAAc,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAU7B,oBAAoB;IAOpB,sBAAsB,CAAC,OAAO,EAAE,oBAAoB,EAAE;IAM7D,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;CAoC3B;AAED,eAAO,MAAM,qBAAqB,eAAkC,CAAC;AAErE,qBAAa,gBAAiB,YAAW,UAAU;IAGrC,OAAO,CAAC,QAAQ,CAAC,aAAa;IAAwB,OAAO,CAAC,QAAQ,CAAC,MAAM;IAFzF,OAAO,CAAC,OAAO,CAAiB;gBAEH,aAAa,EAAE,oBAAoB,EAAmB,MAAM,EAAE,OAAO;IAElG,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;IASjD,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"}
|
|
@@ -0,0 +1,172 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
// 想要通过 MCP 的方式暴露 Opensumi 的 IDE 能力,就需要 Node.js 层打通 MCP 的通信
|
|
3
|
+
// 因为大部分 MCP 功能的实现在前端,因此需要再这里做前后端通信
|
|
4
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
5
|
+
exports.BuiltinMCPServer = exports.TokenBuiltinMCPServer = exports.SumiMCPServerBackend = void 0;
|
|
6
|
+
const tslib_1 = require("tslib");
|
|
7
|
+
const index_js_1 = require("@modelcontextprotocol/sdk/server/index.js");
|
|
8
|
+
const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
|
|
9
|
+
const di_1 = require("@opensumi/di");
|
|
10
|
+
const ide_connection_1 = require("@opensumi/ide-connection");
|
|
11
|
+
const ide_core_common_1 = require("@opensumi/ide-core-common");
|
|
12
|
+
const ide_core_node_1 = require("@opensumi/ide-core-node");
|
|
13
|
+
const tool_invocation_registry_1 = require("../../common/tool-invocation-registry");
|
|
14
|
+
const mcp_server_manager_impl_1 = require("../mcp-server-manager-impl");
|
|
15
|
+
// 每个 BrowserTab 都对应了一个 SumiMCPServerBackend 实例
|
|
16
|
+
// SumiMCPServerBackend 需要做的事情:
|
|
17
|
+
// 维护 Browser 端工具的注册和调用
|
|
18
|
+
// 处理第三方 MCP Server 的注册和调用
|
|
19
|
+
let SumiMCPServerBackend = class SumiMCPServerBackend extends ide_connection_1.RPCService {
|
|
20
|
+
constructor() {
|
|
21
|
+
super();
|
|
22
|
+
// 对应 BrowserTab 的 clientId
|
|
23
|
+
this.clientId = '';
|
|
24
|
+
this.mcpServerManager = new mcp_server_manager_impl_1.MCPServerManagerImpl(this.toolInvocationRegistryManager, this.logger);
|
|
25
|
+
}
|
|
26
|
+
setConnectionClientId(clientId) {
|
|
27
|
+
this.clientId = clientId;
|
|
28
|
+
this.mcpServerManager.setClientId(clientId);
|
|
29
|
+
}
|
|
30
|
+
async getMCPTools() {
|
|
31
|
+
if (!this.client) {
|
|
32
|
+
throw new Error('SUMI MCP RPC Client not initialized');
|
|
33
|
+
}
|
|
34
|
+
// 获取 MCP 工具
|
|
35
|
+
const tools = await this.client.$getMCPTools();
|
|
36
|
+
this.logger.log('[Node backend] SUMI MCP tools', tools);
|
|
37
|
+
return tools;
|
|
38
|
+
}
|
|
39
|
+
async callMCPTool(name, args) {
|
|
40
|
+
if (!this.client) {
|
|
41
|
+
throw new Error('SUMI MCP RPC Client not initialized');
|
|
42
|
+
}
|
|
43
|
+
return await this.client.$callMCPTool(name, args);
|
|
44
|
+
}
|
|
45
|
+
getServer() {
|
|
46
|
+
return this.server;
|
|
47
|
+
}
|
|
48
|
+
// TODO 这里涉及到 Chat Stream Call 中带上 ClientID,具体方案需要进一步讨论
|
|
49
|
+
async getAllMCPTools() {
|
|
50
|
+
const registry = this.toolInvocationRegistryManager.getRegistry(this.clientId);
|
|
51
|
+
return registry.getAllFunctions().map((tool) => ({
|
|
52
|
+
name: tool.name || 'no-name',
|
|
53
|
+
description: tool.description || 'no-description',
|
|
54
|
+
inputSchema: tool.parameters,
|
|
55
|
+
providerName: tool.providerName || 'no-provider-name',
|
|
56
|
+
}));
|
|
57
|
+
}
|
|
58
|
+
async initBuiltinMCPServer() {
|
|
59
|
+
var _a;
|
|
60
|
+
const builtinMCPServer = new BuiltinMCPServer(this, this.logger);
|
|
61
|
+
this.mcpServerManager.setClientId(this.clientId);
|
|
62
|
+
await this.mcpServerManager.initBuiltinServer(builtinMCPServer);
|
|
63
|
+
(_a = this.client) === null || _a === void 0 ? void 0 : _a.$updateMCPServers();
|
|
64
|
+
}
|
|
65
|
+
async initExternalMCPServers(servers) {
|
|
66
|
+
var _a;
|
|
67
|
+
this.mcpServerManager.setClientId(this.clientId);
|
|
68
|
+
await this.mcpServerManager.addExternalMCPServers(servers);
|
|
69
|
+
(_a = this.client) === null || _a === void 0 ? void 0 : _a.$updateMCPServers();
|
|
70
|
+
}
|
|
71
|
+
async initExposedMCPServer() {
|
|
72
|
+
// 初始化 MCP Server
|
|
73
|
+
this.server = new index_js_1.Server({
|
|
74
|
+
name: 'sumi-ide-mcp-server',
|
|
75
|
+
version: '0.2.0',
|
|
76
|
+
}, {
|
|
77
|
+
capabilities: {
|
|
78
|
+
tools: {},
|
|
79
|
+
},
|
|
80
|
+
});
|
|
81
|
+
// 设置工具列表请求处理器
|
|
82
|
+
this.server.setRequestHandler(types_js_1.ListToolsRequestSchema, async () => {
|
|
83
|
+
const tools = await this.getMCPTools();
|
|
84
|
+
return { tools };
|
|
85
|
+
});
|
|
86
|
+
// 设置工具调用请求处理器
|
|
87
|
+
this.server.setRequestHandler(types_js_1.CallToolRequestSchema, async (request) => {
|
|
88
|
+
try {
|
|
89
|
+
const { name, arguments: args } = request.params;
|
|
90
|
+
return await this.callMCPTool(name, args);
|
|
91
|
+
}
|
|
92
|
+
catch (error) {
|
|
93
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
94
|
+
return {
|
|
95
|
+
content: [{ type: 'text', text: `Error: ${errorMessage}` }],
|
|
96
|
+
isError: true,
|
|
97
|
+
};
|
|
98
|
+
}
|
|
99
|
+
});
|
|
100
|
+
return this.server;
|
|
101
|
+
}
|
|
102
|
+
};
|
|
103
|
+
exports.SumiMCPServerBackend = SumiMCPServerBackend;
|
|
104
|
+
tslib_1.__decorate([
|
|
105
|
+
(0, di_1.Autowired)(tool_invocation_registry_1.ToolInvocationRegistryManager),
|
|
106
|
+
tslib_1.__metadata("design:type", Object)
|
|
107
|
+
], SumiMCPServerBackend.prototype, "toolInvocationRegistryManager", void 0);
|
|
108
|
+
tslib_1.__decorate([
|
|
109
|
+
(0, di_1.Autowired)(ide_core_node_1.INodeLogger),
|
|
110
|
+
tslib_1.__metadata("design:type", Object)
|
|
111
|
+
], SumiMCPServerBackend.prototype, "logger", void 0);
|
|
112
|
+
exports.SumiMCPServerBackend = SumiMCPServerBackend = tslib_1.__decorate([
|
|
113
|
+
(0, di_1.Injectable)({ multiple: true }),
|
|
114
|
+
tslib_1.__metadata("design:paramtypes", [])
|
|
115
|
+
], SumiMCPServerBackend);
|
|
116
|
+
exports.TokenBuiltinMCPServer = Symbol('TokenBuiltinMCPServer');
|
|
117
|
+
class BuiltinMCPServer {
|
|
118
|
+
constructor(sumiMCPServer, logger) {
|
|
119
|
+
this.sumiMCPServer = sumiMCPServer;
|
|
120
|
+
this.logger = logger;
|
|
121
|
+
this.started = true;
|
|
122
|
+
}
|
|
123
|
+
isStarted() {
|
|
124
|
+
return this.started;
|
|
125
|
+
}
|
|
126
|
+
getServerName() {
|
|
127
|
+
return 'sumi-builtin';
|
|
128
|
+
}
|
|
129
|
+
async start() {
|
|
130
|
+
if (this.started) {
|
|
131
|
+
return;
|
|
132
|
+
}
|
|
133
|
+
// TODO 考虑 MCP Server 的对外暴露
|
|
134
|
+
// await this.sumiMCPServer.initMCPServer();
|
|
135
|
+
this.started = true;
|
|
136
|
+
}
|
|
137
|
+
async callTool(toolName, arg_string) {
|
|
138
|
+
if (!this.started) {
|
|
139
|
+
throw new Error('MCP Server not started');
|
|
140
|
+
}
|
|
141
|
+
let args;
|
|
142
|
+
try {
|
|
143
|
+
args = JSON.parse(arg_string);
|
|
144
|
+
}
|
|
145
|
+
catch (error) {
|
|
146
|
+
this.logger.error(`Failed to parse arguments for calling tool "${toolName}" in Builtin MCP server.
|
|
147
|
+
Invalid JSON: ${arg_string}`, error);
|
|
148
|
+
throw error;
|
|
149
|
+
}
|
|
150
|
+
return this.sumiMCPServer.callMCPTool(toolName, args);
|
|
151
|
+
}
|
|
152
|
+
async getTools() {
|
|
153
|
+
if (!this.started) {
|
|
154
|
+
throw new Error('MCP Server not started');
|
|
155
|
+
}
|
|
156
|
+
const tools = await this.sumiMCPServer.getMCPTools();
|
|
157
|
+
this.logger.debug('[BuiltinMCPServer] getTools', tools);
|
|
158
|
+
return { tools };
|
|
159
|
+
}
|
|
160
|
+
update(_command, _args, _env) {
|
|
161
|
+
// No-op for builtin server as it doesn't need command/args/env updates
|
|
162
|
+
}
|
|
163
|
+
stop() {
|
|
164
|
+
if (!this.started) {
|
|
165
|
+
return;
|
|
166
|
+
}
|
|
167
|
+
// No explicit cleanup needed for in-memory server
|
|
168
|
+
this.started = false;
|
|
169
|
+
}
|
|
170
|
+
}
|
|
171
|
+
exports.BuiltinMCPServer = BuiltinMCPServer;
|
|
172
|
+
//# sourceMappingURL=sumi-mcp-server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
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;AACtD,+DAAoD;AACpD,2DAAsD;AAItD,oFAAsH;AAGtH,wEAAkE;AAElE,+CAA+C;AAC/C,+BAA+B;AAC/B,uBAAuB;AACvB,0BAA0B;AAGnB,IAAM,oBAAoB,GAA1B,MAAM,oBAAqB,SAAQ,2BAAkC;IAgB1E;QACE,KAAK,EAAE,CAAC;QAJV,2BAA2B;QACnB,aAAQ,GAAW,EAAE,CAAC;QAI5B,IAAI,CAAC,gBAAgB,GAAG,IAAI,8CAAoB,CAAC,IAAI,CAAC,6BAA6B,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACpG,CAAC;IAEM,qBAAqB,CAAC,QAAgB;QAC3C,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IAC9C,CAAC;IAED,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,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;QAC/C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;QACxD,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,uDAAuD;IACvD,KAAK,CAAC,cAAc;QAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,6BAA6B,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/E,OAAO,QAAQ,CAAC,eAAe,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC/C,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,SAAS;YAC5B,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,gBAAgB;YACjD,WAAW,EAAE,IAAI,CAAC,UAAU;YAC5B,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,kBAAkB;SACtD,CAAC,CAAC,CAAC;IACN,CAAC;IAEM,KAAK,CAAC,oBAAoB;;QAC/B,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACjE,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;QAChE,MAAA,IAAI,CAAC,MAAM,0CAAE,iBAAiB,EAAE,CAAC;IACnC,CAAC;IAEM,KAAK,CAAC,sBAAsB,CAAC,OAA+B;;QACjE,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,IAAI,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAC3D,MAAA,IAAI,CAAC,MAAM,0CAAE,iBAAiB,EAAE,CAAC;IACnC,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;AA3GY,oDAAoB;AAMd;IADhB,IAAA,cAAS,EAAC,wDAA6B,CAAC;;2EACsC;AAG9D;IADhB,IAAA,cAAS,EAAC,2BAAW,CAAC;;oDACU;+BATtB,oBAAoB;IADhC,IAAA,eAAU,EAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;GAClB,oBAAoB,CA2GhC;AAEY,QAAA,qBAAqB,GAAG,MAAM,CAAC,uBAAuB,CAAC,CAAC;AAErE,MAAa,gBAAgB;IAG3B,YAA6B,aAAmC,EAAmB,MAAe;QAArE,kBAAa,GAAb,aAAa,CAAsB;QAAmB,WAAM,GAAN,MAAM,CAAS;QAF1F,YAAO,GAAY,IAAI,CAAC;IAEqE,CAAC;IAEtG,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,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,+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,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;QACxD,OAAO,EAAE,KAAK,EAAS,CAAC;IAC1B,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;AA5DD,4CA4DC"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
import { ILogger } from '@opensumi/ide-core-common';
|
|
2
|
+
import { MCPServerDescription, MCPServerManager } from '../common/mcp-server-manager';
|
|
3
|
+
import { IToolInvocationRegistryManager } from '../common/tool-invocation-registry';
|
|
4
|
+
import { BuiltinMCPServer } from './mcp/sumi-mcp-server';
|
|
5
|
+
import { IMCPServer } from './mcp-server';
|
|
6
|
+
export declare class MCPServerManagerImpl implements MCPServerManager {
|
|
7
|
+
private readonly toolInvocationRegistryManager;
|
|
8
|
+
private readonly logger;
|
|
9
|
+
protected servers: Map<string, IMCPServer>;
|
|
10
|
+
private clientId;
|
|
11
|
+
constructor(toolInvocationRegistryManager: IToolInvocationRegistryManager, logger: ILogger);
|
|
12
|
+
setClientId(clientId: string): void;
|
|
13
|
+
stopServer(serverName: string): Promise<void>;
|
|
14
|
+
getStartedServers(): Promise<string[]>;
|
|
15
|
+
callTool(serverName: string, toolName: string, arg_string: string): ReturnType<IMCPServer['callTool']>;
|
|
16
|
+
startServer(serverName: string): Promise<void>;
|
|
17
|
+
getServerNames(): Promise<string[]>;
|
|
18
|
+
private convertToToolRequest;
|
|
19
|
+
registerTools(serverName: string): Promise<void>;
|
|
20
|
+
getTools(serverName: string): ReturnType<IMCPServer['getTools']>;
|
|
21
|
+
addOrUpdateServer(description: MCPServerDescription): void;
|
|
22
|
+
addOrUpdateServerDirectly(server: IMCPServer): void;
|
|
23
|
+
initBuiltinServer(builtinMCPServer: BuiltinMCPServer): Promise<void>;
|
|
24
|
+
addExternalMCPServers(servers: MCPServerDescription[]): Promise<void>;
|
|
25
|
+
removeServer(name: string): void;
|
|
26
|
+
}
|
|
27
|
+
//# sourceMappingURL=mcp-server-manager-impl.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-server-manager-impl.d.ts","sourceRoot":"","sources":["../../src/node/mcp-server-manager-impl.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,2BAA2B,CAAC;AAEpD,OAAO,EAAE,oBAAoB,EAAE,gBAAgB,EAAW,MAAM,8BAA8B,CAAC;AAC/F,OAAO,EAAE,8BAA8B,EAAe,MAAM,oCAAoC,CAAC;AAEjG,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,UAAU,EAAiB,MAAM,cAAc,CAAC;AAGzD,qBAAa,oBAAqB,YAAW,gBAAgB;IAOzD,OAAO,CAAC,QAAQ,CAAC,6BAA6B;IAC9C,OAAO,CAAC,QAAQ,CAAC,MAAM;IAPzB,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAa;IAGvD,OAAO,CAAC,QAAQ,CAAS;gBAGN,6BAA6B,EAAE,8BAA8B,EAC7D,MAAM,EAAE,OAAO;IAGlC,WAAW,CAAC,QAAQ,EAAE,MAAM;IAItB,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,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAehD,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;IAI7C,iBAAiB,CAAC,gBAAgB,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAKpE,qBAAqB,CAAC,OAAO,EAAE,oBAAoB,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ3E,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;CASjC"}
|