ishangsf/DeepCodeGeniusWeb-vscode
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: Optimize code completion performance in InlineCompletionProvider

Browse Source
This commit is contained in:
bobo.yang 2024-05-18 16:48:12 +08:00
parent 8e31ea0ea8
commit e12553f2d7
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/contributes/codecomplete/codecomplete.ts
View File

@ -120,9 +120,9 @@ export class InlineCompletionProvider implements vscode.InlineCompletionItemProv
// }
async codeComplete(document: vscode.TextDocument, position: vscode.Position, context: vscode.InlineCompletionContext, token: vscode.CancellationToken): Promise<CodeCompleteResult | undefined> {
const startTime = process.hrtime();
GitDiffWatcher.getInstance().tryRun();
const completeDebug = this.devchatConfig.get("complete_debug");
// create prompt
const fsPath = document.uri.fsPath;
const fileContent = document.getText();
@ -154,12 +154,21 @@ export class InlineCompletionProvider implements vscode.InlineCompletionItemProv
return undefined;
}
const endTime = process.hrtime(startTime);
const duration = endTime[0] + endTime[1] / 1e9;
logger.channel()?.debug(`Make prompt took ${duration} seconds`);
const startTimeLLM = process.hrtime();
const completor = new LLMStreamComplete(token, lines, position.line, position.character);
const response = await completor.llmStreamComplete(prompt);
if (!response || response.code.length === 0) {
return undefined;
}
const endTimeLLM = process.hrtime(startTimeLLM);
const durationLLM = endTimeLLM[0] + endTimeLLM[1] / 1e9;
logger.channel()?.debug(`LLMStreamComplete took ${durationLLM} seconds`);
if (token.isCancellationRequested) {
return undefined;
}
@ -170,8 +179,6 @@ export class InlineCompletionProvider implements vscode.InlineCompletionItemProv
}
async provideInlineCompletionItems(document: vscode.TextDocument, position: vscode.Position, context: vscode.InlineCompletionContext, token: vscode.CancellationToken): Promise<vscode.InlineCompletionItem[] | null> {
const completeDebug = this.devchatConfig.get("complete_debug");
const result = await this.debouncer.debounce();
if (!result) {
return [];