@opensumi/ide-ai-native 3.7.2-next-1739848467.0 → 3.7.2-next-1739945875.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/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 +45 -0
- 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/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/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 +56 -0
- 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 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-server-manager.js","sourceRoot":"","sources":["../../src/common/mcp-server-manager.ts"],"names":[],"mappings":";;;AA4Ca,QAAA,gBAAgB,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC;AAC9C,QAAA,oBAAoB,GAAG,0BAA0B,CAAC"}
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
export declare const ToolParameterSchema: any;
|
|
3
|
+
export type ToolParameter = z.infer<typeof ToolParameterSchema>;
|
|
4
|
+
export interface ToolRequest {
|
|
5
|
+
id: string;
|
|
6
|
+
name: string;
|
|
7
|
+
parameters?: any;
|
|
8
|
+
description?: string;
|
|
9
|
+
handler: (arg_string: string) => Promise<any>;
|
|
10
|
+
providerName?: string;
|
|
11
|
+
}
|
|
12
|
+
export declare namespace ToolRequest {
|
|
13
|
+
function isToolParameter(obj: unknown): obj is ToolParameter;
|
|
14
|
+
}
|
|
15
|
+
export declare const ToolInvocationRegistry: unique symbol;
|
|
16
|
+
/**
|
|
17
|
+
* 为 Agent 提供的所有可用函数调用的注册表
|
|
18
|
+
*/
|
|
19
|
+
export interface ToolInvocationRegistry {
|
|
20
|
+
/**
|
|
21
|
+
* 在注册表中注册一个工具
|
|
22
|
+
*
|
|
23
|
+
* @param tool - 要注册的 `ToolRequest` 对象
|
|
24
|
+
*/
|
|
25
|
+
registerTool(tool: ToolRequest): void;
|
|
26
|
+
/**
|
|
27
|
+
* 从注册表中获取特定的 `ToolRequest`
|
|
28
|
+
*
|
|
29
|
+
* @param toolId - 要获取的工具的唯一标识符
|
|
30
|
+
* @returns 对应提供的工具 ID 的 `ToolRequest` 对象,
|
|
31
|
+
* 如果在注册表中找不到该工具,则返回 `undefined`
|
|
32
|
+
*/
|
|
33
|
+
getFunction(toolId: string): ToolRequest | undefined;
|
|
34
|
+
/**
|
|
35
|
+
* 从注册表中获取多个 `ToolRequest`
|
|
36
|
+
*
|
|
37
|
+
* @param toolIds - 要获取的工具 ID 列表
|
|
38
|
+
* @returns 指定工具 ID 的 `ToolRequest` 对象数组
|
|
39
|
+
* 如果找不到某个工具 ID,将在返回的数组中跳过该工具
|
|
40
|
+
*/
|
|
41
|
+
getFunctions(...toolIds: string[]): ToolRequest[];
|
|
42
|
+
/**
|
|
43
|
+
* 获取当前注册表中的所有 `ToolRequest`
|
|
44
|
+
*
|
|
45
|
+
* @returns 注册表中所有 `ToolRequest` 对象的数组
|
|
46
|
+
*/
|
|
47
|
+
getAllFunctions(): ToolRequest[];
|
|
48
|
+
/**
|
|
49
|
+
* 注销特定工具提供者的所有工具
|
|
50
|
+
*
|
|
51
|
+
* @param providerName - 要移除其工具的工具提供者名称(在 `ToolRequest` 中指定)
|
|
52
|
+
*/
|
|
53
|
+
unregisterAllTools(providerName: string): void;
|
|
54
|
+
}
|
|
55
|
+
export declare const ToolProvider: unique symbol;
|
|
56
|
+
export interface ToolProvider {
|
|
57
|
+
getTool(): ToolRequest;
|
|
58
|
+
}
|
|
59
|
+
export declare class ToolInvocationRegistryImpl implements ToolInvocationRegistry {
|
|
60
|
+
private tools;
|
|
61
|
+
unregisterAllTools(providerName: string): void;
|
|
62
|
+
getAllFunctions(): ToolRequest[];
|
|
63
|
+
registerTool(tool: ToolRequest): void;
|
|
64
|
+
getFunction(toolId: string): ToolRequest | undefined;
|
|
65
|
+
getFunctions(...toolIds: string[]): ToolRequest[];
|
|
66
|
+
}
|
|
67
|
+
/**
|
|
68
|
+
* 管理多个 ToolInvocationRegistry 实例的管理器,每个实例与一个 clientId 关联
|
|
69
|
+
*/
|
|
70
|
+
export interface IToolInvocationRegistryManager {
|
|
71
|
+
/**
|
|
72
|
+
* 获取或创建特定 clientId 的 ToolInvocationRegistry
|
|
73
|
+
*/
|
|
74
|
+
getRegistry(clientId: string): ToolInvocationRegistry;
|
|
75
|
+
/**
|
|
76
|
+
* 移除特定 clientId 的 ToolInvocationRegistry
|
|
77
|
+
*/
|
|
78
|
+
removeRegistry(clientId: string): void;
|
|
79
|
+
/**
|
|
80
|
+
* 检查特定 clientId 是否存在对应的注册表
|
|
81
|
+
*/
|
|
82
|
+
hasRegistry(clientId: string): boolean;
|
|
83
|
+
}
|
|
84
|
+
export declare const ToolInvocationRegistryManager: unique symbol;
|
|
85
|
+
export declare class ToolInvocationRegistryManagerImpl implements IToolInvocationRegistryManager {
|
|
86
|
+
private registries;
|
|
87
|
+
getRegistry(clientId: string): ToolInvocationRegistry;
|
|
88
|
+
removeRegistry(clientId: string): void;
|
|
89
|
+
hasRegistry(clientId: string): boolean;
|
|
90
|
+
}
|
|
91
|
+
//# sourceMappingURL=tool-invocation-registry.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-invocation-registry.d.ts","sourceRoot":"","sources":["../../src/common/tool-invocation-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAMxB,eAAO,MAAM,mBAAmB,KAO9B,CAAC;AAEH,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAEhE,MAAM,WAAW,WAAW;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,GAAG,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAC9C,YAAY,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,yBAAiB,WAAW,CAAC;IACzB,SAAgB,eAAe,CAAC,GAAG,EAAE,OAAO,GAAG,GAAG,IAAI,aAAa,CAElE;CACJ;AAED,eAAO,MAAM,sBAAsB,eAAmC,CAAC;AAEvE;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACnC;;;;OAIG;IACH,YAAY,CAAC,IAAI,EAAE,WAAW,GAAG,IAAI,CAAC;IAEtC;;;;;;OAMG;IACH,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS,CAAC;IAErD;;;;;;OAMG;IACH,YAAY,CAAC,GAAG,OAAO,EAAE,MAAM,EAAE,GAAG,WAAW,EAAE,CAAC;IAElD;;;;OAIG;IACH,eAAe,IAAI,WAAW,EAAE,CAAC;IAEjC;;;;OAIG;IACH,kBAAkB,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;CAClD;AAED,eAAO,MAAM,YAAY,eAAyB,CAAC;AACnD,MAAM,WAAW,YAAY;IACzB,OAAO,IAAI,WAAW,CAAC;CAC1B;AAED,qBAAa,0BAA2B,YAAW,sBAAsB;IACrE,OAAO,CAAC,KAAK,CAA4D;IAEzE,kBAAkB,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI;IAU9C,eAAe,IAAI,WAAW,EAAE;IAIhC,YAAY,CAAC,IAAI,EAAE,WAAW,GAAG,IAAI;IASrC,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAIpD,YAAY,CAAC,GAAG,OAAO,EAAE,MAAM,EAAE,GAAG,WAAW,EAAE;CAWpD;AAED;;GAEG;AACH,MAAM,WAAW,8BAA8B;IAC3C;;OAEG;IACH,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,sBAAsB,CAAC;IAEtD;;OAEG;IACH,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IAEvC;;OAEG;IACH,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC1C;AAED,eAAO,MAAM,6BAA6B,eAA0C,CAAC;AAErF,qBACa,iCAAkC,YAAW,8BAA8B;IACpF,OAAO,CAAC,UAAU,CAAkD;IAEpE,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,sBAAsB;IASrD,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAItC,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;CAGzC"}
|
|
@@ -0,0 +1,90 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.ToolInvocationRegistryManagerImpl = exports.ToolInvocationRegistryManager = exports.ToolInvocationRegistryImpl = exports.ToolProvider = exports.ToolInvocationRegistry = exports.ToolRequest = exports.ToolParameterSchema = void 0;
|
|
4
|
+
const tslib_1 = require("tslib");
|
|
5
|
+
const zod_1 = require("zod");
|
|
6
|
+
const di_1 = require("@opensumi/di");
|
|
7
|
+
exports.ToolParameterSchema = zod_1.z.object({
|
|
8
|
+
type: zod_1.z.enum(['string', 'number', 'boolean', 'object', 'array']),
|
|
9
|
+
description: zod_1.z.string().optional(),
|
|
10
|
+
enum: zod_1.z.array(zod_1.z.any()).optional(),
|
|
11
|
+
items: zod_1.z.lazy(() => exports.ToolParameterSchema).optional(),
|
|
12
|
+
properties: zod_1.z.record(zod_1.z.lazy(() => exports.ToolParameterSchema)).optional(),
|
|
13
|
+
required: zod_1.z.array(zod_1.z.string()).optional(),
|
|
14
|
+
});
|
|
15
|
+
var ToolRequest;
|
|
16
|
+
(function (ToolRequest) {
|
|
17
|
+
function isToolParameter(obj) {
|
|
18
|
+
return exports.ToolParameterSchema.safeParse(obj).success;
|
|
19
|
+
}
|
|
20
|
+
ToolRequest.isToolParameter = isToolParameter;
|
|
21
|
+
})(ToolRequest || (exports.ToolRequest = ToolRequest = {}));
|
|
22
|
+
exports.ToolInvocationRegistry = Symbol('ToolInvocationRegistry');
|
|
23
|
+
exports.ToolProvider = Symbol('ToolProvider');
|
|
24
|
+
class ToolInvocationRegistryImpl {
|
|
25
|
+
constructor() {
|
|
26
|
+
this.tools = new Map();
|
|
27
|
+
}
|
|
28
|
+
unregisterAllTools(providerName) {
|
|
29
|
+
const toolsToRemove = [];
|
|
30
|
+
for (const [id, tool] of this.tools.entries()) {
|
|
31
|
+
if (tool.providerName === providerName) {
|
|
32
|
+
toolsToRemove.push(id);
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
toolsToRemove.forEach((id) => this.tools.delete(id));
|
|
36
|
+
}
|
|
37
|
+
getAllFunctions() {
|
|
38
|
+
return Array.from(this.tools.values());
|
|
39
|
+
}
|
|
40
|
+
registerTool(tool) {
|
|
41
|
+
if (this.tools.has(tool.id)) {
|
|
42
|
+
// TODO: 使用适当的日志机制
|
|
43
|
+
this.tools.set(tool.id, tool);
|
|
44
|
+
}
|
|
45
|
+
else {
|
|
46
|
+
this.tools.set(tool.id, tool);
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
getFunction(toolId) {
|
|
50
|
+
return this.tools.get(toolId);
|
|
51
|
+
}
|
|
52
|
+
getFunctions(...toolIds) {
|
|
53
|
+
const tools = toolIds.map((toolId) => {
|
|
54
|
+
const tool = this.tools.get(toolId);
|
|
55
|
+
if (tool) {
|
|
56
|
+
return tool;
|
|
57
|
+
}
|
|
58
|
+
else {
|
|
59
|
+
throw new Error(`找不到 ID 为 ${toolId} 的函数`);
|
|
60
|
+
}
|
|
61
|
+
});
|
|
62
|
+
return tools;
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
exports.ToolInvocationRegistryImpl = ToolInvocationRegistryImpl;
|
|
66
|
+
exports.ToolInvocationRegistryManager = Symbol('ToolInvocationRegistryManager');
|
|
67
|
+
let ToolInvocationRegistryManagerImpl = class ToolInvocationRegistryManagerImpl {
|
|
68
|
+
constructor() {
|
|
69
|
+
this.registries = new Map();
|
|
70
|
+
}
|
|
71
|
+
getRegistry(clientId) {
|
|
72
|
+
let registry = this.registries.get(clientId);
|
|
73
|
+
if (!registry) {
|
|
74
|
+
registry = new ToolInvocationRegistryImpl();
|
|
75
|
+
this.registries.set(clientId, registry);
|
|
76
|
+
}
|
|
77
|
+
return registry;
|
|
78
|
+
}
|
|
79
|
+
removeRegistry(clientId) {
|
|
80
|
+
this.registries.delete(clientId);
|
|
81
|
+
}
|
|
82
|
+
hasRegistry(clientId) {
|
|
83
|
+
return this.registries.has(clientId);
|
|
84
|
+
}
|
|
85
|
+
};
|
|
86
|
+
exports.ToolInvocationRegistryManagerImpl = ToolInvocationRegistryManagerImpl;
|
|
87
|
+
exports.ToolInvocationRegistryManagerImpl = ToolInvocationRegistryManagerImpl = tslib_1.__decorate([
|
|
88
|
+
(0, di_1.Injectable)()
|
|
89
|
+
], ToolInvocationRegistryManagerImpl);
|
|
90
|
+
//# sourceMappingURL=tool-invocation-registry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-invocation-registry.js","sourceRoot":"","sources":["../../src/common/tool-invocation-registry.ts"],"names":[],"mappings":";;;;AAAA,6BAAwB;AAExB,qCAA0C;AAI7B,QAAA,mBAAmB,GAAG,OAAC,CAAC,MAAM,CAAC;IACxC,IAAI,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IAChE,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC,CAAC,QAAQ,EAAE;IACjC,KAAK,EAAE,OAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,2BAAmB,CAAC,CAAC,QAAQ,EAAE;IACnD,UAAU,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,2BAAmB,CAAC,CAAC,CAAC,QAAQ,EAAE;IAClE,QAAQ,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;CAC3C,CAAC,CAAC;AAaH,IAAiB,WAAW,CAI3B;AAJD,WAAiB,WAAW;IACxB,SAAgB,eAAe,CAAC,GAAY;QACxC,OAAO,2BAAmB,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC;IACtD,CAAC;IAFe,2BAAe,kBAE9B,CAAA;AACL,CAAC,EAJgB,WAAW,2BAAX,WAAW,QAI3B;AAEY,QAAA,sBAAsB,GAAG,MAAM,CAAC,wBAAwB,CAAC,CAAC;AA8C1D,QAAA,YAAY,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;AAKnD,MAAa,0BAA0B;IAAvC;QACY,UAAK,GAA6B,IAAI,GAAG,EAAuB,CAAC;IAwC7E,CAAC;IAtCG,kBAAkB,CAAC,YAAoB;QACnC,MAAM,aAAa,GAAa,EAAE,CAAC;QACnC,KAAK,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,YAAY,KAAK,YAAY,EAAE,CAAC;gBACrC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3B,CAAC;QACL,CAAC;QACD,aAAa,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACzD,CAAC;IAED,eAAe;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,YAAY,CAAC,IAAiB;QAC1B,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YAC1B,kBAAkB;YAClB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAClC,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAClC,CAAC;IACL,CAAC;IAED,WAAW,CAAC,MAAc;QACtB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IAED,YAAY,CAAC,GAAG,OAAiB;QAC7B,MAAM,KAAK,GAAkB,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YAChD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACpC,IAAI,IAAI,EAAE,CAAC;gBACP,OAAO,IAAI,CAAC;YAChB,CAAC;iBAAM,CAAC;gBACJ,MAAM,IAAI,KAAK,CAAC,YAAY,MAAM,MAAM,CAAC,CAAC;YAC9C,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,KAAK,CAAC;IACjB,CAAC;CACJ;AAzCD,gEAyCC;AAsBY,QAAA,6BAA6B,GAAG,MAAM,CAAC,+BAA+B,CAAC,CAAC;AAG9E,IAAM,iCAAiC,GAAvC,MAAM,iCAAiC;IAAvC;QACK,eAAU,GAAwC,IAAI,GAAG,EAAE,CAAC;IAkBxE,CAAC;IAhBG,WAAW,CAAC,QAAgB;QACxB,IAAI,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,QAAQ,GAAG,IAAI,0BAA0B,EAAE,CAAC;YAC5C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,cAAc,CAAC,QAAgB;QAC3B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;IAED,WAAW,CAAC,QAAgB;QACxB,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC;CACJ,CAAA;AAnBY,8EAAiC;4CAAjC,iCAAiC;IAD7C,IAAA,eAAU,GAAE;GACA,iCAAiC,CAmB7C"}
|
package/lib/common/types.d.ts
CHANGED
|
@@ -17,4 +17,21 @@ export interface INearestCodeBlock {
|
|
|
17
17
|
offset: number;
|
|
18
18
|
type?: NearestCodeBlockType;
|
|
19
19
|
}
|
|
20
|
+
export interface IMCPServerProxyService {
|
|
21
|
+
$callMCPTool(name: string, args: any): Promise<{
|
|
22
|
+
content: {
|
|
23
|
+
type: string;
|
|
24
|
+
text: string;
|
|
25
|
+
}[];
|
|
26
|
+
isError?: boolean;
|
|
27
|
+
}>;
|
|
28
|
+
$getMCPTools(): Promise<MCPTool[]>;
|
|
29
|
+
$updateMCPServers(): Promise<void>;
|
|
30
|
+
}
|
|
31
|
+
export interface MCPTool {
|
|
32
|
+
name: string;
|
|
33
|
+
description: string;
|
|
34
|
+
inputSchema: any;
|
|
35
|
+
providerName: string;
|
|
36
|
+
}
|
|
20
37
|
//# sourceMappingURL=types.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/common/types.ts"],"names":[],"mappings":"AAAA,oBAAY,oBAAoB;IAC9B,KAAK,UAAU;IACf,IAAI,SAAS;CACd;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE;QACL,KAAK,EAAE;YACL,IAAI,EAAE,MAAM,CAAC;YACb,SAAS,EAAE,MAAM,CAAC;SACnB,CAAC;QACF,GAAG,EAAE;YACH,IAAI,EAAE,MAAM,CAAC;YACb,SAAS,EAAE,MAAM,CAAC;SACnB,CAAC;KACH,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,oBAAoB,CAAC;CAC7B"}
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/common/types.ts"],"names":[],"mappings":"AAAA,oBAAY,oBAAoB;IAC9B,KAAK,UAAU;IACf,IAAI,SAAS;CACd;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE;QACL,KAAK,EAAE;YACL,IAAI,EAAE,MAAM,CAAC;YACb,SAAS,EAAE,MAAM,CAAC;SACnB,CAAC;QACF,GAAG,EAAE;YACH,IAAI,EAAE,MAAM,CAAC;YACb,SAAS,EAAE,MAAM,CAAC;SACnB,CAAC;KACH,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,oBAAoB,CAAC;CAC7B;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;CACpC;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"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { AnthropicProvider } from '@ai-sdk/anthropic';
|
|
2
|
+
import { IAIBackServiceOption } from '@opensumi/ide-core-common';
|
|
3
|
+
import { BaseLanguageModel } from '../base-language-model';
|
|
4
|
+
export declare const AnthropicModelIdentifier: unique symbol;
|
|
5
|
+
export declare class AnthropicModel extends BaseLanguageModel {
|
|
6
|
+
protected initializeProvider(options: IAIBackServiceOption): AnthropicProvider;
|
|
7
|
+
protected getModelIdentifier(provider: AnthropicProvider): import("ai").LanguageModelV1;
|
|
8
|
+
}
|
|
9
|
+
//# sourceMappingURL=anthropic-language-model.d.ts.map
|
|
@@ -0,0 +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;CAGzD"}
|
|
@@ -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"}
|