update with api feedback

This commit is contained in:
Connor Peet
2025-04-23 11:09:15 -07:00
parent dd5b6c1235
commit 41cc1e2836
14 changed files with 52 additions and 49 deletions
@@ -138,12 +138,12 @@ export class ExtHostLanguageModelTools implements ExtHostLanguageModelToolsShape
};
}
let progress: vscode.Progress<{ message?: string; increment?: number }> | undefined;
let progress: vscode.Progress<{ message?: string | vscode.MarkdownString; increment?: number }> | undefined;
if (isProposedApiEnabled(item.extension, 'toolProgress')) {
progress = {
report: value => {
this._proxy.$acceptToolProgress(dto.callId, {
message: value.message,
message: typeConvert.MarkdownString.fromStrict(value.message),
increment: value.increment,
total: 100,
});
@@ -151,7 +151,8 @@ export class ExtHostLanguageModelTools implements ExtHostLanguageModelToolsShape
};
}
const extensionResult = await raceCancellation(Promise.resolve(item.tool.invoke(options, token, progress!)), token);
// todo: 'any' cast because TS can't handle the overloads
const extensionResult = await raceCancellation(Promise.resolve((item.tool.invoke as any)(options, token, progress!)), token);
if (!extensionResult) {
throw new CancellationError();
}