1 files changed, 15 insertions, 7 deletions
diff --git a/lib/tasks/gitlab/tw/codeowners.rake b/lib/tasks/gitlab/tw/codeowners.rake
index afe2c564247..cea66125fd0 100644
--- a/lib/tasks/gitlab/tw/codeowners.rake
+++ b/lib/tasks/gitlab/tw/codeowners.rake
@@ -21,7 +21,8 @@ namespace :tw do
     CODE_OWNER_RULES = [
       # CodeOwnerRule.new('Activation', ''),
       # CodeOwnerRule.new('Acquisition', ''),
-      # CodeOwnerRule.new('AI Assisted', ''),
+      CodeOwnerRule.new('AI Framework', '@sselhorn'),
+      CodeOwnerRule.new('AI Model Validation', '@sselhorn'),
       CodeOwnerRule.new('Analytics Instrumentation', '@lciutacu'),
       CodeOwnerRule.new('Anti-Abuse', '@phillipwells'),
       CodeOwnerRule.new('Application Performance', '@jglassman1'),
@@ -34,13 +35,14 @@ namespace :tw do
       CodeOwnerRule.new('Container Registry', '@marcel.amirault'),
       CodeOwnerRule.new('Contributor Experience', '@eread'),
       CodeOwnerRule.new('Database', '@aqualls'),
-      # CodeOwnerRule.new('DataOps', ''),
+      CodeOwnerRule.new('DataOps', '@sselhorn'),
       # CodeOwnerRule.new('Delivery', ''),
       CodeOwnerRule.new('Development', '@sselhorn'),
       CodeOwnerRule.new('Distribution', '@axil'),
       CodeOwnerRule.new('Distribution (Charts)', '@axil'),
       CodeOwnerRule.new('Distribution (Omnibus)', '@eread'),
       CodeOwnerRule.new('Documentation Guidelines', '@sselhorn'),
+      CodeOwnerRule.new('Duo Chat', '@sselhorn'),
       CodeOwnerRule.new('Dynamic Analysis', '@rdickenson'),
       CodeOwnerRule.new('IDE', '@ashrafkhamis'),
       CodeOwnerRule.new('Foundations', '@sselhorn'),
@@ -53,7 +55,7 @@ namespace :tw do
       CodeOwnerRule.new('Import and Integrate', '@eread @ashrafkhamis'),
       CodeOwnerRule.new('Infrastructure', '@sselhorn'),
       # CodeOwnerRule.new('Knowledge', ''),
-      # CodeOwnerRule.new('MLOps', '')
+      CodeOwnerRule.new('MLOps', '@sselhorn'),
       # CodeOwnerRule.new('Observability', ''),
       CodeOwnerRule.new('Optimize', '@lciutacu'),
       CodeOwnerRule.new('Organization', '@lciutacu'),
@@ -71,7 +73,7 @@ namespace :tw do
       CodeOwnerRule.new('Runner', '@fneill'),
       CodeOwnerRule.new('Runner SaaS', '@fneill'),
       CodeOwnerRule.new('Security Policies', '@rdickenson'),
-      CodeOwnerRule.new('Source Code', '@aqualls @msedlakjakubowski'),
+      CodeOwnerRule.new('Source Code', ->(path) { path.start_with?('/doc/user') ? '@aqualls' : '@msedlakjakubowski' }),
       CodeOwnerRule.new('Static Analysis', '@rdickenson'),
       CodeOwnerRule.new('Style Guide', '@sselhorn'),
       CodeOwnerRule.new('Tenant Scale', '@lciutacu'),
@@ -100,8 +102,14 @@ namespace :tw do
       end
     end
 
-    def self.writer_for_group(category)
-      CODE_OWNER_RULES.find { |rule| rule.category == category }&.writer
+    def self.writer_for_group(category, path)
+      writer = CODE_OWNER_RULES.find { |rule| rule.category == category }&.writer
+
+      if writer.is_a?(String) || writer.nil?
+        writer
+      else
+        writer.call(path)
+      end
     end
 
     errors = []
@@ -118,7 +126,7 @@ namespace :tw do
         next
       end
 
-      writer = writer_for_group(document.group)
+      writer = writer_for_group(document.group, relative_file)
       next unless writer
 
       mappings << DocumentOwnerMapping.new(relative_file, writer) if document.has_a_valid_group?