Wire model-provided eagerness prompts

Co-authored-by: Copilot App <223556219+Copilot@users.noreply.github.com>

Copilot-Session: 2bce40ed-b23b-4b38-8c2a-11a9b30a08ce

Author: bstee615

extensions/copilot/src/platform/inlineEdits/common/dataTypes/inlineEditsModelsTypes.ts

@@ -3,20 +3,23 @@
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 
-import { IValidator, vArray, vObj, vString } from '../../../configuration/common/validator';
+import { IValidator, vArray, vObj, vString, vUndefined, vUnion } from '../../../configuration/common/validator';
 
 export namespace WireTypes {
 
 	export namespace Capabilities {
 		export type t = {
 			promptStrategy: string;
+			eagernessPrompt?: string;
 		};
 		export function is(obj: unknown): obj is t {
 			return !!obj && typeof obj === 'object' &&
-				typeof (obj as t).promptStrategy === 'string';
+				typeof (obj as t).promptStrategy === 'string' &&
+				((obj as t).eagernessPrompt === undefined || typeof (obj as t).eagernessPrompt === 'string');
 		}
 		export const validator: IValidator<t> = vObj({
 			promptStrategy: vString(),
+			eagernessPrompt: vUnion(vString(), vUndefined()),
 		});
 	}
 
@@ -54,4 +57,3 @@ export namespace WireTypes {
 		}
 	}
 }
-

extensions/copilot/src/platform/inlineEdits/node/inlineEditsModelService.ts

@@ -263,6 +263,7 @@ export class InlineEditsModelService extends Disposable implements IInlineEditsM
 				return {
 					modelName: m.name,
 					promptingStrategy: m.capabilities.promptStrategy,
+					eagernessPrompt: m.capabilities.eagernessPrompt === 'aggressionHighLow' ? m.capabilities.eagernessPrompt : undefined,
 					includeTagsInCurrentFile: false, // FIXME@ulugbekna: determine this based on model capabilities and config
 					source: ModelSource.Fetched,
 					lintOptions: undefined,

extensions/copilot/src/platform/proxyModels/test/node/proxyModelsService.spec.ts

@@ -10,13 +10,46 @@ import { IInstantiationService } from '../../../../util/vs/platform/instantiatio
 import { CopilotToken, createTestExtendedTokenInfo } from '../../../authentication/common/copilotToken';
 import { ICopilotTokenStore } from '../../../authentication/common/copilotTokenStore';
 import { getEditorVersionHeaders, IEnvService } from '../../../env/common/envService';
+import { WireTypes } from '../../../inlineEdits/common/dataTypes/inlineEditsModelsTypes';
 import { FetchOptions, IAbortController, IFetcherService, PaginationOptions, Response, WebSocketConnection } from '../../../networking/common/fetcherService';
 import { createFakeResponse } from '../../../test/node/fetcher';
 import { createPlatformServices } from '../../../test/node/services';
 import { ProxyModelsService } from '../../node/proxyModelsService';
 
 suite('ProxyModelsService', function () {
 
+	test('validates optional eagerness prompts from the models response', function () {
+		const withEagernessPrompt = WireTypes.ModelList.validator.validate({
+			models: [{
+				serviceType: 'NESChat',
+				name: 'test-model',
+				provider: 'test-provider',
+				capabilities: {
+					promptStrategy: 'patchBased02Optimized',
+					eagernessPrompt: 'aggressionHighLow',
+				},
+			}],
+		});
+		const withoutEagernessPrompt = WireTypes.ModelList.validator.validate({
+			models: [{
+				serviceType: 'NESChat',
+				name: 'test-model',
+				provider: 'test-provider',
+				capabilities: {
+					promptStrategy: 'patchBased02Optimized',
+				},
+			}],
+		});
+
+		assert.deepStrictEqual({
+			withEagernessPrompt: withEagernessPrompt.content?.models[0].capabilities.eagernessPrompt,
+			withoutEagernessPrompt: withoutEagernessPrompt.content?.models[0].capabilities.eagernessPrompt,
+		}, {
+			withEagernessPrompt: 'aggressionHighLow',
+			withoutEagernessPrompt: undefined,
+		});
+	});
+
 	test('includes editor-related headers when fetching the models list', async function () {
 		let capturedHeaders: { [name: string]: string } | undefined;