diff --git a/src/vs/workbench/contrib/chat/browser/chatManagement/chatManagement.contribution.ts b/src/vs/workbench/contrib/chat/browser/chatManagement/chatManagement.contribution.ts
index 9258a8f280c2..df2a0d58e1e6 100644
--- a/src/vs/workbench/contrib/chat/browser/chatManagement/chatManagement.contribution.ts
+++ b/src/vs/workbench/contrib/chat/browser/chatManagement/chatManagement.contribution.ts
@@ -110,6 +110,8 @@ registerAction2(class extends Action2 {
 				ChatContextKeys.Entitlement.planFree,
 				ChatContextKeys.Entitlement.planPro,
 				ChatContextKeys.Entitlement.planProPlus,
+				ChatContextKeys.Entitlement.planBusiness,
+				ChatContextKeys.Entitlement.planEnterprise,
 				ChatContextKeys.Entitlement.internal
 			)),
 			f1: true,
diff --git a/src/vs/workbench/contrib/chat/browser/chatManagement/chatModelsWidget.ts b/src/vs/workbench/contrib/chat/browser/chatManagement/chatModelsWidget.ts
index c7a62628b104..bd8ffdb7f7bc 100644
--- a/src/vs/workbench/contrib/chat/browser/chatManagement/chatModelsWidget.ts
+++ b/src/vs/workbench/contrib/chat/browser/chatManagement/chatModelsWidget.ts
@@ -1170,8 +1170,13 @@ export class ChatModelsWidget extends Disposable {
 
 			const configurableVendors = this.languageModelsService.getVendors().filter(vendor => vendor.managementCommand || vendor.configuration);
 
-			const hasPlan = this.chatEntitlementService.entitlement !== ChatEntitlement.Unknown && this.chatEntitlementService.entitlement !== ChatEntitlement.Available;
-			this.addButton.enabled = hasPlan && configurableVendors.length > 0;
+			const entitlement = this.chatEntitlementService.entitlement;
+			const supportsAddingModels = this.chatEntitlementService.isInternal
+				|| (entitlement !== ChatEntitlement.Unknown
+					&& entitlement !== ChatEntitlement.Available
+					&& entitlement !== ChatEntitlement.Business
+					&& entitlement !== ChatEntitlement.Enterprise);
+			this.addButton.enabled = supportsAddingModels && configurableVendors.length > 0;
 
 			this.dropdownActions = configurableVendors.map(vendor => toAction({
 				id: `enable-${vendor.vendor}`,