Handle warning and info messages froom SDK (#333851)

* Handle warning and info messages froom SDK

* fix ci

Author: lramos15

src/vs/platform/agentHost/common/agentModelNotices.ts

@@ -0,0 +1,64 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import type { IAgentModelInfo } from './agent.js';
+import type { SessionModelInfo } from './state/protocol/state.js';
+
+const DATA_RETENTION_WARNING_CODE = 'data_retention';
+const PENDING_DEPRECATION_WARNING_CODE = 'model_pending_deprecation';
+
+type ModelMessage = { readonly code: string; readonly message: string };
+type ModelNoticeSource = {
+	readonly warningText?: { readonly dataRetention?: string };
+	readonly infoMessages?: readonly ModelMessage[];
+	readonly warningMessages?: readonly ModelMessage[];
+};
+
+/** Converts SDK model messages to model-picker metadata. */
+export function createAgentModelNoticesMeta(source: ModelNoticeSource): Record<string, unknown> | undefined {
+	const warningText: Record<string, string> = {};
+	const infoText: Record<string, string> = {};
+	if (source.warningText?.dataRetention) {
+		warningText[DATA_RETENTION_WARNING_CODE] = source.warningText.dataRetention;
+	}
+	for (const { code, message } of source.infoMessages ?? []) {
+		if (message) {
+			const target = code === PENDING_DEPRECATION_WARNING_CODE ? warningText : infoText;
+			target[code || 'info'] = message;
+		}
+	}
+	for (const { code, message } of source.warningMessages ?? []) {
+		if (message) {
+			warningText[code || 'warning'] = message;
+		}
+	}
+	const rowWarning = source.warningMessages?.find(({ message }) => !!message)?.message
+		?? warningText[PENDING_DEPRECATION_WARNING_CODE];
+	const result = {
+		...(Object.keys(warningText).length > 0 ? { warningText } : {}),
+		...(Object.keys(infoText).length > 0 ? { infoText } : {}),
+		...(rowWarning ? { rowWarning } : {}),
+	};
+	return Object.keys(result).length > 0 ? result : undefined;
+}
+
+/** Reads model-picker messages from Agent Host metadata. */
+export function readAgentModelNoticesMeta(model: IAgentModelInfo | SessionModelInfo) {
+	const meta = model._meta;
+	return {
+		warningText: asStringDictionary(meta?.warningText),
+		infoText: asStringDictionary(meta?.infoText),
+		rowWarning: typeof meta?.rowWarning === 'string' && meta.rowWarning.length > 0 ? meta.rowWarning : undefined,
+	};
+}
+
+function asStringDictionary(value: unknown): Record<string, string> | undefined {
+	if (!value || typeof value !== 'object' || Array.isArray(value)) {
+		return undefined;
+	}
+	const entries = Object.entries(value).filter((entry): entry is [string, string] =>
+		entry[0].length > 0 && typeof entry[1] === 'string' && entry[1].length > 0);
+	return entries.length > 0 ? Object.fromEntries(entries) : undefined;
+}

src/vs/platform/agentHost/node/copilot/copilotAgent.ts

@@ -36,6 +36,7 @@ import { IAgentHostCheckpointService } from '../../common/agentHostCheckpointSer
 import type { IAgentHostClientTelemetryContext } from '../../common/agentHostTelemetry.js';
 import { IAgentHostReviewService } from '../../common/agentHostReviewService.js';
 import { createPricingMetaFromBilling, hasLongContextSurcharge, normalizeCAPIBilling, type ICAPIModelBilling } from '../../common/agentModelPricing.js';
+import { createAgentModelNoticesMeta } from '../../common/agentModelNotices.js';
 import { createAgentModelByokMeta } from '../../common/agentModelByokMeta.js';
 import { AgentHostConfigKey, agentHostCustomizationConfigSchema, DEFAULT_SESSION_CUSTOMIZATION_DISCOVERY_MODE, toContainerCustomization } from '../../common/agentHostCustomizationConfig.js';
 import { CopilotCliConfigKey, CopilotCliVSCodeAssignmentContextKey, copilotCliConfigSchema, DEFAULT_COPILOT_RUBBER_DUCK_ENABLED, type CopilotSdkLogLevelSetting } from '../../common/copilotCliConfig.js';
@@ -2296,11 +2297,10 @@ export class CopilotAgent extends Disposable implements IAgent {
 		};
 	}
 
