development/vscode
development/vscode
1
0
Fork 0
mirror of https://github.com/Microsoft/vscode synced 2024-08-27 04:49:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

send onLanguageModelAccess activation event (#205995)

Browse source 
https://github.com/microsoft/vscode/issues/205726
This commit is contained in:
Johannes Rieken 2024-02-22 16:39:24 +01:00 committed by GitHub
parent 0b82c6e374
commit 838a59f240
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/vs/workbench/api/browser/mainThreadChatProvider.ts
View file

@ -3,7 +3,6 @@
* Licensed under the MIT License. See License.txt in the project root for license information.
*--------------------------------------------------------------------------------------------*/
import { timeout } from 'vs/base/common/async';
import { CancellationToken } from 'vs/base/common/cancellation';
import { Emitter, Event } from 'vs/base/common/event';
import { Disposable, DisposableMap, DisposableStore, IDisposable, toDisposable } from 'vs/base/common/lifecycle';
@ -82,15 +81,20 @@ export class MainThreadChatProvider implements MainThreadChatProviderShape {
}
async $prepareChatAccess(extension: ExtensionIdentifier, providerId: string, justification?: string): Promise<IChatResponseProviderMetadata | undefined> {
const activate = this._extensionService.activateByEvent(`onLanguageModelAccess:${providerId}`);
const metadata = this._chatProviderService.lookupChatResponseProvider(providerId);
// TODO: This should use a real activation event. Perhaps following what authentication does.
for (let i = 0; i < 3; i++) {
if (metadata) {
return metadata;
}
await timeout(2000);
if (metadata) {
return metadata;
}
return undefined;
await Promise.race([
activate,
Event.toPromise(Event.filter(this._chatProviderService.onDidChangeProviders, e => Boolean(e.added?.includes(providerId))))
]);
return this._chatProviderService.lookupChatResponseProvider(providerId);
}
async $fetchResponse(extension: ExtensionIdentifier, providerId: string, requestId: number, messages: IChatMessage[], options: {}, token: CancellationToken): Promise<any> {