Displays a message in AI provider picker when org configuration is enforced

(#4300, #4338)

Author: sergeibbb

src/plus/ai/aiProviderService.ts

@@ -398,6 +398,14 @@ export class AIProviderService implements Disposable {
 		return new Map<AIProviders, AIProviderDescriptorWithConfiguration>(getSettledValues(promises));
 	}
 
+	hasProvidersDisabledByOrg(): boolean {
+		const orgAiConfig = getOrgAIConfig();
+		return (
+			orgAiConfig.enforceAiProviders &&
+			[...supportedAIProviders.keys()].some(providerId => !isProviderEnabledByOrg(providerId, orgAiConfig))
+		);
+	}
+
 	private async ensureProviderConfigured(provider: AIProviderDescriptorWithType, silent: boolean): Promise<boolean> {
 		if (provider.id === this._provider?.id) return this._provider.configured(silent);
 

src/quickpicks/aiModelPicker.ts

@@ -46,6 +46,8 @@ export async function showAIProviderPicker(
 	const subscription = getSettledValue(subscriptionResult)!;
 	const hasPaidPlan = isSubscriptionPaidPlan(subscription.plan.effective.id) && subscription.account?.verified;
 
+	const hasProvidersDisabledByOrg = container.ai.hasProvidersDisabledByOrg();
+
 	const quickpick = window.createQuickPick<ProviderQuickPickItem>();
 	quickpick.ignoreFocusOut = getQuickPickIgnoreFocusOut();
 	quickpick.title = 'Select AI Provider';
@@ -62,8 +64,17 @@ export async function showAIProviderPicker(
 				  : undefined;
 
 		let addedRequiredKeySeparator = false;
+
 		while (true) {
 			const items: ProviderQuickPickItem[] = [];
+			if (hasProvidersDisabledByOrg) {
+				items.push({
+					label: '',
+					description: '$(lock) Some Al providers have been disabled by your GitKraken Admin',
+					iconPath: new ThemeIcon('blank'),
+					provider: undefined!,
+				});
+			}
 			for (const p of providers.values()) {
 				if (!p.primary && !addedRequiredKeySeparator) {
 					addedRequiredKeySeparator = true;
@@ -96,7 +107,11 @@ export async function showAIProviderPicker(
 					quickpick.onDidHide(() => resolve(undefined)),
 					quickpick.onDidAccept(() => {
 						if (quickpick.activeItems.length !== 0) {
-							resolve(quickpick.activeItems[0]);
+							const selectedItem = quickpick.activeItems[0];
+							// Don't allow selection of the disabled providers message
+							if (selectedItem.provider !== undefined) {
+								resolve(selectedItem);
+							}
 						}
 					}),
 					quickpick.onDidTriggerItemButton(e => {