jan-leila/ai-code
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

canceled ollama requests on token canceled

Browse source
This commit is contained in:
Leyla Becker 2025-07-19 19:50:34 -05:00
parent 02e65c5700
commit f93d24dc39
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/extension.ts
View file

@ -87,7 +87,7 @@ const tokenProvider = async (
document: vscode.TextDocument, document: vscode.TextDocument,
position: vscode.Position, position: vscode.Position,
_context: vscode.InlineCompletionContext, _context: vscode.InlineCompletionContext,
_token: vscode.CancellationToken, token: vscode.CancellationToken,
) => { ) => {
const prefix = document.getText(new vscode.Range(0, 0, position.line, position.character)); const prefix = document.getText(new vscode.Range(0, 0, position.line, position.character));
@ -114,6 +114,10 @@ const tokenProvider = async (
}, },
}); });
token.onCancellationRequested(() => {
response.abort();
});
const resultBuffer: string[] = await new Promise(async (resolve, reject) => { const resultBuffer: string[] = await new Promise(async (resolve, reject) => {
const buffer: string[] = []; const buffer: string[] = [];
const timeout = setTimeout(() => { const timeout = setTimeout(() => {