@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.
Files changed (227) hide show
  1. package/lib/browser/ai-core.contribution.d.ts +3 -0
  2. package/lib/browser/ai-core.contribution.d.ts.map +1 -1
  3. package/lib/browser/ai-core.contribution.js +68 -2
  4. package/lib/browser/ai-core.contribution.js.map +1 -1
  5. package/lib/browser/chat/chat-model.d.ts +2 -2
  6. package/lib/browser/chat/chat-model.d.ts.map +1 -1
  7. package/lib/browser/chat/chat-model.js +16 -5
  8. package/lib/browser/chat/chat-model.js.map +1 -1
  9. package/lib/browser/chat/chat-proxy.service.d.ts +4 -0
  10. package/lib/browser/chat/chat-proxy.service.d.ts.map +1 -1
  11. package/lib/browser/chat/chat-proxy.service.js +43 -0
  12. package/lib/browser/chat/chat-proxy.service.js.map +1 -1
  13. package/lib/browser/chat/chat.view.d.ts.map +1 -1
  14. package/lib/browser/chat/chat.view.js +29 -2
  15. package/lib/browser/chat/chat.view.js.map +1 -1
  16. package/lib/browser/components/ChatContext/ContextSelector.d.ts +12 -0
  17. package/lib/browser/components/ChatContext/ContextSelector.d.ts.map +1 -0
  18. package/lib/browser/components/ChatContext/ContextSelector.js +113 -0
  19. package/lib/browser/components/ChatContext/ContextSelector.js.map +1 -0
  20. package/lib/browser/components/ChatContext/index.d.ts +4 -0
  21. package/lib/browser/components/ChatContext/index.d.ts.map +1 -0
  22. package/lib/browser/components/ChatContext/index.js +84 -0
  23. package/lib/browser/components/ChatContext/index.js.map +1 -0
  24. package/lib/browser/components/ChatContext/style.module.less +189 -0
  25. package/lib/browser/components/ChatInput.d.ts.map +1 -1
  26. package/lib/browser/components/ChatInput.js.map +1 -1
  27. package/lib/browser/components/ChatReply.d.ts.map +1 -1
  28. package/lib/browser/components/ChatReply.js +25 -0
  29. package/lib/browser/components/ChatReply.js.map +1 -1
  30. package/lib/browser/components/ChatToolRender.d.ts +6 -0
  31. package/lib/browser/components/ChatToolRender.d.ts.map +1 -0
  32. package/lib/browser/components/ChatToolRender.js +53 -0
  33. package/lib/browser/components/ChatToolRender.js.map +1 -0
  34. package/lib/browser/components/ChatToolRender.module.less +86 -0
  35. package/lib/browser/components/components.module.less +32 -31
  36. package/lib/browser/components/utils.d.ts +2 -2
  37. package/lib/browser/context/llm-context.contribution.d.ts +7 -0
  38. package/lib/browser/context/llm-context.contribution.d.ts.map +1 -0
  39. package/lib/browser/context/llm-context.contribution.js +21 -0
  40. package/lib/browser/context/llm-context.contribution.js.map +1 -0
  41. package/lib/browser/context/llm-context.service.d.ts +24 -0
  42. package/lib/browser/context/llm-context.service.d.ts.map +1 -0
  43. package/lib/browser/context/llm-context.service.js +136 -0
  44. package/lib/browser/context/llm-context.service.js.map +1 -0
  45. package/lib/browser/index.d.ts +11 -3
  46. package/lib/browser/index.d.ts.map +1 -1
  47. package/lib/browser/index.js +56 -3
  48. package/lib/browser/index.js.map +1 -1
  49. package/lib/browser/mcp/mcp-server-proxy.service.d.ts +25 -0
  50. package/lib/browser/mcp/mcp-server-proxy.service.d.ts.map +1 -0
  51. package/lib/browser/mcp/mcp-server-proxy.service.js +56 -0
  52. package/lib/browser/mcp/mcp-server-proxy.service.js.map +1 -0
  53. package/lib/browser/mcp/mcp-server.feature.registry.d.ts +16 -0
  54. package/lib/browser/mcp/mcp-server.feature.registry.d.ts.map +1 -0
  55. package/lib/browser/mcp/mcp-server.feature.registry.js +53 -0
  56. package/lib/browser/mcp/mcp-server.feature.registry.js.map +1 -0
  57. package/lib/browser/mcp/mcp-tools-dialog.module.less +44 -0
  58. package/lib/browser/mcp/mcp-tools-dialog.view.d.ts +8 -0
  59. package/lib/browser/mcp/mcp-tools-dialog.view.d.ts.map +1 -0
  60. package/lib/browser/mcp/mcp-tools-dialog.view.js +16 -0
  61. package/lib/browser/mcp/mcp-tools-dialog.view.js.map +1 -0
  62. package/lib/browser/mcp/tools/createNewFileWithText.d.ts +9 -0
  63. package/lib/browser/mcp/tools/createNewFileWithText.d.ts.map +1 -0
  64. package/lib/browser/mcp/tools/createNewFileWithText.js +83 -0
  65. package/lib/browser/mcp/tools/createNewFileWithText.js.map +1 -0
  66. package/lib/browser/mcp/tools/findFilesByNameSubstring.d.ts +9 -0
  67. package/lib/browser/mcp/tools/findFilesByNameSubstring.d.ts.map +1 -0
  68. package/lib/browser/mcp/tools/findFilesByNameSubstring.js +92 -0
  69. package/lib/browser/mcp/tools/findFilesByNameSubstring.js.map +1 -0
  70. package/lib/browser/mcp/tools/getCurrentFilePath.d.ts +8 -0
  71. package/lib/browser/mcp/tools/getCurrentFilePath.d.ts.map +1 -0
  72. package/lib/browser/mcp/tools/getCurrentFilePath.js +49 -0
  73. package/lib/browser/mcp/tools/getCurrentFilePath.js.map +1 -0
  74. package/lib/browser/mcp/tools/getDiagnosticsByPath.d.ts +10 -0
  75. package/lib/browser/mcp/tools/getDiagnosticsByPath.d.ts.map +1 -0
  76. package/lib/browser/mcp/tools/getDiagnosticsByPath.js +119 -0
  77. package/lib/browser/mcp/tools/getDiagnosticsByPath.js.map +1 -0
  78. package/lib/browser/mcp/tools/getFileTextByPath.d.ts +9 -0
  79. package/lib/browser/mcp/tools/getFileTextByPath.d.ts.map +1 -0
  80. package/lib/browser/mcp/tools/getFileTextByPath.js +97 -0
  81. package/lib/browser/mcp/tools/getFileTextByPath.js.map +1 -0
  82. package/lib/browser/mcp/tools/getOpenEditorFileDiagnostics.d.ts +11 -0
  83. package/lib/browser/mcp/tools/getOpenEditorFileDiagnostics.d.ts.map +1 -0
  84. package/lib/browser/mcp/tools/getOpenEditorFileDiagnostics.js +119 -0
  85. package/lib/browser/mcp/tools/getOpenEditorFileDiagnostics.js.map +1 -0
  86. package/lib/browser/mcp/tools/getOpenEditorFileText.d.ts +8 -0
  87. package/lib/browser/mcp/tools/getOpenEditorFileText.d.ts.map +1 -0
  88. package/lib/browser/mcp/tools/getOpenEditorFileText.js +50 -0
  89. package/lib/browser/mcp/tools/getOpenEditorFileText.js.map +1 -0
  90. package/lib/browser/mcp/tools/getSelectedText.d.ts +8 -0
  91. package/lib/browser/mcp/tools/getSelectedText.d.ts.map +1 -0
  92. package/lib/browser/mcp/tools/getSelectedText.js +57 -0
  93. package/lib/browser/mcp/tools/getSelectedText.js.map +1 -0
  94. package/lib/browser/mcp/tools/handlers/ListDir.d.ts +21 -0
  95. package/lib/browser/mcp/tools/handlers/ListDir.d.ts.map +1 -0
  96. package/lib/browser/mcp/tools/handlers/ListDir.js +112 -0
  97. package/lib/browser/mcp/tools/handlers/ListDir.js.map +1 -0
  98. package/lib/browser/mcp/tools/handlers/ReadFile.d.ts +47 -0
  99. package/lib/browser/mcp/tools/handlers/ReadFile.d.ts.map +1 -0
  100. package/lib/browser/mcp/tools/handlers/ReadFile.js +147 -0
  101. package/lib/browser/mcp/tools/handlers/ReadFile.js.map +1 -0
  102. package/lib/browser/mcp/tools/listDir.d.ts +8 -0
  103. package/lib/browser/mcp/tools/listDir.d.ts.map +1 -0
  104. package/lib/browser/mcp/tools/listDir.js +65 -0
  105. package/lib/browser/mcp/tools/listDir.js.map +1 -0
  106. package/lib/browser/mcp/tools/readFile.d.ts +8 -0
  107. package/lib/browser/mcp/tools/readFile.d.ts.map +1 -0
  108. package/lib/browser/mcp/tools/readFile.js +82 -0
  109. package/lib/browser/mcp/tools/readFile.js.map +1 -0
  110. package/lib/browser/mcp/tools/replaceOpenEditorFile.d.ts +8 -0
  111. package/lib/browser/mcp/tools/replaceOpenEditorFile.d.ts.map +1 -0
  112. package/lib/browser/mcp/tools/replaceOpenEditorFile.js +79 -0
  113. package/lib/browser/mcp/tools/replaceOpenEditorFile.js.map +1 -0
  114. package/lib/browser/mcp/tools/replaceOpenEditorFileByDiffPreviewer.d.ts +8 -0
  115. package/lib/browser/mcp/tools/replaceOpenEditorFileByDiffPreviewer.d.ts.map +1 -0
  116. package/lib/browser/mcp/tools/replaceOpenEditorFileByDiffPreviewer.js +84 -0
  117. package/lib/browser/mcp/tools/replaceOpenEditorFileByDiffPreviewer.js.map +1 -0
  118. package/lib/browser/mcp/tools/runTerminalCmd.d.ts +18 -0
  119. package/lib/browser/mcp/tools/runTerminalCmd.d.ts.map +1 -0
  120. package/lib/browser/mcp/tools/runTerminalCmd.js +96 -0
  121. package/lib/browser/mcp/tools/runTerminalCmd.js.map +1 -0
  122. package/lib/browser/preferences/schema.d.ts.map +1 -1
  123. package/lib/browser/preferences/schema.js +60 -0
  124. package/lib/browser/preferences/schema.js.map +1 -1
  125. package/lib/browser/types.d.ts +45 -0
  126. package/lib/browser/types.d.ts.map +1 -1
  127. package/lib/browser/types.js +5 -1
  128. package/lib/browser/types.js.map +1 -1
  129. package/lib/common/index.d.ts +9 -0
  130. package/lib/common/index.d.ts.map +1 -1
  131. package/lib/common/index.js +4 -1
  132. package/lib/common/index.js.map +1 -1
  133. package/lib/common/llm-context.d.ts +37 -0
  134. package/lib/common/llm-context.d.ts.map +1 -0
  135. package/lib/common/llm-context.js +5 -0
  136. package/lib/common/llm-context.js.map +1 -0
  137. package/lib/common/mcp-server-manager.d.ts +40 -0
  138. package/lib/common/mcp-server-manager.d.ts.map +1 -0
  139. package/lib/common/mcp-server-manager.js +6 -0
  140. package/lib/common/mcp-server-manager.js.map +1 -0
  141. package/lib/common/tool-invocation-registry.d.ts +91 -0
  142. package/lib/common/tool-invocation-registry.d.ts.map +1 -0
  143. package/lib/common/tool-invocation-registry.js +90 -0
  144. package/lib/common/tool-invocation-registry.js.map +1 -0
  145. package/lib/common/types.d.ts +17 -0
  146. package/lib/common/types.d.ts.map +1 -1
  147. package/lib/node/anthropic/anthropic-language-model.d.ts +9 -0
  148. package/lib/node/anthropic/anthropic-language-model.d.ts.map +1 -0
  149. package/lib/node/anthropic/anthropic-language-model.js +26 -0
  150. package/lib/node/anthropic/anthropic-language-model.js.map +1 -0
  151. package/lib/node/base-language-model.d.ts +14 -0
  152. package/lib/node/base-language-model.d.ts.map +1 -0
  153. package/lib/node/base-language-model.js +136 -0
  154. package/lib/node/base-language-model.js.map +1 -0
  155. package/lib/node/deepseek/deepseek-language-model.d.ts +9 -0
  156. package/lib/node/deepseek/deepseek-language-model.d.ts.map +1 -0
  157. package/lib/node/deepseek/deepseek-language-model.js +26 -0
  158. package/lib/node/deepseek/deepseek-language-model.js.map +1 -0
  159. package/lib/node/index.d.ts.map +1 -1
  160. package/lib/node/index.js +19 -0
  161. package/lib/node/index.js.map +1 -1
  162. package/lib/node/mcp/sumi-mcp-server.d.ts +91 -0
  163. package/lib/node/mcp/sumi-mcp-server.d.ts.map +1 -0
  164. package/lib/node/mcp/sumi-mcp-server.js +172 -0
  165. package/lib/node/mcp/sumi-mcp-server.js.map +1 -0
  166. package/lib/node/mcp-server-manager-impl.d.ts +27 -0
  167. package/lib/node/mcp-server-manager-impl.d.ts.map +1 -0
  168. package/lib/node/mcp-server-manager-impl.js +127 -0
  169. package/lib/node/mcp-server-manager-impl.js.map +1 -0
  170. package/lib/node/mcp-server.d.ts +207 -0
  171. package/lib/node/mcp-server.d.ts.map +1 -0
  172. package/lib/node/mcp-server.js +91 -0
  173. package/lib/node/mcp-server.js.map +1 -0
  174. package/lib/node/openai/openai-language-model.d.ts +9 -0
  175. package/lib/node/openai/openai-language-model.d.ts.map +1 -0
  176. package/lib/node/openai/openai-language-model.js +29 -0
  177. package/lib/node/openai/openai-language-model.js.map +1 -0
  178. package/package.json +34 -22
  179. package/src/browser/ai-core.contribution.ts +77 -1
  180. package/src/browser/chat/chat-model.ts +24 -6
  181. package/src/browser/chat/chat-proxy.service.ts +42 -0
  182. package/src/browser/chat/chat.view.tsx +59 -6
  183. package/src/browser/components/ChatContext/ContextSelector.tsx +177 -0
  184. package/src/browser/components/ChatContext/index.tsx +135 -0
  185. package/src/browser/components/ChatContext/style.module.less +189 -0
  186. package/src/browser/components/ChatInput.tsx +1 -0
  187. package/src/browser/components/ChatReply.tsx +32 -0
  188. package/src/browser/components/ChatToolRender.module.less +86 -0
  189. package/src/browser/components/ChatToolRender.tsx +77 -0
  190. package/src/browser/components/components.module.less +32 -31
  191. package/src/browser/context/llm-context.contribution.ts +14 -0
  192. package/src/browser/context/llm-context.service.ts +156 -0
  193. package/src/browser/index.ts +68 -4
  194. package/src/browser/mcp/mcp-server-proxy.service.ts +53 -0
  195. package/src/browser/mcp/mcp-server.feature.registry.ts +54 -0
  196. package/src/browser/mcp/mcp-tools-dialog.module.less +44 -0
  197. package/src/browser/mcp/mcp-tools-dialog.view.tsx +24 -0
  198. package/src/browser/mcp/tools/createNewFileWithText.ts +83 -0
  199. package/src/browser/mcp/tools/findFilesByNameSubstring.ts +93 -0
  200. package/src/browser/mcp/tools/getCurrentFilePath.ts +49 -0
  201. package/src/browser/mcp/tools/getDiagnosticsByPath.ts +123 -0
  202. package/src/browser/mcp/tools/getFileTextByPath.ts +97 -0
  203. package/src/browser/mcp/tools/getOpenEditorFileDiagnostics.ts +121 -0
  204. package/src/browser/mcp/tools/getOpenEditorFileText.ts +50 -0
  205. package/src/browser/mcp/tools/getSelectedText.ts +57 -0
  206. package/src/browser/mcp/tools/handlers/ListDir.ts +117 -0
  207. package/src/browser/mcp/tools/handlers/ReadFile.ts +174 -0
  208. package/src/browser/mcp/tools/listDir.ts +66 -0
  209. package/src/browser/mcp/tools/readFile.ts +82 -0
  210. package/src/browser/mcp/tools/replaceOpenEditorFile.ts +80 -0
  211. package/src/browser/mcp/tools/replaceOpenEditorFileByDiffPreviewer.ts +91 -0
  212. package/src/browser/mcp/tools/runTerminalCmd.ts +107 -0
  213. package/src/browser/preferences/schema.ts +60 -0
  214. package/src/browser/types.ts +56 -0
  215. package/src/common/index.ts +14 -0
  216. package/src/common/llm-context.ts +41 -0
  217. package/src/common/mcp-server-manager.ts +46 -0
  218. package/src/common/tool-invocation-registry.ts +170 -0
  219. package/src/common/types.ts +22 -0
  220. package/src/node/anthropic/anthropic-language-model.ts +25 -0
  221. package/src/node/base-language-model.ts +163 -0
  222. package/src/node/deepseek/deepseek-language-model.ts +25 -0
  223. package/src/node/index.ts +21 -0
  224. package/src/node/mcp/sumi-mcp-server.ts +197 -0
  225. package/src/node/mcp-server-manager-impl.ts +148 -0
  226. package/src/node/mcp-server.ts +126 -0
  227. 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"}
@@ -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"}
@@ -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;AAGrD,qBACa,cAAe,SAAQ,UAAU;IAC5C,SAAS,EAAE,QAAQ,EAAE,CAKnB;IAEF,YAAY;;;QAKV;CACH"}
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
  };
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/node/index.ts"],"names":[],"mappings":";;;;AAAA,qCAAoD;AACpD,+DAAkF;AAClF,2DAAqD;AACrD,+FAA4F;AAGrF,IAAM,cAAc,GAApB,MAAM,cAAe,SAAQ,0BAAU;IAAvC;;QACL,cAAS,GAAe;YACtB;gBACE,KAAK,EAAE,oCAAkB;gBACzB,QAAQ,EAAE,qCAAiB;aAC5B;SACF,CAAC;QAEF,iBAAY,GAAG;YACb;gBACE,WAAW,EAAE,mCAAiB;gBAC9B,KAAK,EAAE,oCAAkB;aAC1B;SACF,CAAC;IACJ,CAAC;CAAA,CAAA;AAdY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,eAAU,GAAE;GACA,cAAc,CAc1B"}
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"}