-	/**
-	 * Builds the open `_meta` model picker bag from the SDK's billing and picker metadata.
-	 */
 	private _createModelPickerMeta(modelInfo: CopilotModelInfo, billing: ICAPIModelBilling | undefined): Record<string, unknown> | undefined {
-		return createPricingMetaFromBilling(billing, modelInfo.modelPickerPriceCategory, modelInfo.modelPickerCategory);
+		const pricing = createPricingMetaFromBilling(billing, modelInfo.modelPickerPriceCategory, modelInfo.modelPickerCategory);
+		const notices = isAutoModel(modelInfo.id) ? undefined : createAgentModelNoticesMeta(modelInfo);
+		return pricing || notices ? { ...pricing, ...notices } : undefined;
 	}
 
 	private _createModelConfigSchema(m: CopilotModelInfo, billing: ICAPIModelBilling | undefined): ConfigSchema | undefined {

src/vs/platform/agentHost/test/node/copilotAgent.test.ts

@@ -429,6 +429,9 @@ interface ITestCopilotModelInfo {
 	readonly billing?: CopilotModelInfo['billing'];
 	readonly modelPickerCategory?: CopilotModelInfo['modelPickerCategory'];
 	readonly modelPickerPriceCategory?: CopilotModelInfo['modelPickerPriceCategory'];
+	readonly warningText?: CopilotModelInfo['warningText'];
+	readonly infoMessages?: CopilotModelInfo['infoMessages'];
+	readonly warningMessages?: CopilotModelInfo['warningMessages'];
 	readonly supportedReasoningEfforts?: CopilotModelInfo['supportedReasoningEfforts'];
 }
 
@@ -469,6 +472,9 @@ function toSdkModelInfo(model: ITestCopilotModelInfo): CopilotModelInfo {
 		...(model.billing ? { billing: model.billing } : {}),
 		...(model.modelPickerCategory ? { modelPickerCategory: model.modelPickerCategory } : {}),
 		...(model.modelPickerPriceCategory ? { modelPickerPriceCategory: model.modelPickerPriceCategory } : {}),
+		...(model.warningText ? { warningText: model.warningText } : {}),
+		...(model.infoMessages ? { infoMessages: model.infoMessages } : {}),
+		...(model.warningMessages ? { warningMessages: model.warningMessages } : {}),
 		...(model.supportedReasoningEfforts ? { supportedReasoningEfforts: model.supportedReasoningEfforts } : {}),
 	};
 }
@@ -4853,7 +4859,7 @@ suite('CopilotAgent', () => {
 		}
 	});
 
