@opensumi/ide-ai-native 3.0.3 → 3.0.4-next-1716367246.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 (77) hide show
  1. package/lib/browser/inline-completions/completeProvider.d.ts +5 -3
  2. package/lib/browser/inline-completions/completeProvider.d.ts.map +1 -1
  3. package/lib/browser/inline-completions/completeProvider.js +16 -30
  4. package/lib/browser/inline-completions/completeProvider.js.map +1 -1
  5. package/lib/browser/inline-completions/constants.d.ts +2 -6
  6. package/lib/browser/inline-completions/constants.d.ts.map +1 -1
  7. package/lib/browser/inline-completions/constants.js +81 -4
  8. package/lib/browser/inline-completions/constants.js.map +1 -1
  9. package/lib/browser/inline-completions/prompt/const.d.ts +9 -0
  10. package/lib/browser/inline-completions/prompt/const.d.ts.map +1 -0
  11. package/lib/browser/inline-completions/prompt/const.js +284 -0
  12. package/lib/browser/inline-completions/prompt/const.js.map +1 -0
  13. package/lib/browser/inline-completions/prompt/importedFiles.d.ts +6 -0
  14. package/lib/browser/inline-completions/prompt/importedFiles.d.ts.map +1 -0
  15. package/lib/browser/inline-completions/prompt/importedFiles.js +77 -0
  16. package/lib/browser/inline-completions/prompt/importedFiles.js.map +1 -0
  17. package/lib/browser/inline-completions/prompt/jaccardMatcher.d.ts +31 -0
  18. package/lib/browser/inline-completions/prompt/jaccardMatcher.d.ts.map +1 -0
  19. package/lib/browser/inline-completions/prompt/jaccardMatcher.js +75 -0
  20. package/lib/browser/inline-completions/prompt/jaccardMatcher.js.map +1 -0
  21. package/lib/browser/inline-completions/prompt/languages.d.ts +4 -0
  22. package/lib/browser/inline-completions/prompt/languages.d.ts.map +1 -0
  23. package/lib/browser/inline-completions/prompt/languages.js +67 -0
  24. package/lib/browser/inline-completions/prompt/languages.js.map +1 -0
  25. package/lib/browser/inline-completions/prompt/matcher.d.ts +42 -0
  26. package/lib/browser/inline-completions/prompt/matcher.d.ts.map +1 -0
  27. package/lib/browser/inline-completions/prompt/matcher.js +279 -0
  28. package/lib/browser/inline-completions/prompt/matcher.js.map +1 -0
  29. package/lib/browser/inline-completions/prompt/prompt.d.ts +24 -0
  30. package/lib/browser/inline-completions/prompt/prompt.d.ts.map +1 -0
  31. package/lib/browser/inline-completions/prompt/prompt.js +242 -0
  32. package/lib/browser/inline-completions/prompt/prompt.js.map +1 -0
  33. package/lib/browser/inline-completions/prompt/similarSnippets.d.ts +9 -0
  34. package/lib/browser/inline-completions/prompt/similarSnippets.d.ts.map +1 -0
  35. package/lib/browser/inline-completions/prompt/similarSnippets.js +110 -0
  36. package/lib/browser/inline-completions/prompt/similarSnippets.js.map +1 -0
  37. package/lib/browser/inline-completions/prompt/tokenizer.d.ts +4 -0
  38. package/lib/browser/inline-completions/prompt/tokenizer.d.ts.map +1 -0
  39. package/lib/browser/inline-completions/prompt/tokenizer.js +18 -0
  40. package/lib/browser/inline-completions/prompt/tokenizer.js.map +1 -0
  41. package/lib/browser/inline-completions/provider.d.ts +5 -12
  42. package/lib/browser/inline-completions/provider.d.ts.map +1 -1
  43. package/lib/browser/inline-completions/provider.js +79 -44
  44. package/lib/browser/inline-completions/provider.js.map +1 -1
  45. package/lib/browser/inline-completions/types.d.ts +154 -0
  46. package/lib/browser/inline-completions/types.d.ts.map +1 -0
  47. package/lib/browser/inline-completions/types.js +59 -0
  48. package/lib/browser/inline-completions/types.js.map +1 -0
  49. package/lib/browser/languages/parser.d.ts +6 -0
  50. package/lib/browser/languages/parser.d.ts.map +1 -1
  51. package/lib/browser/languages/parser.js +140 -1
  52. package/lib/browser/languages/parser.js.map +1 -1
  53. package/lib/browser/languages/tree-sitter/language-facts/types.d.ts +1 -1
  54. package/lib/browser/languages/tree-sitter/wasm-manager.d.ts +2 -2
  55. package/lib/browser/languages/tree-sitter/wasm-manager.d.ts.map +1 -1
  56. package/lib/browser/languages/tree-sitter/wasm-manager.js +2 -2
  57. package/lib/browser/languages/tree-sitter/wasm-manager.js.map +1 -1
  58. package/lib/common/utils.d.ts +5 -0
  59. package/lib/common/utils.d.ts.map +1 -0
  60. package/lib/common/utils.js +26 -0
  61. package/lib/common/utils.js.map +1 -0
  62. package/package.json +20 -19
  63. package/src/browser/inline-completions/completeProvider.ts +26 -35
  64. package/src/browser/inline-completions/constants.ts +87 -7
  65. package/src/browser/inline-completions/prompt/const.ts +286 -0
  66. package/src/browser/inline-completions/prompt/importedFiles.ts +92 -0
  67. package/src/browser/inline-completions/prompt/jaccardMatcher.ts +98 -0
  68. package/src/browser/inline-completions/prompt/languages.ts +65 -0
  69. package/src/browser/inline-completions/prompt/matcher.ts +328 -0
  70. package/src/browser/inline-completions/prompt/prompt.ts +297 -0
  71. package/src/browser/inline-completions/prompt/similarSnippets.ts +130 -0
  72. package/src/browser/inline-completions/prompt/tokenizer.ts +16 -0
  73. package/src/browser/inline-completions/provider.ts +107 -42
  74. package/src/browser/inline-completions/types.ts +169 -0
  75. package/src/browser/languages/parser.ts +147 -0
  76. package/src/browser/languages/tree-sitter/wasm-manager.ts +3 -3
  77. package/src/common/utils.ts +23 -0
