gitlab-cloud-connector 1.7.0 → 1.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f2561683322a95d6bc39cd443dd05a9da52d3e68af94c618720944a8f6be3879
-  data.tar.gz: ab54fe6abd39b2b287acb07097110dc6c9a86abbac8e5a78680bac0d48cf27d6
+  metadata.gz: 590617586593b273bb29603802241e367e8121a0cedf14c046932f005642a78b
+  data.tar.gz: 82bc2a0cce424a91902fe13d7ff5fe58bae4a92d67d05c9c4abaf7be0a659995
 SHA512:
-  metadata.gz: d71e3bfabb8607f2bfdfc0afcca62101a80800ed4659b60912a448b80a788849f2b5f55f279628f5de163a6070590fd9d009fefb0e03d48300773bb41eee4c6b
-  data.tar.gz: 50549c2ff8accfa95a8cdf25728d5e216d77eb66877e4e14dd3b272602d31971bec3795210fee85ac9024cc2051f0769d40200de6fd517c5dbbc01e25e560225
+  metadata.gz: 225ed047b89eaaabe116899476e261773bad0cdd7f36fe335e6a248ed22f07b88286bffbf9724e1e432a0eb6f111a195304e49747382b10a658f0dd44078ea1e
+  data.tar.gz: 89821a3abbe51094c4368f5add6d2fd35fe60e481492331a7a22b07d8a4353639f6b36a980dde3d24c868d385269715b82c56ffd654febaf208cf3a4f2ce84eb

data/config/add_ons/duo_core.yml

@@ -0,0 +1,2 @@
+---
+name: duo_core

data/config/unit_primitives/categorize_duo_chat_question.yml

@@ -1,10 +1,12 @@
 ---
 name: categorize_duo_chat_question
-description: This should not be an unit primitive.
+description: This should not be an unit primitive. It is used to categorize user questions for the purpose of product analytics.
 cut_off_date: 2024-10-17T00:00:00+00:00
-group: group::project management
-feature_category: team_planning
+group: group::duo_chat
+feature_category: duo_chat
+documentation_url: https://docs.gitlab.com/development/ai_features/duo_chat/#product-analysis
 backend_services:
   - ai_gateway
 add_ons:
+  - duo_pro
   - duo_enterprise

data/config/unit_primitives/complete_code.yml

@@ -13,4 +13,4 @@ backend_services:
 add_ons:
   - duo_pro
   - duo_enterprise
-  - duo_nano
+  - duo_core

data/config/unit_primitives/duo_chat.yml

@@ -12,4 +12,4 @@ backend_services:
 add_ons:
   - duo_pro
   - duo_enterprise
-  - duo_nano
+  - duo_core

data/config/unit_primitives/generate_code.yml

@@ -13,4 +13,4 @@ backend_services:
 add_ons:
   - duo_pro
   - duo_enterprise
-  - duo_nano
+  - duo_core

data/lib/gitlab/cloud_connector/configuration.rb

@@ -1,18 +1,18 @@
 # frozen_string_literal: true
 
-require_relative 'data_model/yaml_data_loader'
-require_relative 'data_model/abstract_data_loader'
-
 module Gitlab
   module CloudConnector
     module Configuration
       DEFAULT_DATA_LOADER_CLASS = Gitlab::CloudConnector::DataModel::YamlDataLoader
       INVALID_LOADER_MESSAGE = "Data loader must inherit from Gitlab::CloudConnector::DataModel::AbstractDataLoader"
+      INVALID_CONFIG_MESSAGE = "Override configuration must be a Hash or respond to :call"
 
       InvalidDataLoaderError = Class.new(StandardError)
+      InvalidConfigError = Class.new(StandardError)
 
       class << self
         attr_writer :config_dir, :memoize_data
+        attr_reader :override_config
 
         def config_dir
           @config_dir ||= File.expand_path("../../../config", __dir__).freeze
@@ -27,10 +27,15 @@ module Gitlab
           @data_loader_class = klass
         end
 
+        def override_config=(value)
+          validate_override_config!(value)
+          @override_config = value
+        end
+
         def memoize_data?
           return @memoize_data unless @memoize_data.nil?
 
-          (data_loader_class == DEFAULT_DATA_LOADER_CLASS)
+          data_loader_class == DEFAULT_DATA_LOADER_CLASS
         end
 
         def configure
