Fix variables in chatAgents2 API requests, add tests (#195529)

* Fix variables in chatAgents2 API requests * Enable file references and the 'used references' section by default in Insiders * Add integration tests for chat * Fix equality * fix test
2026-04-28 12:33:35 +01:00 · 2023-10-12 21:27:30 -07:00
parent e13be231e7
commit 6c02f61149
7 changed files with 172 additions and 6 deletions
--- a/extensions/vscode-api-tests/src/singlefolder-tests/chat.test.ts
+++ b/extensions/vscode-api-tests/src/singlefolder-tests/chat.test.ts
@@ -0,0 +1,80 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as assert from 'assert';
+import 'mocha';
+import { CancellationToken, chat, ChatAgentRequest, ChatVariableLevel, CompletionItemKind, Disposable, interactive, InteractiveProgress, InteractiveRequest, InteractiveResponseForProgress, InteractiveSession, InteractiveSessionState, Progress, ProviderResult } from 'vscode';
+import { assertNoRpc, closeAllEditors, DeferredPromise, disposeAll } from '../utils';
+
+suite('chat', () => {
+	let disposables: Disposable[] = [];
+	setup(() => {
+		disposables = [];
+	});
+
+	teardown(async function () {
+		assertNoRpc();
+		await closeAllEditors();
+		disposeAll(disposables);
+	});
+
+	function getDeferredForRequest(): DeferredPromise<ChatAgentRequest> {
+		disposables.push(interactive.registerInteractiveSessionProvider('provider', {
+			prepareSession: (_initialState: InteractiveSessionState | undefined, _token: CancellationToken): ProviderResult<InteractiveSession> => {
+				return {
+					requester: { name: 'test' },
+					responder: { name: 'test' },
+				};
+			},
+
+			provideResponseWithProgress: (_request: InteractiveRequest, _progress: Progress<InteractiveProgress>, _token: CancellationToken): ProviderResult<InteractiveResponseForProgress> => {
+				return null;
+			},
+
+			provideSlashCommands: (_session, _token) => {
+				return [{ command: 'hello', title: 'Hello', kind: CompletionItemKind.Text }];
+			},
+
+			removeRequest: (_session: InteractiveSession, _requestId: string): void => {
+				throw new Error('Function not implemented.');
+			}
+		}));
+
+		const deferred = new DeferredPromise<ChatAgentRequest>();
+		const agent = chat.createChatAgent('agent', (request, _context, _progress, _token) => {
+			deferred.complete(request);
+			return null;
+		});
+		agent.slashCommandProvider = {
+			provideSlashCommands: (_token) => {
+				return [{ name: 'hello', description: 'Hello' }];
+			}
+		};
+		disposables.push(agent);
+		return deferred;
+	}
+
+	test('agent and slash command', async () => {
+		const deferred = getDeferredForRequest();
+		interactive.sendInteractiveRequestToProvider('provider', { message: '@agent /hello friend' });
+		const lastResult = await deferred.p;
+		assert.deepStrictEqual(lastResult.slashCommand, { name: 'hello', description: 'Hello' });
+		assert.strictEqual(lastResult.prompt, 'friend');
+	});
+
+	test('agent and variable', async () => {
+		disposables.push(chat.registerVariable('myVar', 'My variable', {
+			resolve(_name, _context, _token) {
+				return [{ level: ChatVariableLevel.Full, value: 'myValue' }];
+			}
+		}));
+
+		const deferred = getDeferredForRequest();
+		interactive.sendInteractiveRequestToProvider('provider', { message: '@agent hi #myVar' });
+		const lastResult = await deferred.p;
+		assert.strictEqual(lastResult.prompt, 'hi [#myVar](values:myVar)');
+		assert.strictEqual(lastResult.variables['myVar'][0].value, 'myValue');
+	});
+});
--- a/extensions/vscode-api-tests/src/singlefolder-tests/interactive.test.ts
+++ b/extensions/vscode-api-tests/src/singlefolder-tests/interactive.test.ts
@@ -0,0 +1,62 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as assert from 'assert';
+import 'mocha';
+import { CancellationToken, CompletionItemKind, Disposable, interactive, InteractiveProgress, InteractiveRequest, InteractiveResponseForProgress, InteractiveSession, InteractiveSessionState, Progress, ProviderResult } from 'vscode';
+import { assertNoRpc, closeAllEditors, DeferredPromise, disposeAll } from '../utils';
+
+suite('InteractiveSessionProvider', () => {
+	let disposables: Disposable[] = [];
+	setup(async () => {
+		disposables = [];
+	});
+
+	teardown(async function () {
+		assertNoRpc();
+		await closeAllEditors();
+		disposeAll(disposables);
+	});
+
+	function getDeferredForRequest(): DeferredPromise<InteractiveRequest> {
+		const deferred = new DeferredPromise<InteractiveRequest>();
+		disposables.push(interactive.registerInteractiveSessionProvider('provider', {
+			prepareSession: (_initialState: InteractiveSessionState | undefined, _token: CancellationToken): ProviderResult<InteractiveSession> => {
+				return {
+					requester: { name: 'test' },
+					responder: { name: 'test' },
+				};
+			},
+
+			provideResponseWithProgress: (request: InteractiveRequest, _progress: Progress<InteractiveProgress>, _token: CancellationToken): ProviderResult<InteractiveResponseForProgress> => {
+				deferred.complete(request);
+				return null;
+			},
+
+			provideSlashCommands: (_session, _token) => {
+				return [{ command: 'hello', title: 'Hello', kind: CompletionItemKind.Text }];
+			},
+
+			removeRequest: (_session: InteractiveSession, _requestId: string): void => {
+				throw new Error('Function not implemented.');
+			}
+		}));
+		return deferred;
+	}
+
+	test('plain text query', async () => {
+		const deferred = getDeferredForRequest();
+		interactive.sendInteractiveRequestToProvider('provider', { message: 'hello' });
+		const lastResult = await deferred.p;
+		assert.strictEqual(lastResult.message, 'hello');
+	});
+
+	test('slash command', async () => {
+		const deferred = getDeferredForRequest();
+		interactive.sendInteractiveRequestToProvider('provider', { message: '/hello' });
+		const lastResult = await deferred.p;
+		assert.strictEqual(lastResult.message, '/hello');
+	});
+});