From cbd309add11138893fdbbf2479b920a80c17dd1a Mon Sep 17 00:00:00 2001
From: Julie Huang <julhuang@gitlab.com>
Date: Wed, 18 Dec 2024 08:34:20 +0000
Subject: [PATCH] Use :manage_self_hosted_models_settings policy to manage
 visibility of Self-hosted models pages in admin dashboard

---
 ee/lib/ee/sidebars/admin/panel.rb | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/ee/lib/ee/sidebars/admin/panel.rb b/ee/lib/ee/sidebars/admin/panel.rb
index 460806eea411b..b73a006640e4f 100644
--- a/ee/lib/ee/sidebars/admin/panel.rb
+++ b/ee/lib/ee/sidebars/admin/panel.rb
@@ -44,18 +44,13 @@ def insert_gilab_duo_menu
             ::Sidebars::Admin::Menus::CodeSuggestionsMenu.new(context)
           )
 
-          return unless self_hosted_models_enabled?
+          return unless Ability.allowed?(context.current_user, :manage_self_hosted_models_settings)
 
           insert_menu_after(
             ::Sidebars::Admin::Menus::CodeSuggestionsMenu,
             ::Sidebars::Admin::Menus::SelfHostedModelsMenu.new(context)
           )
         end
-
-        def self_hosted_models_enabled?
-          License.current&.ultimate? &&
-            GitlabSubscriptions::AddOnPurchase.for_duo_enterprise.active.exists?
-        end
       end
     end
   end
-- 
GitLab