@@ -0,0 +1,77 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.getImportedFile = exports.extractLocalImportContext = exports.getImportedFilesInterface = void 0;
4
+ const ide_core_common_1 = require("@opensumi/ide-core-common");
5
+ const ide_file_service_1 = require("@opensumi/ide-file-service");
6
+ const path_1 = require("@opensumi/ide-utils/lib/path");
7
+ const ide_workspace_1 = require("@opensumi/ide-workspace");
8
+ const service_1 = require("../../languages/service");
9
+ const languages_1 = require("./languages");
10
+ const getImportedFilesInterface = async (code, resource, options, injector) => {
11
+ var _a;
12
+ const languageParserService = injector.get(service_1.LanguageParserService);
13
+ let matches = [];
14
+ const snippets = [];
15
+ const languageParser = languageParserService.createParser(resource.languageId);
16
+ if (languageParser) {
17
+ matches = await languageParser.extractImportPaths(code);
18
+ }
19
+ const maxPromptTime = (_a = options.maxTime) !== null && _a !== void 0 ? _a : 200;
20
+ const startTime = Date.now();
21
+ for (const file of matches) {
22
+ let filePath = '';
23
+ const basePath = resource.uri.parent.codeUri.fsPath;
24
+ if (file.startsWith('@/') || file.startsWith('~') || !file.startsWith('.')) {
25
+ // skip alias
26
+ continue;
27
+ }
28
+ if (file === '.') {
29
+ filePath = new path_1.Path(basePath).join('index.ts').toString();
30
+ }
31
+ else {
32
+ filePath = new path_1.Path(basePath).join(`${file}${languages_1.LANGUAGE_TO_SUFFIX[resource.languageId]}`).toString();
33
+ }
34
+ try {
35
+ const fileService = injector.get(ide_file_service_1.IFileServiceClient);
36
+ const workspaceService = injector.get(ide_workspace_1.IWorkspaceService);
37
+ const fileUri = ide_core_common_1.URI.file(filePath);
38
+ const content = await fileService.readFile(fileUri.codeUri.fsPath);
39
+ const interfaceCode = await (languageParser === null || languageParser === void 0 ? void 0 : languageParser.extractInterfaceOrTypeCode(content.content.toString()));
40
+ if (interfaceCode && interfaceCode.length > 0) {
41
+ const relative = await workspaceService.asRelativePath(fileUri);
42
+ if (relative === null || relative === void 0 ? void 0 : relative.path) {
43
+ snippets.push([relative.path, interfaceCode]);
44
+ }
45
+ }
46
+ }
47
+ catch (_b) { }
48
+ if (Date.now() - startTime > maxPromptTime) {
49
+ break;
50
+ }
51
+ }
52
+ return snippets;
53
+ };
54
+ exports.getImportedFilesInterface = getImportedFilesInterface;
55
+ const extractLocalImportContext = async (resource, options, injector) => {
56
+ // 从代码中提取出所有的 import 语句,并进一步提取 interface 相关代码声明
57
+ const { source } = resource;
58
+ if (resource.languageId === 'typescript') {
59
+ return (0, exports.getImportedFilesInterface)(source, resource, options, injector);
60
+ }
61
+ return [];
62
+ };
63
+ exports.extractLocalImportContext = extractLocalImportContext;
64
+ const getImportedFile = async (context, options, injector) => {
65
+ if (context.language && !options.language.includes(context.language)) {
66
+ return [];
67
+ }
68
+ const content = context.prefix;
69
+ return (0, exports.extractLocalImportContext)({
70
+ source: content,
71
+ uri: context.uri,
72
+ languageId: context.language,
73
+ offset: content.length,
74
+ }, options, injector);
75
+ };
76
+ exports.getImportedFile = getImportedFile;
77
+ //# sourceMappingURL=importedFiles.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"importedFiles.js","sourceRoot":"","sources":["../../../../src/browser/inline-completions/prompt/importedFiles.ts"],"names":[],"mappings":";;;AACA,+DAAgD;AAChD,iEAAgE;AAChE,uDAAoD;AACpD,2DAA4D;AAE5D,qDAAgE;AAGhE,2CAAiD;AAE1C,MAAM,yBAAyB,GAAG,KAAK,EAC5C,IAAY,EACZ,QAA0B,EAC1B,OAA4B,EAC5B,QAAkB,EACgB,EAAE;;IACpC,MAAM,qBAAqB,GAAG,QAAQ,CAAC,GAAG,CAAC,+BAAqB,CAA0B,CAAC;IAC3F,IAAI,OAAO,GAAa,EAAE,CAAC;IAC3B,MAAM,QAAQ,GAA4B,EAAE,CAAC;IAC7C,MAAM,cAAc,GAAG,qBAAqB,CAAC,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IAC/E,IAAI,cAAc,EAAE;QAClB,OAAO,GAAG,MAAM,cAAc,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;KACzD;IACD,MAAM,aAAa,GAAG,MAAA,OAAO,CAAC,OAAO,mCAAI,GAAG,CAAC;IAC7C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE;QAC1B,IAAI,QAAQ,GAAG,EAAE,CAAC;QAClB,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QACpD,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE;YAC1E,aAAa;YACb,SAAS;SACV;QACD,IAAI,IAAI,KAAK,GAAG,EAAE;YAChB,QAAQ,GAAG,IAAI,WAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,QAAQ,EAAE,CAAC;SAC3D;aAAM;YACL,QAAQ,GAAG,IAAI,WAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,GAAG,IAAI,GAAG,8BAAkB,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;SACpG;QACD,IAAI;YACF,MAAM,WAAW,GAAG,QAAQ,CAAC,GAAG,CAAC,qCAAkB,CAAuB,CAAC;YAC3E,MAAM,gBAAgB,GAAG,QAAQ,CAAC,GAAG,CAAC,iCAAiB,CAAsB,CAAC;YAC9E,MAAM,OAAO,GAAG,qBAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACnC,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACnE,MAAM,aAAa,GAAG,MAAM,CAAA,cAAc,aAAd,cAAc,uBAAd,cAAc,CAAE,0BAA0B,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAA,CAAC;YACnG,IAAI,aAAa,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7C,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;gBAChE,IAAI,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,IAAI,EAAE;oBAClB,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC;iBAC/C;aACF;SACF;QAAC,WAAM,GAAE;QACV,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,GAAG,aAAa,EAAE;YAC1C,MAAM;SACP;KACF;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AA7CW,QAAA,yBAAyB,6BA6CpC;AAEK,MAAM,yBAAyB,GAAG,KAAK,EAC5C,QAA0B,EAC1B,OAA4B,EAC5B,QAAkB,EACgB,EAAE;IACpC,+CAA+C;IAC/C,MAAM,EAAE,MAAM,EAAE,GAAG,QAAQ,CAAC;IAE5B,IAAI,QAAQ,CAAC,UAAU,KAAK,YAAY,EAAE;QACxC,OAAO,IAAA,iCAAyB,EAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;KACvE;IACD,OAAO,EAAE,CAAC;AACZ,CAAC,CAAC;AAZW,QAAA,yBAAyB,6BAYpC;AAEK,MAAM,eAAe,GAAG,KAAK,EAClC,OAA2B,EAC3B,OAA4B,EAC5B,QAAkB,EACgB,EAAE;IACpC,IAAI,OAAO,CAAC,QAAQ,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;QACpE,OAAO,EAAE,CAAC;KACX;IACD,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;IAC/B,OAAO,IAAA,iCAAyB,EAC9B;QACE,MAAM,EAAE,OAAO;QACf,GAAG,EAAE,OAAO,CAAC,GAAG;QAChB,UAAU,EAAE,OAAO,CAAC,QAAQ;QAC5B,MAAM,EAAE,OAAO,CAAC,MAAM;KACvB,EACD,OAAO,EACP,QAAQ,CACT,CAAC;AACJ,CAAC,CAAC;AAnBW,QAAA,eAAe,mBAmB1B"}
@@ -0,0 +1,31 @@
1
+ import { ResourceDocument, TokenizerName } from '../types';
2
+ import { WindowedMatcher } from './matcher';
3
+ export interface CursorContextOptions {
4
+ maxLineCount?: number;
5
+ maxTokenLength?: number;
6
+ tokenizerName?: TokenizerName;
7
+ extensionPath?: string;
8
+ }
9
+ export declare const getCursorContext: (resource: ResourceDocument, option?: CursorContextOptions) => {
10
+ context: string;
11
+ lineCount: number;
12
+ tokenLength: number;
13
+ tokenizerName: TokenizerName;
14
+ };
15
+ export declare class FixedWindowSizeJaccardMatcher extends WindowedMatcher {
16
+ static factory(windowSize: number): {
17
+ to: (doc: ResourceDocument) => FixedWindowSizeJaccardMatcher;
18
+ };
19
+ private readonly windowLength;
20
+ constructor(doc: ResourceDocument, windowSize: number);
21
+ id(): string;
22
+ getWindowsDelineations(lines: string[]): [number, number][];
23
+ getCursorContextInfo(resource: ResourceDocument): {
24
+ context: string;
25
+ lineCount: number;
26
+ tokenLength: number;
27
+ tokenizerName: TokenizerName;
28
+ };
29
+ similarityScore(set1: Set<string>, set2: Set<string>): number;
30
+ }
31
+ //# sourceMappingURL=jaccardMatcher.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"jaccardMatcher.d.ts","sourceRoot":"","sources":["../../../../src/browser/inline-completions/prompt/jaccardMatcher.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAE3D,OAAO,EAAE,eAAe,EAAgB,MAAM,WAAW,CAAC;AAG1D,MAAM,WAAW,oBAAoB;IACnC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,eAAO,MAAM,gBAAgB,aACjB,gBAAgB,WAClB,oBAAoB;aAEnB,MAAM;eACJ,MAAM;iBACJ,MAAM;mBACJ,aAAa;CA+B7B,CAAC;AAkBF,qBAAa,6BAA8B,SAAQ,eAAe;IAChE,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM;kBAEnB,gBAAgB;;IAI9B,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;gBAE1B,GAAG,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM;IAKrD,EAAE;IAIF,sBAAsB,CAAC,KAAK,EAAE,MAAM,EAAE;IAItC,oBAAoB,CAAC,QAAQ,EAAE,gBAAgB;;;;;;IAI/C,eAAe,CAAC,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC;CAGrD"}
@@ -0,0 +1,75 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.FixedWindowSizeJaccardMatcher = exports.getCursorContext = void 0;
4
+ const types_1 = require("../types");
5
+ const matcher_1 = require("./matcher");
6
+ const tokenizer_1 = require("./tokenizer");
7
+ const getCursorContext = (resource, option = {}) => {
8
+ const baseOptions = Object.assign({ tokenizerName: types_1.TokenizerName.cl100k_base }, option);
9
+ const tokenizer = (0, tokenizer_1.getTokenizer)(baseOptions.tokenizerName);
10
+ if (baseOptions.maxLineCount && baseOptions.maxLineCount < 0) {
11
+ throw new Error('maxLineCount must be non-negative if defined');
12
+ }
13
+ if (baseOptions.maxTokenLength && baseOptions.maxTokenLength < 0) {
14
+ throw new Error('maxTokenLength must be non-negative if defined');
15
+ }
16
+ if (baseOptions.maxLineCount === 0 || baseOptions.maxTokenLength === 0) {
17
+ return {
18
+ context: '',
19
+ lineCount: 0,
20
+ tokenLength: 0,
21
+ tokenizerName: baseOptions.tokenizerName,
22
+ };
23
+ }
24
+ let context = resource.source.slice(0, resource.offset);
25
+ if (baseOptions.maxLineCount) {
26
+ context = context.split('\n').slice(-baseOptions.maxLineCount).join('\n');
27
+ }
28
+ return {
29
+ context,
30
+ lineCount: context.split('\n').length,
31
+ tokenLength: tokenizer.encode(context).length,
32
+ tokenizerName: baseOptions.tokenizerName,
33
+ };
34
+ };
35
+ exports.getCursorContext = getCursorContext;
36
+ const getBasicWindowDelineations = (windowSize, lineArrays) => {
37
+ const ranges = [];
38
+ const arrayLength = lineArrays.length;
39
+ if (arrayLength === 0) {
40
+ return [];
41
+ }
42
+ if (arrayLength < windowSize) {
43
+ return [[0, arrayLength]];
44
+ }
45
+ for (let i = 0; i < arrayLength - windowSize + 1; i++) {
46
+ ranges.push([i, i + windowSize]);
47
+ }
48
+ return ranges;
49
+ };
50
+ // 固定窗口大小的 Jaccard 匹配器
51
+ class FixedWindowSizeJaccardMatcher extends matcher_1.WindowedMatcher {
52
+ static factory(windowSize) {
53
+ return {
54
+ to: (doc) => new FixedWindowSizeJaccardMatcher(doc, windowSize),
55
+ };
56
+ }
57
+ constructor(doc, windowSize) {
58
+ super(doc);
59
+ this.windowLength = windowSize;
60
+ }
61
+ id() {
62
+ return `fixed:${this.windowLength}`;
63
+ }
64
+ getWindowsDelineations(lines) {
65
+ return getBasicWindowDelineations(this.windowLength, lines);
66
+ }
67
+ getCursorContextInfo(resource) {
68
+ return (0, exports.getCursorContext)(resource, { maxLineCount: this.windowLength });
69
+ }
70
+ similarityScore(set1, set2) {
71
+ return (0, matcher_1.computeScore)(set1, set2);
72
+ }
73
+ }
74
+ exports.FixedWindowSizeJaccardMatcher = FixedWindowSizeJaccardMatcher;
75
+ //# sourceMappingURL=jaccardMatcher.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"jaccardMatcher.js","sourceRoot":"","sources":["../../../../src/browser/inline-completions/prompt/jaccardMatcher.ts"],"names":[],"mappings":";;;AAAA,oCAA2D;AAE3D,uCAA0D;AAC1D,2CAA2C;AASpC,MAAM,gBAAgB,GAAG,CAC9B,QAA0B,EAC1B,SAA+B,EAAE,EAMjC,EAAE;IACF,MAAM,WAAW,mBACf,aAAa,EAAE,qBAAa,CAAC,WAAW,IACrC,MAAM,CACV,CAAC;IACF,MAAM,SAAS,GAAG,IAAA,wBAAY,EAAC,WAAW,CAAC,aAAa,CAAC,CAAC;IAC1D,IAAI,WAAW,CAAC,YAAY,IAAI,WAAW,CAAC,YAAY,GAAG,CAAC,EAAE;QAC5D,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;KACjE;IACD,IAAI,WAAW,CAAC,cAAc,IAAI,WAAW,CAAC,cAAc,GAAG,CAAC,EAAE;QAChE,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;KACnE;IACD,IAAI,WAAW,CAAC,YAAY,KAAK,CAAC,IAAI,WAAW,CAAC,cAAc,KAAK,CAAC,EAAE;QACtE,OAAO;YACL,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,CAAC;YACZ,WAAW,EAAE,CAAC;YACd,aAAa,EAAE,WAAW,CAAC,aAAa;SACzC,CAAC;KACH;IACD,IAAI,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;IACxD,IAAI,WAAW,CAAC,YAAY,EAAE;QAC5B,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;KAC3E;IACD,OAAO;QACL,OAAO;QACP,SAAS,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM;QACrC,WAAW,EAAE,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM;QAC7C,aAAa,EAAE,WAAW,CAAC,aAAa;KACzC,CAAC;AACJ,CAAC,CAAC;AAtCW,QAAA,gBAAgB,oBAsC3B;AAEF,MAAM,0BAA0B,GAAG,CAAC,UAAkB,EAAE,UAAoB,EAAsB,EAAE;IAClG,MAAM,MAAM,GAAuB,EAAE,CAAC;IACtC,MAAM,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC;IACtC,IAAI,WAAW,KAAK,CAAC,EAAE;QACrB,OAAO,EAAE,CAAC;KACX;IACD,IAAI,WAAW,GAAG,UAAU,EAAE;QAC5B,OAAO,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC;KAC3B;IACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,GAAG,UAAU,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE;QACrD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC;KAClC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC;AAEF,sBAAsB;AACtB,MAAa,6BAA8B,SAAQ,yBAAe;IAChE,MAAM,CAAC,OAAO,CAAC,UAAkB;QAC/B,OAAO;YACL,EAAE,EAAE,CAAC,GAAqB,EAAE,EAAE,CAAC,IAAI,6BAA6B,CAAC,GAAG,EAAE,UAAU,CAAC;SAClF,CAAC;IACJ,CAAC;IAID,YAAY,GAAqB,EAAE,UAAkB;QACnD,KAAK,CAAC,GAAG,CAAC,CAAC;QACX,IAAI,CAAC,YAAY,GAAG,UAAU,CAAC;IACjC,CAAC;IAED,EAAE;QACA,OAAO,SAAS,IAAI,CAAC,YAAY,EAAE,CAAC;IACtC,CAAC;IAED,sBAAsB,CAAC,KAAe;QACpC,OAAO,0BAA0B,CAAC,IAAI,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;IAC9D,CAAC;IAED,oBAAoB,CAAC,QAA0B;QAC7C,OAAO,IAAA,wBAAgB,EAAC,QAAQ,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;IACzE,CAAC;IAED,eAAe,CAAC,IAAiB,EAAE,IAAiB;QAClD,OAAO,IAAA,sBAAY,EAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAClC,CAAC;CACF;AA7BD,sEA6BC"}
@@ -0,0 +1,4 @@
1
+ export declare const LANGUAGE_TO_SUFFIX: {
2
+ [key: string]: string;
3
+ };
4
+ //# sourceMappingURL=languages.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"languages.d.ts","sourceRoot":"","sources":["../../../../src/browser/inline-completions/prompt/languages.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,kBAAkB,EAAE;IAC/B,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAAC;CA+DvB,CAAC"}
@@ -0,0 +1,67 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.LANGUAGE_TO_SUFFIX = void 0;
4
+ exports.LANGUAGE_TO_SUFFIX = {
5
+ abap: '.abap',
6
+ bat: '.bat',
7
+ bibtex: '.bib',
8
+ c: '.c',
9
+ clojure: '.clj',
10
+ coffeescript: '.coffee',
11
+ cpp: '.cpp',
12
+ csharp: '.cs',
13
+ css: '.css',
14
+ 'cuda-cpp': '.cu',
15
+ dart: '.dart',
16
+ diff: '.diff',
17
+ dockerfile: '.dockerfile',
18
+ fsharp: '.fs',
19
+ go: '.go',
20
+ groovy: '.groovy',
21
+ haml: '.haml',
22
+ handlebars: '.handlebars',
23
+ hlsl: '.hlsl',
24
+ html: '.html',
25
+ ini: '.ini',
26
+ jade: '.jade',
27
+ java: '.java',
28
+ javascript: '.js',
29
+ javascriptreact: '.jsx',
30
+ json: '.json',
31
+ julia: '.jl',
32
+ latex: '.tex',
33
+ less: '.less',
34
+ lua: '.lua',
35
+ makefile: '.make',
36
+ markdown: '.md',
37
+ 'objective-c': '.m',
38
+ 'objective-cpp': '.mm',
39
+ perl: '.pl',
40
+ perl6: '.6pl',
41
+ php: '.php',
42
+ plaintext: '.txt',
43
+ powershell: '.ps1',
44
+ pug: '.pug',
45
+ python: '.py',
46
+ r: '.r',
47
+ razor: '.cshtml',
48
+ ruby: '.rb',
49
+ rust: '.rs',
50
+ sass: '.sass',
51
+ scss: '.scss',
52
+ shaderlab: '.shader',
53
+ shellscript: '.sh',
54
+ slim: '.slim',
55
+ sql: '.sql',
56
+ stylus: '.styl',
57
+ swift: '.swift',
58
+ tex: '.tex',
59
+ typescript: '.ts',
60
+ typescriptreact: '.tsx',
61
+ vb: '.vb',
62
+ vue: '.vue',
63
+ xml: '.xml',
64
+ xsl: '.xsl',
65
+ yaml: '.yaml',
66
+ };
67
+ //# sourceMappingURL=languages.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"languages.js","sourceRoot":"","sources":["../../../../src/browser/inline-completions/prompt/languages.ts"],"names":[],"mappings":";;;AAAa,QAAA,kBAAkB,GAE3B;IACF,IAAI,EAAE,OAAO;IACb,GAAG,EAAE,MAAM;IACX,MAAM,EAAE,MAAM;IACd,CAAC,EAAE,IAAI;IACP,OAAO,EAAE,MAAM;IACf,YAAY,EAAE,SAAS;IACvB,GAAG,EAAE,MAAM;IACX,MAAM,EAAE,KAAK;IACb,GAAG,EAAE,MAAM;IACX,UAAU,EAAE,KAAK;IACjB,IAAI,EAAE,OAAO;IACb,IAAI,EAAE,OAAO;IACb,UAAU,EAAE,aAAa;IACzB,MAAM,EAAE,KAAK;IACb,EAAE,EAAE,KAAK;IACT,MAAM,EAAE,SAAS;IACjB,IAAI,EAAE,OAAO;IACb,UAAU,EAAE,aAAa;IACzB,IAAI,EAAE,OAAO;IACb,IAAI,EAAE,OAAO;IACb,GAAG,EAAE,MAAM;IACX,IAAI,EAAE,OAAO;IACb,IAAI,EAAE,OAAO;IACb,UAAU,EAAE,KAAK;IACjB,eAAe,EAAE,MAAM;IACvB,IAAI,EAAE,OAAO;IACb,KAAK,EAAE,KAAK;IACZ,KAAK,EAAE,MAAM;IACb,IAAI,EAAE,OAAO;IACb,GAAG,EAAE,MAAM;IACX,QAAQ,EAAE,OAAO;IACjB,QAAQ,EAAE,KAAK;IACf,aAAa,EAAE,IAAI;IACnB,eAAe,EAAE,KAAK;IACtB,IAAI,EAAE,KAAK;IACX,KAAK,EAAE,MAAM;IACb,GAAG,EAAE,MAAM;IACX,SAAS,EAAE,MAAM;IACjB,UAAU,EAAE,MAAM;IAClB,GAAG,EAAE,MAAM;IACX,MAAM,EAAE,KAAK;IACb,CAAC,EAAE,IAAI;IACP,KAAK,EAAE,SAAS;IAChB,IAAI,EAAE,KAAK;IACX,IAAI,EAAE,KAAK;IACX,IAAI,EAAE,OAAO;IACb,IAAI,EAAE,OAAO;IACb,SAAS,EAAE,SAAS;IACpB,WAAW,EAAE,KAAK;IAClB,IAAI,EAAE,OAAO;IACb,GAAG,EAAE,MAAM;IACX,MAAM,EAAE,OAAO;IACf,KAAK,EAAE,QAAQ;IACf,GAAG,EAAE,MAAM;IACX,UAAU,EAAE,KAAK;IACjB,eAAe,EAAE,MAAM;IACvB,EAAE,EAAE,KAAK;IACT,GAAG,EAAE,MAAM;IACX,GAAG,EAAE,MAAM;IACX,GAAG,EAAE,MAAM;IACX,IAAI,EAAE,OAAO;CACd,CAAC"}
@@ -0,0 +1,42 @@
1
+ import { CursorContext, MatchSimilarSnippet, ResourceDocument, SimilarSnippet, SnippetSelectionOption, SnippetSemantics, SortOption } from '../types';
2
+ export declare class TokensCache {
3
+ private readonly size;
4
+ private readonly keys;
5
+ private readonly cache;
6
+ constructor(size: number);
7
+ put(id: string, tokens: Set<string>[]): void;
8
+ get(id: string): Set<string>[];
9
+ }
10
+ export declare abstract class WindowedMatcher {
11
+ private referenceDoc;
12
+ private tokenizer;
13
+ abstract getCursorContextInfo(doc: ResourceDocument): CursorContext;
14
+ abstract getWindowsDelineations(lines: string[]): [number, number][];
15
+ abstract similarityScore(tokens1: Set<string>, tokens2: Set<string>): number;
16
+ abstract id(): string;
17
+ private referenceTokensCache?;
18
+ constructor(doc: ResourceDocument);
19
+ get referenceTokens(): Set<string>;
20
+ sortScoredSnippets(snippets: SimilarSnippet[], sortRule?: SortOption): SimilarSnippet[];
21
+ retrieveAllSnippets(resource: ResourceDocument, sortRule?: SortOption): SimilarSnippet[];
22
+ findMatches(resource: ResourceDocument, mode?: SnippetSelectionOption, size?: number): MatchSimilarSnippet[] | undefined;
23
+ /**
24
+ * 获取所有匹配的片段
25
+ * @param resource 文档资源
26
+ */
27
+ findBestMatch(resource: ResourceDocument): {
28
+ score: number;
29
+ startLine: number;
30
+ endLine: number;
31
+ snippet: string;
32
+ semantics: SnippetSemantics;
33
+ } | null | undefined;
34
+ /**
35
+ * 获取最高匹配片段
36
+ * @param resource 文档资源
37
+ * @param size 取前几位,默认取前一位
38
+ */
39
+ findTopKMatches(resource: ResourceDocument, size?: number): MatchSimilarSnippet[] | undefined;
40
+ }
41
+ export declare function computeScore(set1: Set<string>, set2: Set<string>): number;
42
+ //# sourceMappingURL=matcher.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"matcher.d.ts","sourceRoot":"","sources":["../../../../src/browser/inline-completions/prompt/matcher.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,aAAa,EACb,mBAAmB,EACnB,gBAAgB,EAChB,cAAc,EACd,sBAAsB,EACtB,gBAAgB,EAChB,UAAU,EACX,MAAM,UAAU,CAAC;AAElB,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAS;IAE9B,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAgB;IAErC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAEf;gBAEK,IAAI,EAAE,MAAM;IAIxB,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE;IAWrC,GAAG,CAAC,EAAE,EAAE,MAAM;CAGf;AA8JD,8BAAsB,eAAe;IACnC,OAAO,CAAC,YAAY,CAAmB;IAEvC,OAAO,CAAC,SAAS,CAAoB;IAErC,QAAQ,CAAC,oBAAoB,CAAC,GAAG,EAAE,gBAAgB,GAAG,aAAa;IACnE,QAAQ,CAAC,sBAAsB,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE;IACpE,QAAQ,CAAC,eAAe,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,GAAG,MAAM;IAC5E,QAAQ,CAAC,EAAE,IAAI,MAAM;IAErB,OAAO,CAAC,oBAAoB,CAAC,CAAc;gBAE/B,GAAG,EAAE,gBAAgB;IAKjC,IAAI,eAAe,gBAOlB;IAED,kBAAkB,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE,QAAQ,aAAwB;IAS/E,mBAAmB,CAAC,QAAQ,EAAE,gBAAgB,EAAE,QAAQ,aAAwB;IA8BhF,WAAW,CACT,QAAQ,EAAE,gBAAgB,EAC1B,IAAI,yBAAmC,EACvC,IAAI,CAAC,EAAE,MAAM,GACZ,mBAAmB,EAAE,GAAG,SAAS;IAQpC;;;OAGG;IACH,aAAa,CAAC,QAAQ,EAAE,gBAAgB;;;;;;;IAexC;;;;OAIG;IACH,eAAe,CAAC,QAAQ,EAAE,gBAAgB,EAAE,IAAI,SAAI,GAAG,mBAAmB,EAAE,GAAG,SAAS;CAqBzF;AAED,wBAAgB,YAAY,CAAC,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,UAQhE"}
@@ -0,0 +1,279 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.computeScore = exports.WindowedMatcher = exports.TokensCache = void 0;
4
+ const types_1 = require("../types");
5
+ class TokensCache {
6
+ constructor(size) {
7
+ this.keys = [];
8
+ this.cache = {};
9
+ this.size = size;
10
+ }
11
+ put(id, tokens) {
12
+ var _a;
13
+ this.cache[id] = tokens;
14
+ if (!this.keys.includes(id)) {
15
+ this.keys.push(id);
16
+ }
17
+ if (this.keys.length > this.size) {
18
+ const key = (_a = this.keys.shift()) !== null && _a !== void 0 ? _a : '';
19
+ delete this.cache[key];
20
+ }
21
+ }
22
+ get(id) {
23
+ return this.cache[id];
24
+ }
25
+ }
26
+ exports.TokensCache = TokensCache;
27
+ const englishStopWords = new Set([
28
+ 'we',
29
+ 'our',
30
+ 'you',
31
+ 'it',
32
+ 'its',
33
+ 'they',
34
+ 'them',
35
+ 'their',
36
+ 'this',
37
+ 'that',
38
+ 'these',
39
+ 'those',
40
+ 'is',
41
+ 'are',
42
+ 'was',
43
+ 'were',
44
+ 'be',
45
+ 'been',
46
+ 'being',
47
+ 'have',
48
+ 'has',
49
+ 'had',
50
+ 'having',
51
+ 'do',
52
+ 'does',
53
+ 'did',
54
+ 'doing',
55
+ 'can',
56
+ 'don',
57
+ 't',
58
+ 's',
59
+ 'will',
60
+ 'would',
61
+ 'should',
62
+ 'what',
63
+ 'which',
64
+ 'who',
65
+ 'when',
66
+ 'where',
67
+ 'why',
68
+ 'how',
69
+ 'a',
70
+ 'an',
71
+ 'the',
72
+ 'and',
73
+ 'or',
74
+ 'not',
75
+ 'no',
76
+ 'but',
77
+ 'because',
78
+ 'as',
79
+ 'until',
80
+ 'again',
81
+ 'further',
82
+ 'then',
83
+ 'once',
84
+ 'here',
85
+ 'there',
86
+ 'all',
87
+ 'any',
88
+ 'both',
89
+ 'each',
90
+ 'few',
91
+ 'more',
92
+ 'most',
93
+ 'other',
94
+ 'some',
95
+ 'such',
96
+ 'above',
97
+ 'below',
98
+ 'to',
99
+ 'during',
100
+ 'before',
101
+ 'after',
102
+ 'of',
103
+ 'at',
104
+ 'by',
105
+ 'about',
106
+ 'between',
107
+ 'into',
108
+ 'through',
109
+ 'from',
110
+ 'up',
111
+ 'down',
112
+ 'in',
113
+ 'out',
114
+ 'on',
115
+ 'off',
116
+ 'over',
117
+ 'under',
118
+ 'only',
119
+ 'own',
120
+ 'same',
121
+ 'so',
122
+ 'than',
123
+ 'too',
124
+ 'very',
125
+ 'just',
126
+ 'now',
127
+ ]);
128
+ const programmingStopWords = new Set([
129
+ 'if',
130
+ 'then',
131
+ 'else',
132
+ 'for',
133
+ 'while',
134
+ 'with',
135
+ 'def',
136
+ 'function',
137
+ 'return',
138
+ 'TODO',
139
+ 'import',
140
+ 'try',
141
+ 'catch',
142
+ 'raise',
143
+ 'finally',
144
+ 'repeat',
145
+ 'switch',
146
+ 'case',
147
+ 'match',
148
+ 'assert',
149
+ 'continue',
150
+ 'break',
151
+ 'const',
152
+ 'class',
153
+ 'enum',
154
+ 'struct',
155
+ 'static',
156
+ 'new',
157
+ 'super',
158
+ 'this',
159
+ 'var',
160
+ ]);
161
+ const allStopWords = new Set([...englishStopWords, ...programmingStopWords]);
162
+ const languageStopWords = new Map();
163
+ function splitIntoWords(text) {
164
+ return text.split(/[^a-zA-Z0-9]/).filter((word) => word.length > 0);
165
+ }
166
+ class DocumentTokenizer {
167
+ constructor(resource) {
168
+ var _a;
169
+ this.stopsForLanguage = (_a = languageStopWords.get(resource.languageId)) !== null && _a !== void 0 ? _a : allStopWords;
170
+ }
171
+ tokenize(text) {
172
+ return new Set(splitIntoWords(text).filter((word) => !this.stopsForLanguage.has(word)));
173
+ }
174
+ }
175
+ const tokensCache = new TokensCache(20);
176
+ class WindowedMatcher {
177
+ constructor(doc) {
178
+ this.referenceDoc = doc;
179
+ this.tokenizer = new DocumentTokenizer(doc);
180
+ }
181
+ get referenceTokens() {
182
+ if (!this.referenceTokensCache) {
183
+ this.referenceTokensCache = new Set(this.tokenizer.tokenize(this.getCursorContextInfo(this.referenceDoc).context));
184
+ }
185
+ return this.referenceTokensCache;
186
+ }
187
+ sortScoredSnippets(snippets, sortRule = types_1.SortOption.Descending) {
188
+ // eslint-disable-next-line no-nested-ternary
189
+ return sortRule === types_1.SortOption.Ascending
190
+ ? snippets.sort((snippet1, snippet2) => (snippet1.score > snippet2.score ? 1 : -1))
191
+ : sortRule === types_1.SortOption.Descending
192
+ ? snippets.sort((snippet1, snippet2) => (snippet1.score > snippet2.score ? -1 : 1))
193
+ : snippets;
194
+ }
195
+ retrieveAllSnippets(resource, sortRule = types_1.SortOption.Descending) {
196
+ var _a;
197
+ const snippets = [];
198
+ if (!resource.source.length || !this.referenceTokens.size) {
199
+ return snippets;
200
+ }
201
+ const sourceArray = resource.source.split('\n');
202
+ const key = `${this.id()}:${resource.source}`;
203
+ const cache = (_a = tokensCache.get(key)) !== null && _a !== void 0 ? _a : [];
204
+ const noCache = !cache.length;
205
+ const tokens = noCache ? sourceArray.map((text) => this.tokenizer.tokenize(text), this.tokenizer) : [];
206
+ for (const [index, [startLine, endLine]] of this.getWindowsDelineations(sourceArray).entries()) {
207
+ if (noCache) {
208
+ const size = new Set();
209
+ tokens.slice(startLine, endLine).forEach((token) => token.forEach((word) => size.add(word)));
210
+ cache.push(size);
211
+ }
212
+ const traget = cache[index];
213
+ const score = this.similarityScore(traget, this.referenceTokens);
214
+ snippets.push({
215
+ score,
216
+ startLine,
217
+ endLine,
218
+ });
219
+ }
220
+ if (noCache) {
221
+ tokensCache.put(key, cache);
222
+ }
223
+ return this.sortScoredSnippets(snippets, sortRule);
224
+ }
225
+ findMatches(resource, mode = types_1.SnippetSelectionOption.BestMatch, size) {
226
+ if (mode === types_1.SnippetSelectionOption.BestMatch) {
227
+ const snippet = this.findBestMatch(resource);
228
+ return snippet ? [snippet] : [];
229
+ }
230
+ return mode === types_1.SnippetSelectionOption.TopK ? this.findTopKMatches(resource, size) : [];
231
+ }
232
+ /**
233
+ * 获取所有匹配的片段
234
+ * @param resource 文档资源
235
+ */
236
+ findBestMatch(resource) {
237
+ if (!resource.source.length || !this.referenceTokens.size) {
238
+ return;
239
+ }
240
+ const lines = resource.source.split('\n');
241
+ const snippets = this.retrieveAllSnippets(resource, types_1.SortOption.Descending);
242
+ return snippets.length !== 0 && snippets[0].score !== 0
243
+ ? Object.assign({ snippet: lines.slice(snippets[0].startLine, snippets[0].endLine).join('\n'), semantics: types_1.SnippetSemantics.Snippet }, snippets[0]) : null;
244
+ }
245
+ /**
246
+ * 获取最高匹配片段
247
+ * @param resource 文档资源
248
+ * @param size 取前几位,默认取前一位
249
+ */
250
+ findTopKMatches(resource, size = 1) {
251
+ if (!resource.source.length || !this.referenceTokens.size || size < 1) {
252
+ return;
253
+ }
254
+ const lines = resource.source.split('\n');
255
+ const snippets = this.retrieveAllSnippets(resource, types_1.SortOption.Descending);
256
+ if (!snippets.length || !snippets[0].score) {
257
+ return;
258
+ }
259
+ const topSnippets = [snippets[0]];
260
+ for (let i = 1; i < snippets.length && topSnippets.length < size; i++) {
261
+ if (topSnippets.findIndex((t) => snippets[i].startLine < t.endLine && snippets[i].endLine > t.startLine) === -1) {
262
+ topSnippets.push(snippets[i]);
263
+ }
264
+ }
265
+ return topSnippets.map((snippet) => (Object.assign({ snippet: lines.slice(snippet.startLine, snippet.endLine).join('\n'), semantics: types_1.SnippetSemantics.Snippet }, snippet)));
266
+ }
267
+ }
268
+ exports.WindowedMatcher = WindowedMatcher;
269
+ function computeScore(set1, set2) {
270
+ let intersectionSize = 0;
271
+ for (const element of set1) {
272
+ if (set2.has(element)) {
273
+ intersectionSize++;
274
+ }
275
+ }
276
+ return intersectionSize / (set1.size + set2.size - intersectionSize);
277
+ }
278
+ exports.computeScore = computeScore;
279
+ //# sourceMappingURL=matcher.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"matcher.js","sourceRoot":"","sources":["../../../../src/browser/inline-completions/prompt/matcher.ts"],"names":[],"mappings":";;;AAAA,oCAQkB;AAElB,MAAa,WAAW;IAStB,YAAY,IAAY;QANP,SAAI,GAAa,EAAE,CAAC;QAEpB,UAAK,GAElB,EAAE,CAAC;QAGL,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,GAAG,CAAC,EAAU,EAAE,MAAqB;;QACnC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC;QACxB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE;YAC3B,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;SACpB;QACD,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,EAAE;YAChC,MAAM,GAAG,GAAG,MAAA,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,mCAAI,EAAE,CAAC;YACpC,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;SACxB;IACH,CAAC;IAED,GAAG,CAAC,EAAU;QACZ,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACxB,CAAC;CACF;AA3BD,kCA2BC;AAED,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;IAC/B,IAAI;IACJ,KAAK;IACL,KAAK;IACL,IAAI;IACJ,KAAK;IACL,MAAM;IACN,MAAM;IACN,OAAO;IACP,MAAM;IACN,MAAM;IACN,OAAO;IACP,OAAO;IACP,IAAI;IACJ,KAAK;IACL,KAAK;IACL,MAAM;IACN,IAAI;IACJ,MAAM;IACN,OAAO;IACP,MAAM;IACN,KAAK;IACL,KAAK;IACL,QAAQ;IACR,IAAI;IACJ,MAAM;IACN,KAAK;IACL,OAAO;IACP,KAAK;IACL,KAAK;IACL,GAAG;IACH,GAAG;IACH,MAAM;IACN,OAAO;IACP,QAAQ;IACR,MAAM;IACN,OAAO;IACP,KAAK;IACL,MAAM;IACN,OAAO;IACP,KAAK;IACL,KAAK;IACL,GAAG;IACH,IAAI;IACJ,KAAK;IACL,KAAK;IACL,IAAI;IACJ,KAAK;IACL,IAAI;IACJ,KAAK;IACL,SAAS;IACT,IAAI;IACJ,OAAO;IACP,OAAO;IACP,SAAS;IACT,MAAM;IACN,MAAM;IACN,MAAM;IACN,OAAO;IACP,KAAK;IACL,KAAK;IACL,MAAM;IACN,MAAM;IACN,KAAK;IACL,MAAM;IACN,MAAM;IACN,OAAO;IACP,MAAM;IACN,MAAM;IACN,OAAO;IACP,OAAO;IACP,IAAI;IACJ,QAAQ;IACR,QAAQ;IACR,OAAO;IACP,IAAI;IACJ,IAAI;IACJ,IAAI;IACJ,OAAO;IACP,SAAS;IACT,MAAM;IACN,SAAS;IACT,MAAM;IACN,IAAI;IACJ,MAAM;IACN,IAAI;IACJ,KAAK;IACL,IAAI;IACJ,KAAK;IACL,MAAM;IACN,OAAO;IACP,MAAM;IACN,KAAK;IACL,MAAM;IACN,IAAI;IACJ,MAAM;IACN,KAAK;IACL,MAAM;IACN,MAAM;IACN,KAAK;CACN,CAAC,CAAC;AACH,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAC;IACnC,IAAI;IACJ,MAAM;IACN,MAAM;IACN,KAAK;IACL,OAAO;IACP,MAAM;IACN,KAAK;IACL,UAAU;IACV,QAAQ;IACR,MAAM;IACN,QAAQ;IACR,KAAK;IACL,OAAO;IACP,OAAO;IACP,SAAS;IACT,QAAQ;IACR,QAAQ;IACR,MAAM;IACN,OAAO;IACP,QAAQ;IACR,UAAU;IACV,OAAO;IACP,OAAO;IACP,OAAO;IACP,MAAM;IACN,QAAQ;IACR,QAAQ;IACR,KAAK;IACL,OAAO;IACP,MAAM;IACN,KAAK;CACN,CAAC,CAAC;AACH,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,gBAAgB,EAAE,GAAG,oBAAoB,CAAC,CAAC,CAAC;AAE7E,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;AAEpC,SAAS,cAAc,CAAC,IAAY;IAClC,OAAO,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC,CAAC,IAAY,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AAC9E,CAAC;AAED,MAAM,iBAAiB;IAGrB,YAAY,QAA0B;;QACpC,IAAI,CAAC,gBAAgB,GAAG,MAAA,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC,mCAAI,YAAY,CAAC;IACrF,CAAC;IAED,QAAQ,CAAC,IAAY;QACnB,OAAO,IAAI,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC1F,CAAC;CACF;AAED,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,EAAE,CAAC,CAAC;AAExC,MAAsB,eAAe;IAYnC,YAAY,GAAqB;QAC/B,IAAI,CAAC,YAAY,GAAG,GAAG,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,eAAe;QACjB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE;YAC9B,IAAI,CAAC,oBAAoB,GAAG,IAAI,GAAG,CACjC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAC9E,CAAC;SACH;QACD,OAAO,IAAI,CAAC,oBAAoB,CAAC;IACnC,CAAC;IAED,kBAAkB,CAAC,QAA0B,EAAE,QAAQ,GAAG,kBAAU,CAAC,UAAU;QAC7E,6CAA6C;QAC7C,OAAO,QAAQ,KAAK,kBAAU,CAAC,SAAS;YACtC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACnF,CAAC,CAAC,QAAQ,KAAK,kBAAU,CAAC,UAAU;gBACpC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBACnF,CAAC,CAAC,QAAQ,CAAC;IACf,CAAC;IAED,mBAAmB,CAAC,QAA0B,EAAE,QAAQ,GAAG,kBAAU,CAAC,UAAU;;QAC9E,MAAM,QAAQ,GAAqB,EAAE,CAAC;QACtC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE;YACzD,OAAO,QAAQ,CAAC;SACjB;QACD,MAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,EAAE,EAAE,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;QAC9C,MAAM,KAAK,GAAkB,MAAA,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,mCAAI,EAAE,CAAC;QACxD,MAAM,OAAO,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC;QAC9B,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACvG,KAAK,MAAM,CAAC,KAAK,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,IAAI,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE,EAAE;YAC9F,IAAI,OAAO,EAAE;gBACX,MAAM,IAAI,GAAgB,IAAI,GAAG,EAAE,CAAC;gBACpC,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAC7F,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;aAClB;YACD,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;YAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;YACjE,QAAQ,CAAC,IAAI,CAAC;gBACZ,KAAK;gBACL,SAAS;gBACT,OAAO;aACR,CAAC,CAAC;SACJ;QACD,IAAI,OAAO,EAAE;YACX,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;SAC7B;QACD,OAAO,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACrD,CAAC;IAED,WAAW,CACT,QAA0B,EAC1B,IAAI,GAAG,8BAAsB,CAAC,SAAS,EACvC,IAAa;QAEb,IAAI,IAAI,KAAK,8BAAsB,CAAC,SAAS,EAAE;YAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;YAC7C,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;SACjC;QACD,OAAO,IAAI,KAAK,8BAAsB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC1F,CAAC;IAED;;;OAGG;IACH,aAAa,CAAC,QAA0B;QACtC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE;YACzD,OAAO;SACR;QACD,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,kBAAU,CAAC,UAAU,CAAC,CAAC;QAC3E,OAAO,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,KAAK,CAAC;YACrD,CAAC,iBACG,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAC3E,SAAS,EAAE,wBAAgB,CAAC,OAAO,IAChC,QAAQ,CAAC,CAAC,CAAC,EAElB,CAAC,CAAC,IAAI,CAAC;IACX,CAAC;IAED;;;;OAIG;IACH,eAAe,CAAC,QAA0B,EAAE,IAAI,GAAG,CAAC;QAClD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,IAAI,IAAI,GAAG,CAAC,EAAE;YACrE,OAAO;SACR;QACD,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,kBAAU,CAAC,UAAU,CAAC,CAAC;QAC3E,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE;YAC1C,OAAO;SACR;QACD,MAAM,WAAW,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,IAAI,WAAW,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE;YACrE,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,OAAO,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE;gBAC/G,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;aAC/B;SACF;QACD,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,iBAClC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EACnE,SAAS,EAAE,wBAAgB,CAAC,OAAO,IAChC,OAAO,EACV,CAAC,CAAC;IACN,CAAC;CACF;AA1HD,0CA0HC;AAED,SAAgB,YAAY,CAAC,IAAiB,EAAE,IAAiB;IAC/D,IAAI,gBAAgB,GAAG,CAAC,CAAC;IACzB,KAAK,MAAM,OAAO,IAAI,IAAI,EAAE;QAC1B,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;YACrB,gBAAgB,EAAE,CAAC;SACpB;KACF;IACD,OAAO,gBAAgB,GAAG,CAAC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,GAAG,gBAAgB,CAAC,CAAC;AACvE,CAAC;AARD,oCAQC"}
@@ -0,0 +1,24 @@
1
+ import { Tiktoken } from 'js-tiktoken';
2
+ import { Injector } from '@opensumi/di';
3
+ import { LanguageParser } from '../../languages/parser';
4
+ import { StrategyType } from '../types';
5
+ import type { ICompletionContext, ICompletionModel, MarkerItem } from '../types';
6
+ import type * as monaco from '@opensumi/ide-monaco';
7
+ export declare const addComment: (commentText?: string) => string;
8
+ export declare const getMarkerByLanguage: (text: string, language: string) => string;
9
+ export declare const getLanguageMarker: (language: string) => string;
10
+ export declare const getPathMarker: (path: string, language: string) => string;
11
+ export declare const getMarkerForSnippets: (text: string, language: string) => string;
12
+ export declare const getCroppedTextByLine: (text: string, maxTokenSize: number, textTokens: number[][], reverse?: boolean) => string;
13
+ /**
14
+ * 裁剪字符函数
15
+ * @param text 文本内容
16
+ * @param maxTokenSize 最大 token 数量
17
+ * @param strategy 分割策略
18
+ * @param language 语言类型,按函数分割仅在 typescript/javascript/typescriptreact 中支持
19
+ * @param reverse 是否反向裁剪,即从后往前裁剪
20
+ */
21
+ export declare const getCroppedText: (text: string, maxTokenSize: number, textTokens: number[][], strategy: StrategyType | undefined, tokenizer: Tiktoken, parser?: LanguageParser, minBlockSize?: number, reverse?: boolean, token?: monaco.CancellationToken) => Promise<string>;
22
+ export declare const getBeforePrompt: (promptPriority: MarkerItem[], context: ICompletionContext, promptConfig: ICompletionModel, leftTokenSize: number, injector: Injector, token: monaco.CancellationToken) => Promise<string | undefined>;
23
+ export declare const getAfterPrompt: (context: ICompletionContext, promptConfig: ICompletionModel, injector: Injector, token: monaco.CancellationToken) => Promise<string>;
24
+ //# sourceMappingURL=prompt.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"prompt.d.ts","sourceRoot":"","sources":["../../../../src/browser/inline-completions/prompt/prompt.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAEvC,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAExC,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAExD,OAAO,EAAE,YAAY,EAAyB,MAAM,UAAU,CAAC;AAK/D,OAAO,KAAK,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AACjF,OAAO,KAAK,KAAK,MAAM,MAAM,sBAAsB,CAAC;AAEpD,eAAO,MAAM,UAAU,iBAAkB,MAAM,WAK9C,CAAC;AAEF,eAAO,MAAM,mBAAmB,SAAU,MAAM,YAAY,MAAM,WAOjE,CAAC;AAEF,eAAO,MAAM,iBAAiB,aAAc,MAAM,WAoBjD,CAAC;AAGF,eAAO,MAAM,aAAa,SAAU,MAAM,YAAY,MAAM,WAA2D,CAAC;AAExH,eAAO,MAAM,oBAAoB,SAAU,MAAM,YAAY,MAAM,WAGlE,CAAC;AAEF,eAAO,MAAM,oBAAoB,SAAU,MAAM,gBAAgB,MAAM,cAAc,MAAM,EAAE,EAAE,8BAsC9F,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,cAAc,SACnB,MAAM,gBACE,MAAM,cACR,MAAM,EAAE,EAAE,iDAEX,QAAQ,WACV,cAAc,oDAGf,OAAO,iBAAiB,KAC/B,QAAQ,MAAM,CA2ChB,CAAC;AAEF,eAAO,MAAM,eAAe,mBACV,UAAU,EAAE,WACnB,kBAAkB,gBACb,gBAAgB,iBACf,MAAM,YACX,QAAQ,SACX,OAAO,iBAAiB,gCA+FhC,CAAC;AAEF,eAAO,MAAM,cAAc,YAChB,kBAAkB,gBACb,gBAAgB,YACpB,QAAQ,SACX,OAAO,iBAAiB,oBAyBhC,CAAC"}