@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.
- package/lib/browser/inline-completions/completeProvider.d.ts +5 -3
- package/lib/browser/inline-completions/completeProvider.d.ts.map +1 -1
- package/lib/browser/inline-completions/completeProvider.js +16 -30
- package/lib/browser/inline-completions/completeProvider.js.map +1 -1
- package/lib/browser/inline-completions/constants.d.ts +2 -6
- package/lib/browser/inline-completions/constants.d.ts.map +1 -1
- package/lib/browser/inline-completions/constants.js +81 -4
- package/lib/browser/inline-completions/constants.js.map +1 -1
- package/lib/browser/inline-completions/prompt/const.d.ts +9 -0
- package/lib/browser/inline-completions/prompt/const.d.ts.map +1 -0
- package/lib/browser/inline-completions/prompt/const.js +284 -0
- package/lib/browser/inline-completions/prompt/const.js.map +1 -0
- package/lib/browser/inline-completions/prompt/importedFiles.d.ts +6 -0
- package/lib/browser/inline-completions/prompt/importedFiles.d.ts.map +1 -0
- package/lib/browser/inline-completions/prompt/importedFiles.js +77 -0
- package/lib/browser/inline-completions/prompt/importedFiles.js.map +1 -0
- package/lib/browser/inline-completions/prompt/jaccardMatcher.d.ts +31 -0
- package/lib/browser/inline-completions/prompt/jaccardMatcher.d.ts.map +1 -0
- package/lib/browser/inline-completions/prompt/jaccardMatcher.js +75 -0
- package/lib/browser/inline-completions/prompt/jaccardMatcher.js.map +1 -0
- package/lib/browser/inline-completions/prompt/languages.d.ts +4 -0
- package/lib/browser/inline-completions/prompt/languages.d.ts.map +1 -0
- package/lib/browser/inline-completions/prompt/languages.js +67 -0
- package/lib/browser/inline-completions/prompt/languages.js.map +1 -0
- package/lib/browser/inline-completions/prompt/matcher.d.ts +42 -0
- package/lib/browser/inline-completions/prompt/matcher.d.ts.map +1 -0
- package/lib/browser/inline-completions/prompt/matcher.js +279 -0
- package/lib/browser/inline-completions/prompt/matcher.js.map +1 -0
- package/lib/browser/inline-completions/prompt/prompt.d.ts +24 -0
- package/lib/browser/inline-completions/prompt/prompt.d.ts.map +1 -0
- package/lib/browser/inline-completions/prompt/prompt.js +242 -0
- package/lib/browser/inline-completions/prompt/prompt.js.map +1 -0
- package/lib/browser/inline-completions/prompt/similarSnippets.d.ts +9 -0
- package/lib/browser/inline-completions/prompt/similarSnippets.d.ts.map +1 -0
- package/lib/browser/inline-completions/prompt/similarSnippets.js +110 -0
- package/lib/browser/inline-completions/prompt/similarSnippets.js.map +1 -0
- package/lib/browser/inline-completions/prompt/tokenizer.d.ts +4 -0
- package/lib/browser/inline-completions/prompt/tokenizer.d.ts.map +1 -0
- package/lib/browser/inline-completions/prompt/tokenizer.js +18 -0
- package/lib/browser/inline-completions/prompt/tokenizer.js.map +1 -0
- package/lib/browser/inline-completions/provider.d.ts +5 -12
- package/lib/browser/inline-completions/provider.d.ts.map +1 -1
- package/lib/browser/inline-completions/provider.js +79 -44
- package/lib/browser/inline-completions/provider.js.map +1 -1
- package/lib/browser/inline-completions/types.d.ts +154 -0
- package/lib/browser/inline-completions/types.d.ts.map +1 -0
- package/lib/browser/inline-completions/types.js +59 -0
- package/lib/browser/inline-completions/types.js.map +1 -0
- package/lib/browser/languages/parser.d.ts +6 -0
- package/lib/browser/languages/parser.d.ts.map +1 -1
- package/lib/browser/languages/parser.js +140 -1
- package/lib/browser/languages/parser.js.map +1 -1
- package/lib/browser/languages/tree-sitter/language-facts/types.d.ts +1 -1
- package/lib/browser/languages/tree-sitter/wasm-manager.d.ts +2 -2
- package/lib/browser/languages/tree-sitter/wasm-manager.d.ts.map +1 -1
- package/lib/browser/languages/tree-sitter/wasm-manager.js +2 -2
- package/lib/browser/languages/tree-sitter/wasm-manager.js.map +1 -1
- package/lib/common/utils.d.ts +5 -0
- package/lib/common/utils.d.ts.map +1 -0
- package/lib/common/utils.js +26 -0
- package/lib/common/utils.js.map +1 -0
- package/package.json +20 -19
- package/src/browser/inline-completions/completeProvider.ts +26 -35
- package/src/browser/inline-completions/constants.ts +87 -7
- package/src/browser/inline-completions/prompt/const.ts +286 -0
- package/src/browser/inline-completions/prompt/importedFiles.ts +92 -0
- package/src/browser/inline-completions/prompt/jaccardMatcher.ts +98 -0
- package/src/browser/inline-completions/prompt/languages.ts +65 -0
- package/src/browser/inline-completions/prompt/matcher.ts +328 -0
- package/src/browser/inline-completions/prompt/prompt.ts +297 -0
- package/src/browser/inline-completions/prompt/similarSnippets.ts +130 -0
- package/src/browser/inline-completions/prompt/tokenizer.ts +16 -0
- package/src/browser/inline-completions/provider.ts +107 -42
- package/src/browser/inline-completions/types.ts +169 -0
- package/src/browser/languages/parser.ts +147 -0
- package/src/browser/languages/tree-sitter/wasm-manager.ts +3 -3
- 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 @@
|
|
|
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"}
|