@@ -44,6 +49,12 @@ module Gitlab
 
           raise InvalidDataLoaderError, INVALID_LOADER_MESSAGE
         end
+
+        def validate_override_config!(value)
+          return if value.nil? || value.is_a?(Hash) || value.respond_to?(:call)
+
+          raise InvalidConfigError, INVALID_CONFIG_MESSAGE
+        end
       end
     end
   end

data/lib/gitlab/cloud_connector/data_model/overridable.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module CloudConnector
+    module DataModel
+      module Overridable
+        OverrideLoadError = Class.new(StandardError)
+
+        extend ActiveSupport::Concern
+
+        def apply_overrides(base_data, model_class)
+          overrides = fetch_relevant_overrides(base_data, model_class)
+          return base_data unless overrides
+
+          base_data.merge(overrides)
+        end
+
+        private
+
+        def fetch_relevant_overrides(base_data, model_class)
+          data = load_override_data
+          return unless data
+
+          model_name = base_data[:name]&.to_sym
+          return unless model_name
+
+          model_type = model_class.model_name.tableize.to_sym
+          data.dig(model_type, model_name)
+        end
+
+        def load_override_data
+          config = Gitlab::CloudConnector::Configuration.override_config
+
+          data = config.respond_to?(:call) ? config.call : config
+          unless data.is_a?(Hash) || data.nil?
+            raise OverrideLoadError, "Invalid override config type: #{data.class}. Expected Hash or nil."
+          end
+
+          data
+        rescue StandardError => e
+          raise e if e.is_a?(OverrideLoadError)
+
+          raise OverrideLoadError, "Failed to load override config: #{e.class}: #{e.message}"
+        end
+      end
+    end
+  end
+end

data/lib/gitlab/cloud_connector/data_model/yaml_data_loader.rb

@@ -2,11 +2,14 @@
 
 require 'yaml'
 require_relative 'abstract_data_loader'
+require_relative 'overridable'
 
 module Gitlab
   module CloudConnector
     module DataModel
       class YamlDataLoader < AbstractDataLoader
+        include Overridable
+
         def load!
           Dir.glob(data_files_path).map { |file| load_model_from_file(file) }
         end
@@ -18,16 +21,21 @@ module Gitlab
         end
 
         def load_model_from_file(file_path)
-          data = YAML.safe_load(
-            File.read(file_path),
-            permitted_classes: [Time],
-            symbolize_names: true
-          ).deep_symbolize_keys!
+          raw_data = load_yaml_file(file_path)
+          data = apply_overrides(raw_data, model_class)
 
           model_class.new(**data)
         rescue StandardError => e
           raise "Error loading file #{file_path}: #{e.message}"
         end
+
+        def load_yaml_file(file_path)
+          YAML.safe_load(
+            File.read(file_path),
+            permitted_classes: [Time],
+            symbolize_names: true
+          ) || {}
+        end
       end
     end
   end

data/lib/gitlab/cloud_connector/version.rb

@@ -2,6 +2,6 @@
 
 module Gitlab
   module CloudConnector
-    VERSION = '1.7.0'
+    VERSION = '1.9.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gitlab-cloud-connector
 version: !ruby/object:Gem::Version
-  version: 1.7.0
+  version: 1.9.0
 platform: ruby
 authors:
 - Nikola Milojevic
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-04-03 00:00:00.000000000 Z
+date: 2025-04-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -96,8 +96,8 @@ files:
 - README.md
 - Rakefile
 - config/add_ons/duo_amazon_q.yml
+- config/add_ons/duo_core.yml
 - config/add_ons/duo_enterprise.yml
-- config/add_ons/duo_nano.yml
 - config/add_ons/duo_pro.yml
 - config/backend_services/ai_gateway.yml
 - config/backend_services/ai_gateway_agent.yml
@@ -169,6 +169,7 @@ files:
 - lib/gitlab/cloud_connector/data_model/abstract_data_loader.rb
 - lib/gitlab/cloud_connector/data_model/associations.rb
 - lib/gitlab/cloud_connector/data_model/base.rb
+- lib/gitlab/cloud_connector/data_model/overridable.rb
 - lib/gitlab/cloud_connector/data_model/yaml_data_loader.rb
 - lib/gitlab/cloud_connector/json_web_token.rb
 - lib/gitlab/cloud_connector/version.rb

data/config/add_ons/duo_nano.yml

@@ -1,2 +0,0 @@
----
-name: duo_nano