gitlab-cloud-connector 1.12.0 → 1.14.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5ad05b8bdbe28bb01760e2778f4fd95949d4ab927ee76a78b0177aafea43d912
-  data.tar.gz: b2405278b9f29f0c3d696c44f2aeb0d3f06578e8c3af28189a33e3d998e5a626
+  metadata.gz: cb83b610b630d8bec2117b6c5473856274b4cf19824a9b96c729808647b8c55c
+  data.tar.gz: f95a96e59789f000725c2b9ef87fb166ce17d23082f6460125d3bde5d45f7f75
 SHA512:
-  metadata.gz: 36c31411737f12736f99cc95bda8e3eae88eec3dc13f27727b5654b2cdd4df79d40aa86abcb1c4f2a6b7f5d500f2508a64bb610cbece045fbd479f29ffbb68de
-  data.tar.gz: f0b345b64f19ffed81dd9a28ea67ea66d63759a762b353cb49762d3b947804eb940ffd60eb13559869b1169de0ee74c4fff0b8ee1aebad353f2cfad3f752841b
+  metadata.gz: c8c12b5077dbbe30f6ebb538f5a90c4ad867418200f1502088cd53a276a8469ad3778c532ccfb8336b2745abc7a0d48abbcab9b599f65b77f30bac66457289fd
+  data.tar.gz: cc05305427fa886ce5e96662dfc6b3d50a67b1180fbe75aff0b3732a29dc8a07127b222b864b2cf531db4c4c0f92785e0e6e06f95ba22d9490a7c39357194e13

data/README.md

@@ -61,14 +61,14 @@ duo_chat.backend_services.map(&:jwt_aud)
 ## Contributing
 
 ### Test your changes locally
-  
-To test changes to the Ruby gem in your environment, you need to source gem contents from you local dir:  
 
