@opensumi/ide-ai-native 3.8.1-next-1741250387.0 → 3.8.1-next-1741251026.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.contextkeys.d.ts +1 -1
- package/lib/browser/ai-core.contextkeys.d.ts.map +1 -1
- package/lib/browser/ai-core.contextkeys.js +1 -1
- package/lib/browser/ai-core.contextkeys.js.map +1 -1
- package/lib/browser/ai-core.contribution.d.ts +5 -0
- package/lib/browser/ai-core.contribution.d.ts.map +1 -1
- package/lib/browser/ai-core.contribution.js +71 -9
- package/lib/browser/ai-core.contribution.js.map +1 -1
- package/lib/browser/chat/chat-agent.service.d.ts +11 -3
- package/lib/browser/chat/chat-agent.service.d.ts.map +1 -1
- package/lib/browser/chat/chat-agent.service.js +43 -5
- package/lib/browser/chat/chat-agent.service.js.map +1 -1
- package/lib/browser/chat/chat-manager.service.d.ts +1 -0
- package/lib/browser/chat/chat-manager.service.d.ts.map +1 -1
- package/lib/browser/chat/chat-manager.service.js +6 -8
- package/lib/browser/chat/chat-manager.service.js.map +1 -1
- package/lib/browser/chat/chat-model.d.ts +4 -1
- package/lib/browser/chat/chat-model.d.ts.map +1 -1
- package/lib/browser/chat/chat-model.js +57 -2
- package/lib/browser/chat/chat-model.js.map +1 -1
- package/lib/browser/chat/chat-proxy.service.d.ts.map +1 -1
- package/lib/browser/chat/chat-proxy.service.js +13 -6
- package/lib/browser/chat/chat-proxy.service.js.map +1 -1
- package/lib/browser/chat/chat.api.service.d.ts +1 -2
- package/lib/browser/chat/chat.api.service.d.ts.map +1 -1
- package/lib/browser/chat/chat.api.service.js +0 -4
- package/lib/browser/chat/chat.api.service.js.map +1 -1
- package/lib/browser/chat/chat.internal.service.d.ts +2 -0
- package/lib/browser/chat/chat.internal.service.d.ts.map +1 -1
- package/lib/browser/chat/chat.internal.service.js +3 -0
- package/lib/browser/chat/chat.internal.service.js.map +1 -1
- package/lib/browser/chat/chat.module.less +1 -2
- package/lib/browser/chat/chat.view.d.ts.map +1 -1
- package/lib/browser/chat/chat.view.js +46 -29
- package/lib/browser/chat/chat.view.js.map +1 -1
- package/lib/browser/components/ChatContext/index.js +2 -2
- package/lib/browser/components/ChatContext/index.js.map +1 -1
- package/lib/browser/components/ChatInput.d.ts.map +1 -1
- package/lib/browser/components/ChatInput.js +29 -1
- package/lib/browser/components/ChatInput.js.map +1 -1
- package/lib/browser/components/ChatReply.js +6 -1
- package/lib/browser/components/ChatReply.js.map +1 -1
- package/lib/browser/components/ChatThinking.d.ts +0 -2
- package/lib/browser/components/ChatThinking.d.ts.map +1 -1
- package/lib/browser/components/ChatThinking.js +2 -10
- package/lib/browser/components/ChatThinking.js.map +1 -1
- package/lib/browser/components/ChatToolRender.d.ts.map +1 -1
- package/lib/browser/components/ChatToolRender.js +2 -3
- package/lib/browser/components/ChatToolRender.js.map +1 -1
- package/lib/browser/components/WelcomeMsg.js +1 -1
- package/lib/browser/components/WelcomeMsg.js.map +1 -1
- package/lib/browser/components/chat-history.module.less +2 -1
- package/lib/browser/components/components.module.less +22 -0
- package/lib/browser/context/llm-context.service.d.ts +18 -5
- package/lib/browser/context/llm-context.service.d.ts.map +1 -1
- package/lib/browser/context/llm-context.service.js +80 -47
- package/lib/browser/context/llm-context.service.js.map +1 -1
- package/lib/browser/contrib/inline-completions/inline-completions.controller.js +1 -1
- package/lib/browser/contrib/inline-completions/inline-completions.controller.js.map +1 -1
- package/lib/browser/contrib/inline-completions/prompt/matcher.js +2 -2
- package/lib/browser/contrib/inline-completions/prompt/similarSnippets.d.ts +1 -1
- package/lib/browser/contrib/inline-completions/prompt/similarSnippets.js +2 -2
- package/lib/browser/contrib/intelligent-completions/index.d.ts +2 -1
- package/lib/browser/contrib/intelligent-completions/index.d.ts.map +1 -1
- package/lib/browser/contrib/intelligent-completions/index.js +4 -1
- package/lib/browser/contrib/intelligent-completions/index.js.map +1 -1
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.contribution.js +2 -2
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.contribution.js.map +1 -1
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.controller.d.ts.map +1 -1
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.controller.js +6 -5
- package/lib/browser/contrib/intelligent-completions/intelligent-completions.controller.js.map +1 -1
- package/lib/browser/contrib/intelligent-completions/view/code-edits-previewer.d.ts.map +1 -1
- package/lib/browser/contrib/intelligent-completions/view/code-edits-previewer.js +4 -2
- package/lib/browser/contrib/intelligent-completions/view/code-edits-previewer.js.map +1 -1
- package/lib/browser/contrib/intelligent-completions/view/default.d.ts.map +1 -1
- package/lib/browser/contrib/intelligent-completions/view/default.js +23 -11
- package/lib/browser/contrib/intelligent-completions/view/default.js.map +1 -1
- package/lib/browser/index.d.ts.map +1 -1
- package/lib/browser/index.js +4 -0
- package/lib/browser/index.js.map +1 -1
- package/lib/browser/layout/layout.module.less +4 -4
- package/lib/browser/mcp/base-apply.service.d.ts +26 -11
- package/lib/browser/mcp/base-apply.service.d.ts.map +1 -1
- package/lib/browser/mcp/base-apply.service.js +255 -103
- package/lib/browser/mcp/base-apply.service.js.map +1 -1
- package/lib/browser/mcp/config/components/mcp-config.module.less +178 -0
- package/lib/browser/mcp/config/components/mcp-config.view.d.ts +3 -0
- package/lib/browser/mcp/config/components/mcp-config.view.d.ts.map +1 -0
- package/lib/browser/mcp/config/components/mcp-config.view.js +150 -0
- package/lib/browser/mcp/config/components/mcp-config.view.js.map +1 -0
- package/lib/browser/mcp/config/components/mcp-server-form.d.ts +16 -0
- package/lib/browser/mcp/config/components/mcp-server-form.d.ts.map +1 -0
- package/lib/browser/mcp/config/components/mcp-server-form.js +84 -0
- package/lib/browser/mcp/config/components/mcp-server-form.js.map +1 -0
- package/lib/browser/mcp/config/components/mcp-server-form.module.less +78 -0
- package/lib/browser/mcp/config/mcp-config.commands.d.ts +10 -0
- package/lib/browser/mcp/config/mcp-config.commands.d.ts.map +1 -0
- package/lib/browser/mcp/config/mcp-config.commands.js +35 -0
- package/lib/browser/mcp/config/mcp-config.commands.js.map +1 -0
- package/lib/browser/mcp/config/mcp-config.contribution.d.ts +16 -0
- package/lib/browser/mcp/config/mcp-config.contribution.d.ts.map +1 -0
- package/lib/browser/mcp/config/mcp-config.contribution.js +62 -0
- package/lib/browser/mcp/config/mcp-config.contribution.js.map +1 -0
- package/lib/browser/mcp/mcp-server-proxy.service.d.ts +6 -0
- package/lib/browser/mcp/mcp-server-proxy.service.d.ts.map +1 -1
- package/lib/browser/mcp/mcp-server-proxy.service.js +10 -1
- package/lib/browser/mcp/mcp-server-proxy.service.js.map +1 -1
- package/lib/browser/mcp/mcp-server.feature.registry.d.ts +8 -0
- package/lib/browser/mcp/mcp-server.feature.registry.d.ts.map +1 -1
- package/lib/browser/mcp/mcp-server.feature.registry.js +28 -3
- package/lib/browser/mcp/mcp-server.feature.registry.js.map +1 -1
- package/lib/browser/mcp/tools/components/EditFile.d.ts.map +1 -1
- package/lib/browser/mcp/tools/components/EditFile.js +15 -9
- package/lib/browser/mcp/tools/components/EditFile.js.map +1 -1
- package/lib/browser/mcp/tools/components/ExpandableFileList.d.ts +13 -0
- package/lib/browser/mcp/tools/components/ExpandableFileList.d.ts.map +1 -0
- package/lib/browser/mcp/tools/components/{SearchResult.js → ExpandableFileList.js} +29 -19
- package/lib/browser/mcp/tools/components/ExpandableFileList.js.map +1 -0
- package/lib/browser/mcp/tools/components/Terminal.d.ts.map +1 -1
- package/lib/browser/mcp/tools/components/Terminal.js +10 -2
- package/lib/browser/mcp/tools/components/Terminal.js.map +1 -1
- package/lib/browser/mcp/tools/components/index.module.less +4 -0
- package/lib/browser/mcp/tools/createNewFileWithText.d.ts +1 -0
- package/lib/browser/mcp/tools/createNewFileWithText.d.ts.map +1 -1
- package/lib/browser/mcp/tools/createNewFileWithText.js +19 -11
- package/lib/browser/mcp/tools/createNewFileWithText.js.map +1 -1
- package/lib/browser/mcp/tools/fileSearch.d.ts +1 -0
- package/lib/browser/mcp/tools/fileSearch.d.ts.map +1 -1
- package/lib/browser/mcp/tools/fileSearch.js +14 -5
- package/lib/browser/mcp/tools/fileSearch.js.map +1 -1
- package/lib/browser/mcp/tools/getDiagnosticsByPath.d.ts.map +1 -1
- package/lib/browser/mcp/tools/getDiagnosticsByPath.js +1 -0
- package/lib/browser/mcp/tools/getDiagnosticsByPath.js.map +1 -1
- package/lib/browser/mcp/tools/grepSearch.d.ts.map +1 -1
- package/lib/browser/mcp/tools/grepSearch.js +6 -3
- package/lib/browser/mcp/tools/grepSearch.js.map +1 -1
- package/lib/browser/mcp/tools/handlers/EditFile.js +1 -1
- package/lib/browser/mcp/tools/handlers/EditFile.js.map +1 -1
- package/lib/browser/mcp/tools/handlers/ListDir.d.ts +1 -0
- package/lib/browser/mcp/tools/handlers/ListDir.d.ts.map +1 -1
- package/lib/browser/mcp/tools/handlers/ListDir.js +3 -0
- package/lib/browser/mcp/tools/handlers/ListDir.js.map +1 -1
- package/lib/browser/mcp/tools/handlers/RunCommand.d.ts.map +1 -1
- package/lib/browser/mcp/tools/handlers/RunCommand.js +2 -0
- package/lib/browser/mcp/tools/handlers/RunCommand.js.map +1 -1
- package/lib/browser/mcp/tools/listDir.d.ts +1 -0
- package/lib/browser/mcp/tools/listDir.d.ts.map +1 -1
- package/lib/browser/mcp/tools/listDir.js +35 -4
- package/lib/browser/mcp/tools/listDir.js.map +1 -1
- package/lib/browser/mcp/tools/runTerminalCmd.d.ts.map +1 -1
- package/lib/browser/mcp/tools/runTerminalCmd.js +1 -0
- package/lib/browser/mcp/tools/runTerminalCmd.js.map +1 -1
- package/lib/browser/model/msg-history-manager.d.ts +2 -2
- package/lib/browser/model/msg-history-manager.d.ts.map +1 -1
- package/lib/browser/model/msg-history-manager.js +4 -10
- package/lib/browser/model/msg-history-manager.js.map +1 -1
- package/lib/browser/preferences/schema.d.ts.map +1 -1
- package/lib/browser/preferences/schema.js +22 -2
- package/lib/browser/preferences/schema.js.map +1 -1
- package/lib/browser/types.d.ts +4 -0
- package/lib/browser/types.d.ts.map +1 -1
- package/lib/browser/widget/inline-chat/inline-chat-controller.d.ts.map +1 -1
- package/lib/browser/widget/inline-chat/inline-chat-controller.js +6 -1
- package/lib/browser/widget/inline-chat/inline-chat-controller.js.map +1 -1
- package/lib/browser/widget/inline-diff/inline-diff-manager.d.ts.map +1 -1
- package/lib/browser/widget/inline-diff/inline-diff-manager.js +68 -8
- package/lib/browser/widget/inline-diff/inline-diff-manager.js.map +1 -1
- package/lib/browser/widget/inline-diff/inline-diff-previewer.d.ts +10 -4
- package/lib/browser/widget/inline-diff/inline-diff-previewer.d.ts.map +1 -1
- package/lib/browser/widget/inline-diff/inline-diff-previewer.js +14 -3
- package/lib/browser/widget/inline-diff/inline-diff-previewer.js.map +1 -1
- package/lib/browser/widget/inline-diff/inline-diff-widget.module.less +25 -4
- package/lib/browser/widget/inline-diff/inline-diff.controller.d.ts +3 -3
- package/lib/browser/widget/inline-diff/inline-diff.controller.d.ts.map +1 -1
- package/lib/browser/widget/inline-diff/inline-diff.controller.js +10 -5
- package/lib/browser/widget/inline-diff/inline-diff.controller.js.map +1 -1
- package/lib/browser/widget/inline-stream-diff/inline-stream-diff.handler.d.ts +46 -17
- package/lib/browser/widget/inline-stream-diff/inline-stream-diff.handler.d.ts.map +1 -1
- package/lib/browser/widget/inline-stream-diff/inline-stream-diff.handler.js +110 -53
- package/lib/browser/widget/inline-stream-diff/inline-stream-diff.handler.js.map +1 -1
- package/lib/browser/widget/inline-stream-diff/live-preview.decoration.d.ts +4 -0
- package/lib/browser/widget/inline-stream-diff/live-preview.decoration.d.ts.map +1 -1
- package/lib/browser/widget/inline-stream-diff/live-preview.decoration.js +26 -1
- package/lib/browser/widget/inline-stream-diff/live-preview.decoration.js.map +1 -1
- package/lib/common/index.d.ts +13 -4
- package/lib/common/index.d.ts.map +1 -1
- package/lib/common/index.js +5 -1
- package/lib/common/index.js.map +1 -1
- package/lib/common/llm-context.d.ts +13 -9
- package/lib/common/llm-context.d.ts.map +1 -1
- package/lib/common/llm-context.js.map +1 -1
- package/lib/common/mcp-server-manager.d.ts +17 -1
- package/lib/common/mcp-server-manager.d.ts.map +1 -1
- package/lib/common/mcp-server-manager.js.map +1 -1
- package/lib/common/model.d.ts +12 -0
- package/lib/common/model.d.ts.map +1 -0
- package/lib/common/model.js +83 -0
- package/lib/common/model.js.map +1 -0
- package/lib/common/prompts/context-prompt-provider.d.ts +4 -3
- package/lib/common/prompts/context-prompt-provider.d.ts.map +1 -1
- package/lib/common/prompts/context-prompt-provider.js +33 -22
- package/lib/common/prompts/context-prompt-provider.js.map +1 -1
- package/lib/common/tool-invocation-registry.d.ts +2 -2
- package/lib/common/tool-invocation-registry.d.ts.map +1 -1
- package/lib/common/tool-invocation-registry.js +1 -1
- package/lib/common/tool-invocation-registry.js.map +1 -1
- package/lib/common/types.d.ts +8 -0
- package/lib/common/types.d.ts.map +1 -1
- package/lib/common/utils.d.ts.map +1 -1
- package/lib/common/utils.js +2 -1
- package/lib/common/utils.js.map +1 -1
- package/lib/node/anthropic/anthropic-language-model.d.ts +3 -1
- package/lib/node/anthropic/anthropic-language-model.d.ts.map +1 -1
- package/lib/node/anthropic/anthropic-language-model.js +6 -2
- package/lib/node/anthropic/anthropic-language-model.js.map +1 -1
- package/lib/node/base-language-model.d.ts +6 -3
- package/lib/node/base-language-model.d.ts.map +1 -1
- package/lib/node/base-language-model.js +14 -31
- package/lib/node/base-language-model.js.map +1 -1
- package/lib/node/deepseek/deepseek-language-model.d.ts +3 -1
- package/lib/node/deepseek/deepseek-language-model.d.ts.map +1 -1
- package/lib/node/deepseek/deepseek-language-model.js +6 -2
- package/lib/node/deepseek/deepseek-language-model.js.map +1 -1
- package/lib/node/mcp/sumi-mcp-server.d.ts +17 -3
- package/lib/node/mcp/sumi-mcp-server.d.ts.map +1 -1
- package/lib/node/mcp/sumi-mcp-server.js +59 -6
- package/lib/node/mcp/sumi-mcp-server.js.map +1 -1
- package/lib/node/mcp-server-manager-impl.d.ts +4 -3
- package/lib/node/mcp-server-manager-impl.d.ts.map +1 -1
- package/lib/node/mcp-server-manager-impl.js +26 -6
- package/lib/node/mcp-server-manager-impl.js.map +1 -1
- package/lib/node/mcp-server.d.ts +5 -16
- package/lib/node/mcp-server.d.ts.map +1 -1
- package/lib/node/mcp-server.js +12 -6
- package/lib/node/mcp-server.js.map +1 -1
- package/lib/node/openai/openai-language-model.d.ts +4 -2
- package/lib/node/openai/openai-language-model.d.ts.map +1 -1
- package/lib/node/openai/openai-language-model.js +6 -4
- package/lib/node/openai/openai-language-model.js.map +1 -1
- package/lib/node/openai-compatible/openai-compatible-language-model.d.ts +10 -0
- package/lib/node/openai-compatible/openai-compatible-language-model.d.ts.map +1 -0
- package/lib/node/openai-compatible/openai-compatible-language-model.js +32 -0
- package/lib/node/openai-compatible/openai-compatible-language-model.js.map +1 -0
- package/package.json +27 -26
- package/src/browser/ai-core.contextkeys.ts +3 -3
- package/src/browser/ai-core.contribution.ts +89 -12
- package/src/browser/chat/chat-agent.service.ts +53 -9
- package/src/browser/chat/chat-manager.service.ts +10 -8
- package/src/browser/chat/chat-model.ts +63 -5
- package/src/browser/chat/chat-proxy.service.ts +16 -8
- package/src/browser/chat/chat.api.service.ts +1 -5
- package/src/browser/chat/chat.internal.service.ts +4 -0
- package/src/browser/chat/chat.module.less +1 -2
- package/src/browser/chat/chat.view.tsx +63 -69
- package/src/browser/components/ChatContext/index.tsx +2 -2
- package/src/browser/components/ChatInput.tsx +81 -6
- package/src/browser/components/ChatReply.tsx +6 -6
- package/src/browser/components/ChatThinking.tsx +3 -9
- package/src/browser/components/ChatToolRender.tsx +1 -2
- package/src/browser/components/WelcomeMsg.tsx +1 -1
- package/src/browser/components/chat-history.module.less +2 -1
- package/src/browser/components/components.module.less +22 -0
- package/src/browser/context/llm-context.service.ts +93 -54
- package/src/browser/contrib/inline-completions/inline-completions.controller.ts +1 -1
- package/src/browser/contrib/inline-completions/prompt/matcher.ts +2 -2
- package/src/browser/contrib/inline-completions/prompt/similarSnippets.ts +2 -2
- package/src/browser/contrib/intelligent-completions/index.ts +5 -1
- package/src/browser/contrib/intelligent-completions/intelligent-completions.contribution.ts +3 -3
- package/src/browser/contrib/intelligent-completions/intelligent-completions.controller.ts +7 -6
- package/src/browser/contrib/intelligent-completions/view/code-edits-previewer.ts +4 -2
- package/src/browser/contrib/intelligent-completions/view/default.ts +34 -19
- package/src/browser/index.ts +4 -0
- package/src/browser/layout/layout.module.less +4 -4
- package/src/browser/mcp/base-apply.service.ts +306 -109
- package/src/browser/mcp/config/components/mcp-config.module.less +178 -0
- package/src/browser/mcp/config/components/mcp-config.view.tsx +215 -0
- package/src/browser/mcp/config/components/mcp-server-form.module.less +78 -0
- package/src/browser/mcp/config/components/mcp-server-form.tsx +144 -0
- package/src/browser/mcp/config/mcp-config.commands.ts +29 -0
- package/src/browser/mcp/config/mcp-config.contribution.ts +65 -0
- package/src/browser/mcp/mcp-server-proxy.service.ts +14 -2
- package/src/browser/mcp/mcp-server.feature.registry.ts +35 -4
- package/src/browser/mcp/tools/components/EditFile.tsx +16 -9
- package/src/browser/mcp/tools/components/ExpandableFileList.tsx +133 -0
- package/src/browser/mcp/tools/components/Terminal.tsx +13 -2
- package/src/browser/mcp/tools/components/index.module.less +4 -0
- package/src/browser/mcp/tools/createNewFileWithText.ts +21 -12
- package/src/browser/mcp/tools/fileSearch.ts +14 -4
- package/src/browser/mcp/tools/getDiagnosticsByPath.ts +1 -0
- package/src/browser/mcp/tools/grepSearch.ts +6 -3
- package/src/browser/mcp/tools/handlers/EditFile.ts +1 -1
- package/src/browser/mcp/tools/handlers/ListDir.ts +4 -0
- package/src/browser/mcp/tools/handlers/RunCommand.ts +2 -0
- package/src/browser/mcp/tools/listDir.ts +36 -5
- package/src/browser/mcp/tools/runTerminalCmd.ts +1 -0
- package/src/browser/model/msg-history-manager.ts +5 -11
- package/src/browser/preferences/schema.ts +22 -2
- package/src/browser/types.ts +1 -0
- package/src/browser/widget/inline-chat/inline-chat-controller.ts +5 -1
- package/src/browser/widget/inline-diff/inline-diff-manager.tsx +143 -21
- package/src/browser/widget/inline-diff/inline-diff-previewer.ts +25 -7
- package/src/browser/widget/inline-diff/inline-diff-widget.module.less +25 -4
- package/src/browser/widget/inline-diff/inline-diff.controller.ts +16 -8
- package/src/browser/widget/inline-stream-diff/inline-stream-diff.handler.tsx +139 -68
- package/src/browser/widget/inline-stream-diff/live-preview.decoration.tsx +30 -1
- package/src/common/index.ts +15 -4
- package/src/common/llm-context.ts +10 -4
- package/src/common/mcp-server-manager.ts +17 -1
- package/src/common/model.ts +90 -0
- package/src/common/prompts/context-prompt-provider.ts +38 -29
- package/src/common/tool-invocation-registry.ts +2 -2
- package/src/common/types.ts +8 -0
- package/src/common/utils.ts +3 -1
- package/src/node/anthropic/anthropic-language-model.ts +7 -2
- package/src/node/base-language-model.ts +25 -38
- package/src/node/deepseek/deepseek-language-model.ts +7 -2
- package/src/node/mcp/sumi-mcp-server.ts +67 -9
- package/src/node/mcp-server-manager-impl.ts +30 -9
- package/src/node/mcp-server.ts +11 -14
- package/src/node/openai/openai-language-model.ts +8 -4
- package/src/node/openai-compatible/openai-compatible-language-model.ts +30 -0
- package/lib/browser/mcp/tools/components/SearchResult.d.ts +0 -11
- package/lib/browser/mcp/tools/components/SearchResult.d.ts.map +0 -1
- package/lib/browser/mcp/tools/components/SearchResult.js.map +0 -1
- package/src/browser/mcp/tools/components/SearchResult.tsx +0 -92
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/common/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAErD,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;IAEH,YAAY,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAEnC,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/common/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAErD,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;IAEH,YAAY,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAEnC,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAEnC,WAAW,IAAI,OAAO,CAAC,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC,CAAC;IAEpE,YAAY,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEhD,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAChD;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,GAAG,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,eAAe,CAAC;IACxB,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,eAAe,EAAE,OAAO,EAAE,CAAC;KAC5B,CAAC;CACH;AAED,MAAM,MAAM,eAAe,GAAG,YAAY,GAAG,SAAS,GAAG,SAAS,GAAG,UAAU,GAAG,QAAQ,GAAG,WAAW,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/common/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;
|
|
1
|
+
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/common/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAO5D,eAAO,MAAM,kBAAkB,aAAc,oBAAoB,YAShE,CAAC;AAEF,eAAO,MAAM,kBAAkB,aAAc,oBAAoB,YAAiD,CAAC;AAEnH,eAAO,MAAM,eAAe,aAAc,oBAAoB,YAK7D,CAAC;AAGF,eAAO,MAAM,iBAAiB,YAAa,MAAM,KAAG,MAuBnD,CAAC;AAEF,eAAO,MAAM,WAAW,aAAc,MAAM,gCAA0E,CAAC"}
|
package/lib/common/utils.js
CHANGED
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.getToolName = exports.extractCodeBlocks = exports.isDocumentValid = exports.isDocumentTooShort = exports.isDocumentTooLarge = void 0;
|
|
4
|
+
const index_1 = require("./index");
|
|
4
5
|
const BACK_QUOTE_3_SYMBOL = '```';
|
|
5
6
|
const MIN_PROMPT_CHARS = 10;
|
|
6
7
|
const isDocumentTooLarge = (document) => {
|
|
@@ -47,6 +48,6 @@ const extractCodeBlocks = (content) => {
|
|
|
47
48
|
return newContents.join('\n');
|
|
48
49
|
};
|
|
49
50
|
exports.extractCodeBlocks = extractCodeBlocks;
|
|
50
|
-
const getToolName = (toolName, serverName =
|
|
51
|
+
const getToolName = (toolName, serverName = index_1.BUILTIN_MCP_SERVER_NAME) => `mcp_${serverName}_${toolName}`;
|
|
51
52
|
exports.getToolName = getToolName;
|
|
52
53
|
//# sourceMappingURL=utils.js.map
|
package/lib/common/utils.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/common/utils.ts"],"names":[],"mappings":";;;AAEA,MAAM,mBAAmB,GAAG,KAAK,CAAC;AAClC,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAErB,MAAM,kBAAkB,GAAG,CAAC,QAA8B,EAAE,EAAE;IACnE,IAAI,CAAC;QACH,QAAQ,CAAC,OAAO,EAAE,CAAC;IACrB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,CAAC,YAAY,UAAU,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AATW,QAAA,kBAAkB,sBAS7B;AAEK,MAAM,kBAAkB,GAAG,CAAC,QAA8B,EAAE,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,MAAM,GAAG,gBAAgB,CAAC;AAAtG,QAAA,kBAAkB,sBAAoF;AAE5G,MAAM,eAAe,GAAG,CAAC,QAA8B,EAAE,EAAE;IAChE,IAAI,IAAA,0BAAkB,EAAC,QAAQ,CAAC,IAAI,IAAA,0BAAkB,EAAC,QAAQ,CAAC,EAAE,CAAC;QACjE,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AALW,QAAA,eAAe,mBAK1B;AAEF,eAAe;AACR,MAAM,iBAAiB,GAAG,CAAC,OAAe,EAAU,EAAE;IAC3D,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAElC,IAAI,WAAW,GAAa,EAAE,CAAC;IAC/B,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;QACxB,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAC5D,OAAO,GAAG,IAAI,CAAC;YACf,SAAS,GAAG,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;aAAM,IAAI,OAAO,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAClE,OAAO,GAAG,KAAK,CAAC;YAChB,MAAM,OAAO,GAAG,CAAC,CAAC;YAClB,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,OAAO,IAAI,SAAS,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;YACnC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChC,CAAC,CAAC;AAvBW,QAAA,iBAAiB,qBAuB5B;AAEK,MAAM,WAAW,GAAG,CAAC,QAAgB,EAAE,UAAU,GAAG,
|
|
1
|
+
{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/common/utils.ts"],"names":[],"mappings":";;;AAEA,mCAAkD;AAElD,MAAM,mBAAmB,GAAG,KAAK,CAAC;AAClC,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAErB,MAAM,kBAAkB,GAAG,CAAC,QAA8B,EAAE,EAAE;IACnE,IAAI,CAAC;QACH,QAAQ,CAAC,OAAO,EAAE,CAAC;IACrB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,CAAC,YAAY,UAAU,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AATW,QAAA,kBAAkB,sBAS7B;AAEK,MAAM,kBAAkB,GAAG,CAAC,QAA8B,EAAE,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,MAAM,GAAG,gBAAgB,CAAC;AAAtG,QAAA,kBAAkB,sBAAoF;AAE5G,MAAM,eAAe,GAAG,CAAC,QAA8B,EAAE,EAAE;IAChE,IAAI,IAAA,0BAAkB,EAAC,QAAQ,CAAC,IAAI,IAAA,0BAAkB,EAAC,QAAQ,CAAC,EAAE,CAAC;QACjE,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AALW,QAAA,eAAe,mBAK1B;AAEF,eAAe;AACR,MAAM,iBAAiB,GAAG,CAAC,OAAe,EAAU,EAAE;IAC3D,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAElC,IAAI,WAAW,GAAa,EAAE,CAAC;IAC/B,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;QACxB,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAC5D,OAAO,GAAG,IAAI,CAAC;YACf,SAAS,GAAG,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;aAAM,IAAI,OAAO,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAClE,OAAO,GAAG,KAAK,CAAC;YAChB,MAAM,OAAO,GAAG,CAAC,CAAC;YAClB,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,OAAO,IAAI,SAAS,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;YACnC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChC,CAAC,CAAC;AAvBW,QAAA,iBAAiB,qBAuB5B;AAEK,MAAM,WAAW,GAAG,CAAC,QAAgB,EAAE,UAAU,GAAG,+BAAuB,EAAE,EAAE,CAAC,OAAO,UAAU,IAAI,QAAQ,EAAE,CAAC;AAA1G,QAAA,WAAW,eAA+F"}
|
|
@@ -1,9 +1,11 @@
|
|
|
1
1
|
import { AnthropicProvider } from '@ai-sdk/anthropic';
|
|
2
2
|
import { IAIBackServiceOption } from '@opensumi/ide-core-common';
|
|
3
|
+
import { ModelInfo } from '../../common/model';
|
|
3
4
|
import { BaseLanguageModel } from '../base-language-model';
|
|
4
5
|
export declare const AnthropicModelIdentifier: unique symbol;
|
|
5
6
|
export declare class AnthropicModel extends BaseLanguageModel {
|
|
6
7
|
protected initializeProvider(options: IAIBackServiceOption): AnthropicProvider;
|
|
7
|
-
protected getModelIdentifier(provider: AnthropicProvider): import("ai").LanguageModelV1;
|
|
8
|
+
protected getModelIdentifier(provider: AnthropicProvider, modelId?: string): import("ai").LanguageModelV1;
|
|
9
|
+
protected getModelInfo(modelId: string): ModelInfo | undefined;
|
|
8
10
|
}
|
|
9
11
|
//# sourceMappingURL=anthropic-language-model.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
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;AAGvE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAGjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAE3D,eAAO,MAAM,wBAAwB,eAAqC,CAAC;AAE3E,qBACa,cAAe,SAAQ,iBAAiB;IACnD,SAAS,CAAC,kBAAkB,CAAC,OAAO,EAAE,oBAAoB,GAAG,iBAAiB;IAS9E,SAAS,CAAC,kBAAkB,CAAC,QAAQ,EAAE,iBAAiB;
|
|
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;AAGvE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAGjE,OAAO,EAAE,SAAS,EAAmB,MAAM,oBAAoB,CAAC;AAChE,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAE3D,eAAO,MAAM,wBAAwB,eAAqC,CAAC;AAE3E,qBACa,cAAe,SAAQ,iBAAiB;IACnD,SAAS,CAAC,kBAAkB,CAAC,OAAO,EAAE,oBAAoB,GAAG,iBAAiB;IAS9E,SAAS,CAAC,kBAAkB,CAAC,QAAQ,EAAE,iBAAiB,EAAE,OAAO,SAA+B;IAIhG,SAAS,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;CAG/D"}
|
|
@@ -5,6 +5,7 @@ const tslib_1 = require("tslib");
|
|
|
5
5
|
const anthropic_1 = require("@ai-sdk/anthropic");
|
|
6
6
|
const di_1 = require("@opensumi/di");
|
|
7
7
|
const ai_native_1 = require("@opensumi/ide-core-common/lib/settings/ai-native");
|
|
8
|
+
const model_1 = require("../../common/model");
|
|
8
9
|
const base_language_model_1 = require("../base-language-model");
|
|
9
10
|
exports.AnthropicModelIdentifier = Symbol('AnthropicModelIdentifier');
|
|
10
11
|
let AnthropicModel = class AnthropicModel extends base_language_model_1.BaseLanguageModel {
|
|
@@ -15,8 +16,11 @@ let AnthropicModel = class AnthropicModel extends base_language_model_1.BaseLang
|
|
|
15
16
|
}
|
|
16
17
|
return (0, anthropic_1.createAnthropic)({ apiKey });
|
|
17
18
|
}
|
|
18
|
-
getModelIdentifier(provider) {
|
|
19
|
-
return provider(
|
|
19
|
+
getModelIdentifier(provider, modelId = 'claude-3-5-sonnet-20241022') {
|
|
20
|
+
return provider(modelId);
|
|
21
|
+
}
|
|
22
|
+
getModelInfo(modelId) {
|
|
23
|
+
return model_1.anthropicModels[modelId];
|
|
20
24
|
}
|
|
21
25
|
};
|
|
22
26
|
exports.AnthropicModel = AnthropicModel;
|
|
@@ -1 +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;
|
|
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,8CAAgE;AAChE,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,EAAE,OAAO,GAAG,4BAA4B;QAC9F,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAES,YAAY,CAAC,OAAe;QACpC,OAAO,uBAAe,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;CACF,CAAA;AAjBY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,eAAU,GAAE;GACA,cAAc,CAiB1B"}
|
|
@@ -1,14 +1,17 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { CoreMessage } from 'ai';
|
|
2
|
+
import { IAIBackServiceOption } from '@opensumi/ide-core-common';
|
|
2
3
|
import { ChatReadableStream } from '@opensumi/ide-core-node';
|
|
3
4
|
import { CancellationToken } from '@opensumi/ide-utils';
|
|
5
|
+
import { ModelInfo } from '../common';
|
|
4
6
|
import { IToolInvocationRegistryManager, ToolRequest } from '../common/tool-invocation-registry';
|
|
5
7
|
export declare abstract class BaseLanguageModel {
|
|
8
|
+
static ModelOptions: Record<string, ModelInfo>;
|
|
6
9
|
protected readonly toolInvocationRegistryManager: IToolInvocationRegistryManager;
|
|
7
10
|
protected abstract initializeProvider(options: IAIBackServiceOption): any;
|
|
8
|
-
private convertChatMessageRole;
|
|
9
11
|
request(request: string, chatReadableStream: ChatReadableStream, options: IAIBackServiceOption, cancellationToken?: CancellationToken): Promise<any>;
|
|
10
12
|
private convertToolRequestToAITool;
|
|
11
13
|
protected abstract getModelIdentifier(provider: any, modelId?: string): any;
|
|
12
|
-
protected
|
|
14
|
+
protected abstract getModelInfo(modelId: string): ModelInfo | undefined;
|
|
15
|
+
protected handleStreamingRequest(provider: any, request: string, tools: ToolRequest[], chatReadableStream: ChatReadableStream, history?: CoreMessage[], modelId?: string, providerOptions?: Record<string, any>, trimTexts?: [string, string], systemPrompt?: string, maxTokens?: number, cancellationToken?: CancellationToken): Promise<any>;
|
|
13
16
|
}
|
|
14
17
|
//# sourceMappingURL=base-language-model.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"base-language-model.d.ts","sourceRoot":"","sources":["../../src/node/base-language-model.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"base-language-model.d.ts","sourceRoot":"","sources":["../../src/node/base-language-model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAsD,MAAM,IAAI,CAAC;AAGrF,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AACjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAExD,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EACL,8BAA8B,EAE9B,WAAW,EACZ,MAAM,oCAAoC,CAAC;AAE5C,8BACsB,iBAAiB;IACrC,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAG/C,SAAS,CAAC,QAAQ,CAAC,6BAA6B,EAAE,8BAA8B,CAAC;IAEjF,SAAS,CAAC,QAAQ,CAAC,kBAAkB,CAAC,OAAO,EAAE,oBAAoB,GAAG,GAAG;IAEnE,OAAO,CACX,OAAO,EAAE,MAAM,EACf,kBAAkB,EAAE,kBAAkB,EACtC,OAAO,EAAE,oBAAoB,EAC7B,iBAAiB,CAAC,EAAE,iBAAiB,GACpC,OAAO,CAAC,GAAG,CAAC;IA0Bf,OAAO,CAAC,0BAA0B;IAUlC,SAAS,CAAC,QAAQ,CAAC,kBAAkB,CAAC,QAAQ,EAAE,GAAG,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,GAAG;IAE3E,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;cAEvD,sBAAsB,CACpC,QAAQ,EAAE,GAAG,EACb,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,WAAW,EAAE,EACpB,kBAAkB,EAAE,kBAAkB,EACtC,OAAO,GAAE,WAAW,EAAO,EAC3B,OAAO,CAAC,EAAE,MAAM,EAChB,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EACrC,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,EAC5B,YAAY,CAAC,EAAE,MAAM,EACrB,SAAS,CAAC,EAAE,MAAM,EAClB,iBAAiB,CAAC,EAAE,iBAAiB,GACpC,OAAO,CAAC,GAAG,CAAC;CAiJhB"}
|
|
@@ -6,29 +6,16 @@ const ai_1 = require("ai");
|
|
|
6
6
|
const di_1 = require("@opensumi/di");
|
|
7
7
|
const tool_invocation_registry_1 = require("../common/tool-invocation-registry");
|
|
8
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
9
|
async request(request, chatReadableStream, options, cancellationToken) {
|
|
24
10
|
const provider = this.initializeProvider(options);
|
|
25
11
|
const clientId = options.clientId;
|
|
26
|
-
|
|
27
|
-
|
|
12
|
+
let allFunctions = [];
|
|
13
|
+
// 如果没有传入 clientId,则不使用工具
|
|
14
|
+
if (clientId) {
|
|
15
|
+
const registry = this.toolInvocationRegistryManager.getRegistry(clientId);
|
|
16
|
+
allFunctions = options.noTool ? [] : registry.getAllFunctions();
|
|
28
17
|
}
|
|
29
|
-
|
|
30
|
-
const allFunctions = options.noTool ? [] : registry.getAllFunctions();
|
|
31
|
-
return this.handleStreamingRequest(provider, request, allFunctions, chatReadableStream, options.history || [], options.modelId, options.temperature, options.topP, options.topK, options.providerOptions, options.trimTexts, cancellationToken);
|
|
18
|
+
return this.handleStreamingRequest(provider, request, allFunctions, chatReadableStream, options.history || [], options.modelId, options.providerOptions, options.trimTexts, options.system, options.maxTokens, cancellationToken);
|
|
32
19
|
}
|
|
33
20
|
convertToolRequestToAITool(toolRequest) {
|
|
34
21
|
return (0, ai_1.tool)({
|
|
@@ -38,7 +25,7 @@ let BaseLanguageModel = class BaseLanguageModel {
|
|
|
38
25
|
execute: async (args, options) => await toolRequest.handler(JSON.stringify(args), options),
|
|
39
26
|
});
|
|
40
27
|
}
|
|
41
|
-
async handleStreamingRequest(provider, request, tools, chatReadableStream, history = [], modelId,
|
|
28
|
+
async handleStreamingRequest(provider, request, tools, chatReadableStream, history = [], modelId, providerOptions, trimTexts, systemPrompt, maxTokens, cancellationToken) {
|
|
42
29
|
try {
|
|
43
30
|
const aiTools = Object.fromEntries(tools.map((tool) => [tool.name, this.convertToolRequestToAITool(tool)]));
|
|
44
31
|
const abortController = new AbortController();
|
|
@@ -47,24 +34,20 @@ let BaseLanguageModel = class BaseLanguageModel {
|
|
|
47
34
|
abortController.abort();
|
|
48
35
|
});
|
|
49
36
|
}
|
|
50
|
-
const messages = [
|
|
51
|
-
|
|
52
|
-
role: this.convertChatMessageRole(msg.role), // 这个 SDK 包里的类型不太好完全对应,
|
|
53
|
-
content: msg.content,
|
|
54
|
-
})),
|
|
55
|
-
{ role: 'user', content: request },
|
|
56
|
-
];
|
|
37
|
+
const messages = [...history, { role: 'user', content: request }];
|
|
38
|
+
const modelInfo = modelId ? this.getModelInfo(modelId) : undefined;
|
|
57
39
|
const stream = (0, ai_1.streamText)({
|
|
58
40
|
model: this.getModelIdentifier(provider, modelId),
|
|
59
|
-
maxTokens: 4096,
|
|
60
41
|
tools: aiTools,
|
|
61
42
|
messages,
|
|
62
43
|
abortSignal: abortController.signal,
|
|
63
44
|
experimental_toolCallStreaming: true,
|
|
64
45
|
maxSteps: 12,
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
46
|
+
maxTokens,
|
|
47
|
+
temperature: (modelInfo === null || modelInfo === void 0 ? void 0 : modelInfo.temperature) || 0,
|
|
48
|
+
topP: (modelInfo === null || modelInfo === void 0 ? void 0 : modelInfo.topP) || 0.8,
|
|
49
|
+
topK: (modelInfo === null || modelInfo === void 0 ? void 0 : modelInfo.topK) || 1,
|
|
50
|
+
system: systemPrompt,
|
|
68
51
|
providerOptions,
|
|
69
52
|
});
|
|
70
53
|
// 状态跟踪变量
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"base-language-model.js","sourceRoot":"","sources":["../../src/node/base-language-model.ts"],"names":[],"mappings":";;;;AAAA,2BAAqF;AAErF,qCAAqD;
|
|
1
|
+
{"version":3,"file":"base-language-model.js","sourceRoot":"","sources":["../../src/node/base-language-model.ts"],"names":[],"mappings":";;;;AAAA,2BAAqF;AAErF,qCAAqD;AAMrD,iFAI4C;AAGrC,IAAe,iBAAiB,GAAhC,MAAe,iBAAiB;IAQrC,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;QAElC,IAAI,YAAY,GAAkB,EAAE,CAAC;QACrC,yBAAyB;QACzB,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,IAAI,CAAC,6BAA6B,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAC1E,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC;QAClE,CAAC;QAED,OAAO,IAAI,CAAC,sBAAsB,CAChC,QAAQ,EACR,OAAO,EACP,YAAY,EACZ,kBAAkB,EAClB,OAAO,CAAC,OAAO,IAAI,EAAE,EACrB,OAAO,CAAC,OAAO,EACf,OAAO,CAAC,eAAe,EACvB,OAAO,CAAC,SAAS,EACjB,OAAO,CAAC,MAAM,EACd,OAAO,CAAC,SAAS,EACjB,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,OAA6B,EAAE,EAAE,CAC1D,MAAM,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC;SAC3D,CAAC,CAAC;IACL,CAAC;IAMS,KAAK,CAAC,sBAAsB,CACpC,QAAa,EACb,OAAe,EACf,KAAoB,EACpB,kBAAsC,EACtC,UAAyB,EAAE,EAC3B,OAAgB,EAChB,eAAqC,EACrC,SAA4B,EAC5B,YAAqB,EACrB,SAAkB,EAClB,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,CAAC,GAAG,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YACjF,MAAM,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACnE,MAAM,MAAM,GAAG,IAAA,eAAU,EAAC;gBACxB,KAAK,EAAE,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,OAAO,CAAC;gBACjD,KAAK,EAAE,OAAO;gBACd,QAAQ;gBACR,WAAW,EAAE,eAAe,CAAC,MAAM;gBACnC,8BAA8B,EAAE,IAAI;gBACpC,QAAQ,EAAE,EAAE;gBACZ,SAAS;gBACT,WAAW,EAAE,CAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,WAAW,KAAI,CAAC;gBACxC,IAAI,EAAE,CAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,IAAI,KAAI,GAAG;gBAC5B,IAAI,EAAE,CAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,IAAI,KAAI,CAAC;gBAC1B,MAAM,EAAE,YAAY;gBACpB,eAAe;aAChB,CAAC,CAAC;YAEH,SAAS;YACT,IAAI,YAAY,GAAG,IAAI,CAAC;YACxB,IAAI,YAAY,GAAG,EAAE,CAAC;YACtB,MAAM,YAAY,GAAa,EAAE,CAAC;YAClC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;gBAC5C,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAChC,IAAI,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,MAAM,EAAE,CAAC;wBACtB,eAAe;wBACf,YAAY,IAAI,KAAK,CAAC,SAAS,CAAC;wBAEhC,qBAAqB;wBACrB,IAAI,YAAY,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;4BACxD,YAAY,GAAG,YAAY,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;4BAChG,YAAY,GAAG,KAAK,CAAC;wBACvB,CAAC;wBAED,0BAA0B;wBAC1B,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAEvC,wBAAwB;wBACxB,YAAY,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;wBAEjC,gBAAgB;wBAChB,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BACrB,YAAY,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;wBAC9B,CAAC;wBAED,4BAA4B;wBAC5B,OAAO,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BAC/B,mBAAmB;4BACnB,MAAM,UAAU,GAAG,YAAY,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC;4BAC/C,kBAAkB,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;wBACxE,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,kBAAkB,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC;oBAC7E,CAAC;gBACH,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,IAAI,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAG,CAAC,CAAC,EAAE,CAAC;gBACnB,qBAAqB;gBAErB,eAAe;gBACf,IAAI,YAAY,EAAE,CAAC;oBACjB,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAClC,CAAC;gBAED,gBAAgB;gBAChB,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC5B,IAAI,QAAQ,GAAG,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBAErD,IAAI,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBACpC,OAAO;wBACP,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;wBACxE,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC;oBACnD,CAAC;gBACH,CAAC;gBAED,WAAW;gBACX,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC7C,MAAM,UAAU,GAAG,CAAC,KAAK,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC;oBACjD,MAAM,UAAU,GAAG,YAAY,CAAC,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;oBAC9D,kBAAkB,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;gBACxE,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;AAlNqB,8CAAiB;AAIlB;IADlB,IAAA,cAAS,EAAC,wDAA6B,CAAC;;wEACwC;4BAJ7D,iBAAiB;IADtC,IAAA,eAAU,GAAE;GACS,iBAAiB,CAkNtC"}
|
|
@@ -1,9 +1,11 @@
|
|
|
1
1
|
import { DeepSeekProvider } from '@ai-sdk/deepseek';
|
|
2
2
|
import { IAIBackServiceOption } from '@opensumi/ide-core-common';
|
|
3
|
+
import { ModelInfo } from '../../common';
|
|
3
4
|
import { BaseLanguageModel } from '../base-language-model';
|
|
4
5
|
export declare const DeepSeekModelIdentifier: unique symbol;
|
|
5
6
|
export declare class DeepSeekModel extends BaseLanguageModel {
|
|
6
7
|
protected initializeProvider(options: IAIBackServiceOption): DeepSeekProvider;
|
|
7
|
-
protected getModelIdentifier(provider: DeepSeekProvider): import("ai").LanguageModelV1;
|
|
8
|
+
protected getModelIdentifier(provider: DeepSeekProvider, modelId?: string): import("ai").LanguageModelV1;
|
|
9
|
+
protected getModelInfo(modelId: string): ModelInfo | undefined;
|
|
8
10
|
}
|
|
9
11
|
//# sourceMappingURL=deepseek-language-model.d.ts.map
|
|
@@ -1 +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;
|
|
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,SAAS,EAAkB,MAAM,cAAc,CAAC;AACzD,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,EAAE,OAAO,SAAkB;IAIlF,SAAS,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;CAG/D"}
|
|
@@ -5,6 +5,7 @@ const tslib_1 = require("tslib");
|
|
|
5
5
|
const deepseek_1 = require("@ai-sdk/deepseek");
|
|
6
6
|
const di_1 = require("@opensumi/di");
|
|
7
7
|
const ai_native_1 = require("@opensumi/ide-core-common/lib/settings/ai-native");
|
|
8
|
+
const common_1 = require("../../common");
|
|
8
9
|
const base_language_model_1 = require("../base-language-model");
|
|
9
10
|
exports.DeepSeekModelIdentifier = Symbol('DeepSeekModelIdentifier');
|
|
10
11
|
let DeepSeekModel = class DeepSeekModel extends base_language_model_1.BaseLanguageModel {
|
|
@@ -15,8 +16,11 @@ let DeepSeekModel = class DeepSeekModel extends base_language_model_1.BaseLangua
|
|
|
15
16
|
}
|
|
16
17
|
return (0, deepseek_1.createDeepSeek)({ apiKey });
|
|
17
18
|
}
|
|
18
|
-
getModelIdentifier(provider) {
|
|
19
|
-
return provider(
|
|
19
|
+
getModelIdentifier(provider, modelId = 'deepseek-chat') {
|
|
20
|
+
return provider(modelId);
|
|
21
|
+
}
|
|
22
|
+
getModelInfo(modelId) {
|
|
23
|
+
return common_1.deepSeekModels[modelId];
|
|
20
24
|
}
|
|
21
25
|
};
|
|
22
26
|
exports.DeepSeekModel = DeepSeekModel;
|
|
@@ -1 +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;
|
|
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,yCAAyD;AACzD,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,EAAE,OAAO,GAAG,eAAe;QAChF,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAES,YAAY,CAAC,OAAe;QACpC,OAAO,uBAAc,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;CACF,CAAA;AAjBY,sCAAa;wBAAb,aAAa;IADzB,IAAA,eAAU,GAAE;GACA,aAAa,CAiBzB"}
|
|
@@ -3,9 +3,8 @@ import { Server } from '@modelcontextprotocol/sdk/server/index.js';
|
|
|
3
3
|
import { RPCService } from '@opensumi/ide-connection';
|
|
4
4
|
import { ILogger } from '@opensumi/ide-core-common';
|
|
5
5
|
import { ISumiMCPServerBackend } from '../../common';
|
|
6
|
-
import { MCPServerDescription } from '../../common/mcp-server-manager';
|
|
6
|
+
import { IMCPServer, MCPServerDescription } from '../../common/mcp-server-manager';
|
|
7
7
|
import { IMCPServerProxyService, MCPTool } from '../../common/types';
|
|
8
|
-
import { IMCPServer } from '../mcp-server';
|
|
9
8
|
export declare class SumiMCPServerBackend extends RPCService<IMCPServerProxyService> implements ISumiMCPServerBackend {
|
|
10
9
|
private readonly mcpServerManager;
|
|
11
10
|
private readonly toolInvocationRegistryManager;
|
|
@@ -46,7 +45,7 @@ export declare class SumiMCPServerBackend extends RPCService<IMCPServerProxyServ
|
|
|
46
45
|
} | undefined;
|
|
47
46
|
}> | undefined;
|
|
48
47
|
getAllMCPTools(): Promise<MCPTool[]>;
|
|
49
|
-
initBuiltinMCPServer(): Promise<void>;
|
|
48
|
+
initBuiltinMCPServer(enabled: boolean): Promise<void>;
|
|
50
49
|
initExternalMCPServers(servers: MCPServerDescription[]): Promise<void>;
|
|
51
50
|
initExposedMCPServer(): Promise<Server<{
|
|
52
51
|
method: string;
|
|
@@ -71,6 +70,21 @@ export declare class SumiMCPServerBackend extends RPCService<IMCPServerProxyServ
|
|
|
71
70
|
[x: string]: unknown;
|
|
72
71
|
} | undefined;
|
|
73
72
|
}>>;
|
|
73
|
+
getServers(): Promise<({
|
|
74
|
+
name: string;
|
|
75
|
+
isStarted: boolean;
|
|
76
|
+
type: string;
|
|
77
|
+
tools: string[];
|
|
78
|
+
command?: undefined;
|
|
79
|
+
} | {
|
|
80
|
+
name: string;
|
|
81
|
+
isStarted: boolean;
|
|
82
|
+
type: string;
|
|
83
|
+
command: string;
|
|
84
|
+
tools: string[];
|
|
85
|
+
})[]>;
|
|
86
|
+
startServer(serverName: string): Promise<void>;
|
|
87
|
+
stopServer(serverName: string): Promise<void>;
|
|
74
88
|
}
|
|
75
89
|
export declare const TokenBuiltinMCPServer: unique symbol;
|
|
76
90
|
export declare class BuiltinMCPServer implements IMCPServer {
|
|
@@ -1 +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,
|
|
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,EAA2B,qBAAqB,EAAE,MAAM,cAAc,CAAC;AAC9E,OAAO,EAAE,UAAU,EAAE,oBAAoB,EAAE,MAAM,iCAAiC,CAAC;AAEnF,OAAO,EAAE,sBAAsB,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AASrE,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;IASX,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG;;;;;;;IAOzC,SAAS;;;;;;;;;;;;;;;;;;;;;;;IAKH,cAAc,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAU7B,oBAAoB,CAAC,OAAO,EAAE,OAAO;IAOrC,sBAAsB,CAAC,OAAO,EAAE,oBAAoB,EAAE;IAM7D,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;IAqCpB,UAAU;;;;;;;;;;;;;IAkDV,WAAW,CAAC,UAAU,EAAE,MAAM;IAK9B,UAAU,CAAC,UAAU,EAAE,MAAM;CAIpC;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,CAAkB;gBAEJ,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,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC;IAqBhF,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"}
|
|
@@ -10,7 +10,9 @@ const di_1 = require("@opensumi/di");
|
|
|
10
10
|
const ide_connection_1 = require("@opensumi/ide-connection");
|
|
11
11
|
const ide_core_common_1 = require("@opensumi/ide-core-common");
|
|
12
12
|
const ide_core_node_1 = require("@opensumi/ide-core-node");
|
|
13
|
+
const common_1 = require("../../common");
|
|
13
14
|
const tool_invocation_registry_1 = require("../../common/tool-invocation-registry");
|
|
15
|
+
const mcp_server_1 = require("../mcp-server");
|
|
14
16
|
const mcp_server_manager_impl_1 = require("../mcp-server-manager-impl");
|
|
15
17
|
// 每个 BrowserTab 都对应了一个 SumiMCPServerBackend 实例
|
|
16
18
|
// SumiMCPServerBackend 需要做的事情:
|
|
@@ -33,7 +35,6 @@ let SumiMCPServerBackend = class SumiMCPServerBackend extends ide_connection_1.R
|
|
|
33
35
|
}
|
|
34
36
|
// 获取 MCP 工具
|
|
35
37
|
const tools = await this.client.$getMCPTools();
|
|
36
|
-
this.logger.log('[Node backend] SUMI MCP tools', tools);
|
|
37
38
|
return tools;
|
|
38
39
|
}
|
|
39
40
|
async callMCPTool(name, args) {
|
|
@@ -55,11 +56,11 @@ let SumiMCPServerBackend = class SumiMCPServerBackend extends ide_connection_1.R
|
|
|
55
56
|
providerName: tool.providerName || 'no-provider-name',
|
|
56
57
|
}));
|
|
57
58
|
}
|
|
58
|
-
async initBuiltinMCPServer() {
|
|
59
|
+
async initBuiltinMCPServer(enabled) {
|
|
59
60
|
var _a;
|
|
60
61
|
const builtinMCPServer = new BuiltinMCPServer(this, this.logger);
|
|
61
62
|
this.mcpServerManager.setClientId(this.clientId);
|
|
62
|
-
await this.mcpServerManager.initBuiltinServer(builtinMCPServer);
|
|
63
|
+
await this.mcpServerManager.initBuiltinServer(builtinMCPServer, enabled);
|
|
63
64
|
(_a = this.client) === null || _a === void 0 ? void 0 : _a.$updateMCPServers();
|
|
64
65
|
}
|
|
65
66
|
async initExternalMCPServers(servers) {
|
|
@@ -99,6 +100,59 @@ let SumiMCPServerBackend = class SumiMCPServerBackend extends ide_connection_1.R
|
|
|
99
100
|
});
|
|
100
101
|
return this.server;
|
|
101
102
|
}
|
|
103
|
+
async getServers() {
|
|
104
|
+
const servers = Array.from(this.mcpServerManager.getServers().entries());
|
|
105
|
+
const serverInfos = await Promise.all(servers.map(async ([serverName, server]) => {
|
|
106
|
+
var _a;
|
|
107
|
+
let toolNames = [];
|
|
108
|
+
if (server.isStarted()) {
|
|
109
|
+
// 只获取正在运行的 MCP Server 的工具列表
|
|
110
|
+
const toolsResponse = await server.getTools();
|
|
111
|
+
toolNames = toolsResponse.tools.map((tool) => tool.name);
|
|
112
|
+
}
|
|
113
|
+
// OpenSumi 内置的 MCP Server
|
|
114
|
+
if (serverName === common_1.BUILTIN_MCP_SERVER_NAME) {
|
|
115
|
+
return {
|
|
116
|
+
name: server.getServerName(),
|
|
117
|
+
isStarted: server.isStarted(),
|
|
118
|
+
type: 'builtin rpc',
|
|
119
|
+
tools: toolNames,
|
|
120
|
+
};
|
|
121
|
+
}
|
|
122
|
+
// 第三方 Stdio 类型的 MCP Server
|
|
123
|
+
if (server instanceof mcp_server_1.StdioMCPServerImpl) {
|
|
124
|
+
return {
|
|
125
|
+
name: server.getServerName(),
|
|
126
|
+
isStarted: server.isStarted(),
|
|
127
|
+
type: 'stdio',
|
|
128
|
+
command: server.command + ' ' + (((_a = server.args) === null || _a === void 0 ? void 0 : _a.join(' ')) || ''),
|
|
129
|
+
tools: toolNames,
|
|
130
|
+
};
|
|
131
|
+
}
|
|
132
|
+
// TODO SSE 类型的 MCP Server
|
|
133
|
+
return {
|
|
134
|
+
name: server.getServerName(),
|
|
135
|
+
isStarted: server.isStarted(),
|
|
136
|
+
type: '[MOCK] stdio',
|
|
137
|
+
command: '[MOCK] npx sumi-ide-mcp-server',
|
|
138
|
+
tools: toolNames,
|
|
139
|
+
};
|
|
140
|
+
}));
|
|
141
|
+
// 将 builtin server 放在第一位
|
|
142
|
+
const builtinServer = serverInfos.find((server) => server.name === common_1.BUILTIN_MCP_SERVER_NAME);
|
|
143
|
+
const otherServers = serverInfos.filter((server) => server.name !== common_1.BUILTIN_MCP_SERVER_NAME);
|
|
144
|
+
return builtinServer ? [builtinServer, ...otherServers] : otherServers;
|
|
145
|
+
}
|
|
146
|
+
async startServer(serverName) {
|
|
147
|
+
var _a;
|
|
148
|
+
await this.mcpServerManager.startServer(serverName);
|
|
149
|
+
(_a = this.client) === null || _a === void 0 ? void 0 : _a.$updateMCPServers();
|
|
150
|
+
}
|
|
151
|
+
async stopServer(serverName) {
|
|
152
|
+
var _a;
|
|
153
|
+
await this.mcpServerManager.stopServer(serverName);
|
|
154
|
+
(_a = this.client) === null || _a === void 0 ? void 0 : _a.$updateMCPServers();
|
|
155
|
+
}
|
|
102
156
|
};
|
|
103
157
|
exports.SumiMCPServerBackend = SumiMCPServerBackend;
|
|
104
158
|
tslib_1.__decorate([
|
|
@@ -118,13 +172,13 @@ class BuiltinMCPServer {
|
|
|
118
172
|
constructor(sumiMCPServer, logger) {
|
|
119
173
|
this.sumiMCPServer = sumiMCPServer;
|
|
120
174
|
this.logger = logger;
|
|
121
|
-
this.started =
|
|
175
|
+
this.started = false;
|
|
122
176
|
}
|
|
123
177
|
isStarted() {
|
|
124
178
|
return this.started;
|
|
125
179
|
}
|
|
126
180
|
getServerName() {
|
|
127
|
-
return
|
|
181
|
+
return common_1.BUILTIN_MCP_SERVER_NAME;
|
|
128
182
|
}
|
|
129
183
|
async start() {
|
|
130
184
|
if (this.started) {
|
|
@@ -157,7 +211,6 @@ class BuiltinMCPServer {
|
|
|
157
211
|
throw new Error('MCP Server not started');
|
|
158
212
|
}
|
|
159
213
|
const tools = await this.sumiMCPServer.getMCPTools();
|
|
160
|
-
this.logger.debug('[BuiltinMCPServer] getTools', tools);
|
|
161
214
|
return { tools };
|
|
162
215
|
}
|
|
163
216
|
update(_command, _args, _env) {
|
|
@@ -1 +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;
|
|
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;AAEtD,yCAA8E;AAE9E,oFAAsH;AAEtH,8CAAmD;AACnD,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,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,CAAC,OAAgB;;QAChD,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,EAAE,OAAO,CAAC,CAAC;QACzE,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;IAED,KAAK,CAAC,UAAU;QACd,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QACzE,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CACnC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,UAAU,EAAE,MAAM,CAAC,EAAE,EAAE;;YACzC,IAAI,SAAS,GAAa,EAAE,CAAC;YAC7B,IAAI,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC;gBACvB,4BAA4B;gBAC5B,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAC9C,SAAS,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3D,CAAC;YAED,0BAA0B;YAC1B,IAAI,UAAU,KAAK,gCAAuB,EAAE,CAAC;gBAC3C,OAAO;oBACL,IAAI,EAAE,MAAM,CAAC,aAAa,EAAE;oBAC5B,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE;oBAC7B,IAAI,EAAE,aAAa;oBACnB,KAAK,EAAE,SAAS;iBACjB,CAAC;YACJ,CAAC;YAED,2BAA2B;YAC3B,IAAI,MAAM,YAAY,+BAAkB,EAAE,CAAC;gBACzC,OAAO;oBACL,IAAI,EAAE,MAAM,CAAC,aAAa,EAAE;oBAC5B,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE;oBAC7B,IAAI,EAAE,OAAO;oBACb,OAAO,EAAE,MAAM,CAAC,OAAO,GAAG,GAAG,GAAG,CAAC,CAAA,MAAA,MAAM,CAAC,IAAI,0CAAE,IAAI,CAAC,GAAG,CAAC,KAAI,EAAE,CAAC;oBAC9D,KAAK,EAAE,SAAS;iBACjB,CAAC;YACJ,CAAC;YAED,0BAA0B;YAE1B,OAAO;gBACL,IAAI,EAAE,MAAM,CAAC,aAAa,EAAE;gBAC5B,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE;gBAC7B,IAAI,EAAE,cAAc;gBACpB,OAAO,EAAE,gCAAgC;gBACzC,KAAK,EAAE,SAAS;aACjB,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;QAEF,yBAAyB;QACzB,MAAM,aAAa,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,gCAAuB,CAAC,CAAC;QAC5F,MAAM,YAAY,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,gCAAuB,CAAC,CAAC;QAC7F,OAAO,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,EAAE,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC;IACzE,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,UAAkB;;QAClC,MAAM,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QACpD,MAAA,IAAI,CAAC,MAAM,0CAAE,iBAAiB,EAAE,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,UAAkB;;QACjC,MAAM,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QACnD,MAAA,IAAI,CAAC,MAAM,0CAAE,iBAAiB,EAAE,CAAC;IACnC,CAAC;CACF,CAAA;AAtKY,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,CAsKhC;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,KAAK,CAAC;IAEoE,CAAC;IAEtG,SAAS;QACP,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,aAAa;QACX,OAAO,gCAAuB,CAAC;IACjC,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,EAAE,UAAkB;QACrE,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;YAC9C,GAAG,IAAI;YACP,UAAU;SACX,CAAC,CAAC;IACL,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,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;AA9DD,4CA8DC"}
|
|
@@ -1,15 +1,16 @@
|
|
|
1
1
|
import { ILogger } from '@opensumi/ide-core-common';
|
|
2
|
-
import { MCPServerDescription, MCPServerManager } from '../common/mcp-server-manager';
|
|
2
|
+
import { IMCPServer, MCPServerDescription, MCPServerManager } from '../common/mcp-server-manager';
|
|
3
3
|
import { IToolInvocationRegistryManager } from '../common/tool-invocation-registry';
|
|
4
4
|
import { BuiltinMCPServer } from './mcp/sumi-mcp-server';
|
|
5
|
-
import { IMCPServer } from './mcp-server';
|
|
6
5
|
export declare class MCPServerManagerImpl implements MCPServerManager {
|
|
7
6
|
private readonly toolInvocationRegistryManager;
|
|
8
7
|
private readonly logger;
|
|
9
8
|
protected servers: Map<string, IMCPServer>;
|
|
10
9
|
private clientId;
|
|
10
|
+
getServers(): Map<string, IMCPServer>;
|
|
11
11
|
constructor(toolInvocationRegistryManager: IToolInvocationRegistryManager, logger: ILogger);
|
|
12
12
|
setClientId(clientId: string): void;
|
|
13
|
+
private unregisterServerTools;
|
|
13
14
|
stopServer(serverName: string): Promise<void>;
|
|
14
15
|
getStartedServers(): Promise<string[]>;
|
|
15
16
|
callTool(serverName: string, toolName: string, toolCallId: string, arg_string: string): ReturnType<IMCPServer['callTool']>;
|
|
@@ -20,7 +21,7 @@ export declare class MCPServerManagerImpl implements MCPServerManager {
|
|
|
20
21
|
getTools(serverName: string): ReturnType<IMCPServer['getTools']>;
|
|
21
22
|
addOrUpdateServer(description: MCPServerDescription): void;
|
|
22
23
|
addOrUpdateServerDirectly(server: IMCPServer): void;
|
|
23
|
-
initBuiltinServer(builtinMCPServer: BuiltinMCPServer): Promise<void>;
|
|
24
|
+
initBuiltinServer(builtinMCPServer: BuiltinMCPServer, enabled?: boolean): Promise<void>;
|
|
24
25
|
addExternalMCPServers(servers: MCPServerDescription[]): Promise<void>;
|
|
25
26
|
removeServer(name: string): void;
|
|
26
27
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"mcp-server-manager-impl.d.ts","sourceRoot":"","sources":["../../src/node/mcp-server-manager-impl.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,2BAA2B,CAAC;AAEpD,OAAO,EAAE,oBAAoB,EAAE,gBAAgB,EAAW,MAAM,8BAA8B,CAAC;
|
|
1
|
+
{"version":3,"file":"mcp-server-manager-impl.d.ts","sourceRoot":"","sources":["../../src/node/mcp-server-manager-impl.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,2BAA2B,CAAC;AAEpD,OAAO,EAAE,UAAU,EAAE,oBAAoB,EAAE,gBAAgB,EAAW,MAAM,8BAA8B,CAAC;AAC3G,OAAO,EAAE,8BAA8B,EAAe,MAAM,oCAAoC,CAAC;AAGjG,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAGzD,qBAAa,oBAAqB,YAAW,gBAAgB;IAWzD,OAAO,CAAC,QAAQ,CAAC,6BAA6B;IAC9C,OAAO,CAAC,QAAQ,CAAC,MAAM;IAXzB,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAa;IAGvD,OAAO,CAAC,QAAQ,CAAS;IAEzB,UAAU,IAAI,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC;gBAKlB,6BAA6B,EAAE,8BAA8B,EAC7D,MAAM,EAAE,OAAO;IAGlC,WAAW,CAAC,QAAQ,EAAE,MAAM;IAI5B,OAAO,CAAC,qBAAqB;IAKvB,UAAU,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAW7C,iBAAiB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAU5C,QAAQ,CACN,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,UAAU,EAAE,MAAM,GACjB,UAAU,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IAQ/B,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAS9C,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;IAK7C,iBAAiB,CAAC,gBAAgB,EAAE,gBAAgB,EAAE,OAAO,GAAE,OAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ7F,qBAAqB,CAAC,OAAO,EAAE,oBAAoB,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAW3E,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;CAYjC"}
|
|
@@ -5,6 +5,9 @@ const utils_1 = require("../common/utils");
|
|
|
5
5
|
const mcp_server_1 = require("./mcp-server");
|
|
6
6
|
// 这应该是 Browser Tab 维度的,每个 Tab 对应一个 MCPServerManagerImpl
|
|
7
7
|
class MCPServerManagerImpl {
|
|
8
|
+
getServers() {
|
|
9
|
+
return this.servers;
|
|
10
|
+
}
|
|
8
11
|
constructor(toolInvocationRegistryManager, logger) {
|
|
9
12
|
this.toolInvocationRegistryManager = toolInvocationRegistryManager;
|
|
10
13
|
this.logger = logger;
|
|
@@ -13,13 +16,19 @@ class MCPServerManagerImpl {
|
|
|
13
16
|
setClientId(clientId) {
|
|
14
17
|
this.clientId = clientId;
|
|
15
18
|
}
|
|
19
|
+
unregisterServerTools(serverName) {
|
|
20
|
+
const registry = this.toolInvocationRegistryManager.getRegistry(this.clientId);
|
|
21
|
+
registry.unregisterProviderTools(serverName);
|
|
22
|
+
}
|
|
16
23
|
async stopServer(serverName) {
|
|
17
24
|
const server = this.servers.get(serverName);
|
|
18
25
|
if (!server) {
|
|
19
26
|
throw new Error(`MCP server "${serverName}" not found.`);
|
|
20
27
|
}
|
|
21
|
-
server.stop();
|
|
22
|
-
|
|
28
|
+
await server.stop();
|
|
29
|
+
// 停止服务器后,需要从注册表中移除该服务器的所有工具
|
|
30
|
+
this.unregisterServerTools(serverName);
|
|
31
|
+
this.logger.log(`MCP server "${serverName}" stopped and tools unregistered.`);
|
|
23
32
|
}
|
|
24
33
|
async getStartedServers() {
|
|
25
34
|
const startedServers = [];
|
|
@@ -43,6 +52,7 @@ class MCPServerManagerImpl {
|
|
|
43
52
|
throw new Error(`MCP server "${serverName}" not found.`);
|
|
44
53
|
}
|
|
45
54
|
await server.start();
|
|
55
|
+
await this.registerTools(serverName);
|
|
46
56
|
}
|
|
47
57
|
async getServerNames() {
|
|
48
58
|
return Array.from(this.servers.keys());
|
|
@@ -95,29 +105,39 @@ class MCPServerManagerImpl {
|
|
|
95
105
|
existingServer.update(command, args, env);
|
|
96
106
|
}
|
|
97
107
|
else {
|
|
98
|
-
const newServer = new mcp_server_1.
|
|
108
|
+
const newServer = new mcp_server_1.StdioMCPServerImpl(name, command, args, env, this.logger);
|
|
99
109
|
this.servers.set(name, newServer);
|
|
100
110
|
}
|
|
101
111
|
}
|
|
102
112
|
addOrUpdateServerDirectly(server) {
|
|
103
113
|
this.servers.set(server.getServerName(), server);
|
|
104
114
|
}
|
|
105
|
-
|
|
115
|
+
// enabled 为 true 时,会自动启动内置服务器, 并注册工具
|
|
116
|
+
async initBuiltinServer(builtinMCPServer, enabled = true) {
|
|
106
117
|
this.addOrUpdateServerDirectly(builtinMCPServer);
|
|
107
|
-
|
|
118
|
+
if (enabled) {
|
|
119
|
+
await builtinMCPServer.start();
|
|
120
|
+
await this.registerTools(builtinMCPServer.getServerName());
|
|
121
|
+
}
|
|
108
122
|
}
|
|
109
123
|
async addExternalMCPServers(servers) {
|
|
110
124
|
for (const server of servers) {
|
|
111
125
|
this.addOrUpdateServer(server);
|
|
126
|
+
if (!server.enabled) {
|
|
127
|
+
// 如果是 enabled 为 false 的 server,则不进行启动
|
|
128
|
+
continue;
|
|
129
|
+
}
|
|
112
130
|
await this.startServer(server.name);
|
|
113
|
-
await this.registerTools(server.name);
|
|
114
131
|
}
|
|
115
132
|
}
|
|
116
133
|
removeServer(name) {
|
|
117
134
|
const server = this.servers.get(name);
|
|
118
135
|
if (server) {
|
|
119
136
|
server.stop();
|
|
137
|
+
// 移除服务器时,也需要从注册表中移除该服务器的所有工具
|
|
138
|
+
this.unregisterServerTools(name);
|
|
120
139
|
this.servers.delete(name);
|
|
140
|
+
this.logger.log(`MCP server "${name}" removed and tools unregistered.`);
|
|
121
141
|
}
|
|
122
142
|
else {
|
|
123
143
|
this.logger.warn(`MCP server "${name}" not found.`);
|