From a320e1230dde778cc68962948d15df11cbf6bbbd Mon Sep 17 00:00:00 2001 From: Sandeep Somavarapu Date: Mon, 19 Jan 2026 20:23:53 +0100 Subject: [PATCH] enable lm management editor for business and enterprise (#288966) * enable lm management editor for business and enterprise * show for internal --- .../chatManagement/chatManagement.contribution.ts | 2 ++ .../chat/browser/chatManagement/chatModelsWidget.ts | 9 +++++++-- 2 files changed, 9 insertions(+), 2 deletions(-) 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}`,