-	test('models include picker and promo metadata when the SDK provides it', async () => {
+	test('models include picker, notice, and promo metadata when the SDK provides it', async () => {
 		const agent = createTestAgent(disposables, {
 			copilotClient: new TestCopilotClient([], [{
 				id: 'claude-sonnet',
@@ -4878,6 +4884,16 @@ suite('CopilotAgent', () => {
 				},
 				modelPickerCategory: 'powerful',
 				modelPickerPriceCategory: 'medium',
+				warningText: {
+					dataRetention: 'Prompts are retained for 30 days.',
+				},
+				infoMessages: [
+					{ code: 'model_pending_deprecation', message: 'Claude Sonnet will be retired soon.' },
+					{ code: 'model_relocated', message: 'Claude Sonnet now serves from a new region.' },
+				],
+				warningMessages: [
+					{ code: 'model_degraded', message: 'Claude Sonnet is currently degraded.' },
+				],
 			}]),
 		});
 		try {
@@ -4894,6 +4910,15 @@ suite('CopilotAgent', () => {
 				longContextOutputCost: 22.5,
 				priceCategory: 'medium',
 				category: 'powerful',
+				warningText: {
+					data_retention: 'Prompts are retained for 30 days.',
+					model_pending_deprecation: 'Claude Sonnet will be retired soon.',
+					model_degraded: 'Claude Sonnet is currently degraded.',
+				},
+				infoText: {
+					model_relocated: 'Claude Sonnet now serves from a new region.',
+				},
+				rowWarning: 'Claude Sonnet is currently degraded.',
 				promo: {
 					id: 'summer-sale',
 					discountPercent: 25,

src/vs/workbench/contrib/chat/browser/agentSessions/agentHost/agentHostLanguageModelProvider.ts

@@ -4,9 +4,11 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { CancellationToken } from '../../../../../../base/common/cancellation.js';
+import { Codicon } from '../../../../../../base/common/codicons.js';
 import { Emitter } from '../../../../../../base/common/event.js';
 import { Disposable } from '../../../../../../base/common/lifecycle.js';
 import { localize } from '../../../../../../nls.js';
+import { readAgentModelNoticesMeta } from '../../../../../../platform/agentHost/common/agentModelNotices.js';
 import { ConfigSchema, SessionModelInfo } from '../../../../../../platform/agentHost/common/state/sessionState.js';
 import { readAgentModelPricingMeta } from '../../../../../../platform/agentHost/common/agentModelPricing.js';
 import { readAgentModelByokIdentifier } from '../../../../../../platform/agentHost/common/agentModelByokMeta.js';
@@ -68,16 +70,17 @@ export class AgentHostLanguageModelProvider extends Disposable implements ILangu
 				const multiplierNumeric = pricing.multiplierNumeric;
 				// "Auto" advertises the auto-mode discount (detail) + description (tooltip). microsoft/vscode#321778, #321659.
 				const isAuto = m.id === AUTO_RAW_MODEL_ID;
+				const notices = isAuto ? undefined : readAgentModelNoticesMeta(m);
 				const discountPercent = pricing.discountPercent;
 				// Guard against a non-finite or out-of-range value from the open `_meta` bag so we never render
 				// nonsense like "Infinity% discount"; the documented range is a whole number in (0, 100].
 				const hasDiscount = typeof discountPercent === 'number' && discountPercent > 0 && discountPercent <= 100;
 				const detail = isAuto && hasDiscount
 					? localize('agentHost.auto.discount', "{0}% discount", discountPercent)
 					: undefined;
-				const tooltip = isAuto
+				const tooltip = notices?.rowWarning ?? (isAuto
 					? ILanguageModelChatMetadata.getAutoModelDescription(hasDiscount ? discountPercent : undefined)
-					: undefined;
+					: undefined);
 				const modelGroup = this._modelGroupFor(m);
 				const byokModelIdentifier = readAgentModelByokIdentifier(m);
 				return {
@@ -95,6 +98,9 @@ export class AgentHostLanguageModelProvider extends Disposable implements ILangu
 						maxOutputTokens: m.maxOutputTokens ?? 0,
 						isDefaultForLocation: {},
 						isUserSelectable: true,
+						statusIcon: notices?.rowWarning ? Codicon.warning : undefined,
+						warningText: notices?.warningText,
+						infoText: notices?.infoText,
 						pricing: multiplierNumeric !== undefined ? `${multiplierNumeric}x` : undefined,
 						multiplierNumeric,
 						inputCost: pricing.inputCost,

src/vs/workbench/contrib/chat/test/browser/agentSessions/agentHostLanguageModelProvider.test.ts

@@ -5,6 +5,7 @@
 
 import assert from 'assert';
 import { CancellationToken } from '../../../../../../base/common/cancellation.js';
+import { Codicon } from '../../../../../../base/common/codicons.js';
 import { ensureNoDisposablesAreLeakedInTestSuite } from '../../../../../../base/test/common/utils.js';
 import { SessionModelInfo } from '../../../../../../platform/agentHost/common/state/sessionState.js';
 import { ILanguageModelChatMetadata } from '../../../common/languageModels.js';
@@ -136,6 +137,28 @@ suite('AgentHostLanguageModelProvider', () => {
 		]);
 	});
 
+	test('carries model notices and flags row warnings', async () => {
+		const provider = createProvider();
+		provider.updateModels([makeModel('gpt-5', {
+			warningText: { model_degraded: 'GPT-5 is currently degraded.' },
+			infoText: { model_relocated: 'GPT-5 now serves from a new region.' },
+			rowWarning: 'GPT-5 is currently degraded.',
+		})]);
+
+		const metadata = (await provider.provideLanguageModelChatInfo(undefined, CancellationToken.None))[0].metadata;
+		assert.deepStrictEqual({
+			tooltip: metadata.tooltip,
+			statusIcon: metadata.statusIcon?.id,
+			warningText: metadata.warningText,
+			infoText: metadata.infoText,
+		}, {
+			tooltip: 'GPT-5 is currently degraded.',
+			statusIcon: Codicon.warning.id,
+			warningText: { model_degraded: 'GPT-5 is currently degraded.' },
+			infoText: { model_relocated: 'GPT-5 now serves from a new region.' },
+		});
+	});
+
 	test('derives the picker group from the model-id prefix, not the harness provider', async () => {
 		const provider = createProvider();
 		// The agent host reports every model under the harness provider (`copilotcli`);