-1. add a `path` attribute to the `gem` definition in the `Gemfile:
+To test changes to the Ruby gem in your environment, you need to source gem contents from you local dir:
 
-```ruby
-gem "gitlab-cloud-connector", "~> 1.5", require: 'gitlab/cloud_connector', feature_category: :cloud_connector, path: '<path to gitlab-cloud-connector folder on your machine>/src/ruby/'
-```
+1. Set `path` in `Gemfile` to point to the library project source:
+
+   ```ruby
+   gem "gitlab-cloud-connector", "~> 1.5", require: 'gitlab/cloud_connector', feature_category: :cloud_connector, path: '/path/to/gitlab-cloud-connector/src/ruby/'
+   ```
 
 1. Run `bundle install` to apply the changes
 

data/config/services/duo_chat.yml

@@ -7,6 +7,7 @@ unit_primitives:
   - ask_epic
   - ask_issue
   - ask_merge_request
+  - codebase_search
   - documentation_search
   - duo_chat
   - explain_code

data/config/services/self_hosted_models.yml

@@ -8,6 +8,7 @@ unit_primitives:
   - ask_epic
   - ask_issue
   - ask_merge_request
+  - codebase_search
   - complete_code
   - generate_code
   - documentation_search

data/config/unit_primitives/codebase_search.yml

@@ -0,0 +1,16 @@
+---
+name: codebase_search
+description: Perform semantic search on a GitLab repository for a given query.
+cut_off_date: 2024-10-17T00:00:00+00:00
+group: group::code creation
+feature_category: duo_chat
+documentation_url: "" # To be added: https://gitlab.com/gitlab-org/gitlab/-/issues/537573
+milestone: "18.1"
+introduced_by_url: "https://gitlab.com/gitlab-org/cloud-connector/gitlab-cloud-connector/-/merge_requests/149"
+unit_primitive_issue_url: https://gitlab.com/gitlab-org/cloud-connector/gitlab-cloud-connector/-/issues/72
+backend_services:
+  - ai_gateway
+add_ons:
+  - duo_pro
+  - duo_enterprise
+  - duo_amazon_q

data/config/unit_primitives/generate_embeddings_codebase.yml

@@ -0,0 +1,17 @@
+---
+name: generate_embeddings_codebase
+description: >
+    Generates embeddings for Chat with Codebase feature.
+    This unit primitive serves two purposes:
+    1. System requests (majority): Used to populate a vector store for search functionality. These requests do not have an associated user.
+    2. User requests (minority): Generated when Chat with Codebase feature is used. These requests include user.
+group: group::code suggestions
+feature_category: code_suggestions
+unit_primitive_issue_url: https://gitlab.com/gitlab-org/gitlab/-/issues/538567
+introduced_by_url: https://gitlab.com/gitlab-org/cloud-connector/gitlab-cloud-connector/-/merge_requests/146
+cut_off_date: 2024-02-15T00:00:00+00:00
+backend_services:
+  - ai_gateway
+add_ons:
+  - duo_pro
+  - duo_enterprise

data/config/unit_primitives/semantic_search_issue.yml

@@ -1,6 +1,10 @@
 ---
 name: semantic_search_issue
-description: Perform semantic search on gitlab issues for a given query.
+description: >
+    Performs semantic search on GitLab work items for a given query.
+    This unit primitive serves two purposes:
+    1. System requests (majority): Used to populate a vector store for search functionality. These requests do not have an associated user or namespace IDs.
+    2. User requests (minority): Generated when a work item search is performed using the Global Search feature for enabled projects. These requests include user and namespace IDs.
 group: group::global search
 feature_category: global_search
 backend_services:

data/lib/gitlab/cloud_connector/configuration.rb

@@ -11,7 +11,7 @@ module Gitlab
       InvalidConfigError = Class.new(StandardError)
 
       class << self
-        attr_writer :config_dir, :memoize_data
+        attr_writer :config_dir
         attr_reader :override_config
 
         def config_dir
@@ -32,12 +32,6 @@ module Gitlab
           @override_config = value
         end
 
-        def memoize_data?
-          return @memoize_data unless @memoize_data.nil?
-
-          data_loader_class == DEFAULT_DATA_LOADER_CLASS
-        end
-
         def configure
           yield self if block_given?
         end

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

@@ -23,17 +23,13 @@ module Gitlab
           end
 
           def all
-            if Configuration.memoize_data?
-              @all ||= data_loader.load!
-            else
-              data_loader.load!
-            end
+            data_loader.load!
           end
 
           private
 
           def data_loader
-            Configuration.data_loader_class.new(self)
+            @data_loader ||= Configuration.data_loader_class.new(self)
           end
         end
 

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

@@ -11,11 +11,15 @@ module Gitlab
         include Overridable
 
         def load!
-          Dir.glob(data_files_path).map { |file| load_model_from_file(file) }
+          load_data_from_files
         end
 
         private
 
+        def load_data_from_files
+          @load_data_from_files ||= Dir.glob(data_files_path).map { |file| load_model_from_file(file) }
+        end
+
         def data_files_path
           File.join(CloudConnector::Configuration.config_dir, model_class.model_name.tableize, '*.yml')
         end

data/lib/gitlab/cloud_connector/version.rb

@@ -2,6 +2,6 @@
 
 module Gitlab
   module CloudConnector
-    VERSION = '1.12.0'
+    VERSION = '1.14.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gitlab-cloud-connector
 version: !ruby/object:Gem::Version
-  version: 1.12.0
+  version: 1.14.0
 platform: ruby
 authors:
 - Nikola Milojevic
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-05-09 00:00:00.000000000 Z
+date: 2025-05-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -128,6 +128,7 @@ files:
 - config/unit_primitives/ask_merge_request.yml
 - config/unit_primitives/categorize_duo_chat_question.yml
 - config/unit_primitives/code_suggestions.yml
+- config/unit_primitives/codebase_search.yml
 - config/unit_primitives/complete_code.yml
 - config/unit_primitives/description_composer.yml
 - config/unit_primitives/documentation_search.yml
@@ -139,6 +140,7 @@ files:
 - config/unit_primitives/generate_code.yml
 - config/unit_primitives/generate_commit_message.yml
 - config/unit_primitives/generate_cube_query.yml
+- config/unit_primitives/generate_embeddings_codebase.yml
 - config/unit_primitives/generate_issue_description.yml
 - config/unit_primitives/glab_ask_git_command.yml
 - config/unit_primitives/include_dependency_context.yml