Support degraded model status (#1148)

Author: lramos15

src/extension/conversation/vscode-node/languageModelAccess.ts

@@ -129,7 +129,9 @@ export class LanguageModelAccess extends Disposable implements IExtensionContrib
 
 			const sanitizedModelName = endpoint.name.replace(/\(Preview\)/g, '').trim();
 			let modelDescription: string | undefined;
-			if (endpoint.model === AutoChatEndpoint.id) {
+			if (endpoint.degradationReason) {
+				modelDescription = endpoint.degradationReason;
+			} else if (endpoint.model === AutoChatEndpoint.id) {
 				if (this._authenticationService.copilotToken?.isNoAuthUser) {
 					modelDescription = localize('languageModel.autoTooltipNoAuth', 'Auto selects the best model for your request based on capacity and performance.');
 				} else {
@@ -178,6 +180,7 @@ export class LanguageModelAccess extends Disposable implements IExtensionContrib
 				tooltip: modelDescription,
 				detail: modelDetail,
 				category: modelCategory,
+				statusIcon: endpoint.degradationReason ? new vscode.ThemeIcon('warning') : undefined,
 				version: endpoint.version,
 				maxInputTokens: endpoint.modelMaxPromptTokens - baseCount - BaseTokensPerCompletion,
 				maxOutputTokens: endpoint.maxOutputTokens,

src/platform/endpoint/common/endpointProvider.ts

@@ -81,6 +81,7 @@ export interface IModelAPIResponse {
 	is_chat_default: boolean;
 	is_chat_fallback: boolean;
 	version: string;
+	warning_message?: string;
 	billing?: { is_premium: boolean; multiplier: number; restricted_to?: string[] };
 	capabilities: IChatModelCapabilities | ICompletionModelCapabilities | IEmbeddingModelCapabilities;
 	supported_endpoints?: ModelSupportedEndpoint[];

src/platform/endpoint/node/chatEndpoint.ts

@@ -223,6 +223,10 @@ export class ChatEndpoint implements IChatEndpoint {
 		return !!(enableResponsesApi && this._modelMetadata.supported_endpoints?.includes(ModelSupportedEndpoint.Responses));
 	}
 
+	public get degradationReason(): string | undefined {
+		return this._modelMetadata.warning_message;
+	}
+
 	public get policy(): 'enabled' | { terms: string } {
 		if (!this._policyDetails) {
 			return 'enabled';

src/platform/networking/common/networking.ts

@@ -164,6 +164,7 @@ export interface IChatEndpoint extends IEndpoint {
 	readonly supportedEditTools?: readonly EndpointEditToolName[];
 	readonly showInModelPicker: boolean;
 	readonly isPremium?: boolean;
+	readonly degradationReason?: string;
 	readonly multiplier?: number;
 	readonly restrictedToSkus?: string[];
 	readonly isDefault: boolean;