stretchy-model 0.6.0 → 0.6.5

data/lib/{rails → stretchy/rails}/instrumentation/railtie.rb

@@ -3,17 +3,21 @@ module Stretchy
 
     class Railtie < ::Rails::Railtie
 
-      require 'rails/instrumentation/publishers'
-      Stretchy.instrument!
+      require_relative 'publishers'
 
       ActiveSupport::Notifications.subscribe 'search.stretchy' do |name, start, finish, id, payload|
         message = [
-           Rainbow("  #{payload[:klass]}").bright, 
-           "(#{(finish.to_time - start.to_time).round(2)}ms)",
-           Stretchy::Utils.to_curl(payload[:klass].constantize, payload[:search])
+            Rainbow("  #{payload[:klass]}").bright, 
+            "(#{(finish.to_time - start.to_time).round(2)}ms)",
+            Stretchy::Utils.to_curl(payload[:klass].constantize, payload[:search])
         ].join(" ")
         ::Rails.logger.debug(Rainbow(message).color(:yellow))
-     end
+      end
+
+      
+      initializer 'stretchy.instrumentation' do
+        Stretchy.instrument!
+      end
 
       # initializer 'stretchy.set_defaults' do
       #   config.elasticsearch_cache_store = :redis_store
@@ -21,5 +25,6 @@ module Stretchy
       # end
 
     end
+
   end
 end

data/lib/stretchy/relation.rb

@@ -147,6 +147,7 @@ module Stretchy
             "#<#{self.class.name} #{message.join(', ')}>"
           rescue StandardError => e
             e
+            raise e
           end
         end
 

data/lib/stretchy/relations/query_builder.rb

@@ -31,6 +31,18 @@ module Stretchy
         @query_string ||= compact_where(values[:query_string], bool: false)
       end
 
+      def neural_sparse
+        @neural_sparse ||= values[:neural_sparse]
+      end
+
+      def neural
+        @neural ||= values[:neural]
+      end
+
+      def hybrid
+        @hybrid ||= values[:hybrid]
+      end
+
       def must_nots
         @must_nots ||= compact_where(values[:must_not])
       end
@@ -104,9 +116,68 @@ module Stretchy
         query_filters.nil? && or_filters.nil?
       end
 
+      def missing_neural?
+        neural_sparse.nil? && neural.nil? && hybrid.nil?
+      end
+
+      def no_query?
+        missing_bool_query? && missing_query_string? && missing_query_filter? && missing_neural?
+      end
+
       def build_query
-        return if missing_bool_query? && missing_query_string? && missing_query_filter?
+        return if no_query?
         structure.query do
+
+          structure.hybrid do
+            structure.queries do
+                hybrid[:neural].each do |n|
+                  structure.child! do
+                    params = n.dup
+                    field_name, query_text = params.shift
+                    structure.neural do
+                      structure.set! field_name do
+                        structure.query_text query_text
+                        structure.extract! params, *params.keys
+                      end
+                    end
+                  end
+                end
+
+                hybrid[:query].each do |query|
+                  structure.child! do
+                    structure.extract! query, *query.keys
+                  end
+                end
+
+            end
+          end unless hybrid.nil?
+
+          structure.neural_sparse do
+            neural_sparse.each do |args|
+              params = args.dup
+              field_name, query_text = params.shift
+              structure.set! field_name do
+                structure.query_text query_text
+                structure.extract! params, *params.keys
+              end
+            end
+          end unless neural_sparse.blank?
+
+          structure.neural do
+            neural.each do |args|
+              params = args.dup
+              field_name, query = params.shift
+              structure.set! field_name do
+                if query.is_a?(Hash)
+                  structure.extract! query, *query.keys
+                else
+                  structure.query_text query
+                end
+                structure.extract! params, *params.keys
+              end
+            end
+          end unless neural.blank?
+
           structure.regexp do
             build_regexp unless regexes.nil?
           end
@@ -121,12 +192,11 @@ module Stretchy
 
           end unless missing_bool_query? && missing_query_filter?
 
-
-
           structure.query_string do
             structure.extract! query_string_options, *query_string_options.keys
             structure.query query_strings
           end unless query_strings.nil?
+
         end.with_indifferent_access
       end
 

data/lib/stretchy/relations/query_methods.rb

@@ -19,6 +19,9 @@ module Stretchy
         :or_filter,
         :extending,
         :skip_callbacks,
+        :neural_sparse,
+        :neural,
+        :hybrid,
         :regexp
       ]
 
@@ -231,6 +234,35 @@ module Stretchy
       # @see #where
       alias :must :where
 
+
+      concerning :Neural do
+        def neural_sparse(opts)
+          spawn.neural_sparse!(opts)
+        end
+
+        def neural_sparse!(opts) # :nodoc:
+          self.neural_sparse_values += [opts]
+          self
+        end
+
+        def neural(opts)
+          spawn.neural!(opts)
+        end
+
+        def neural!(opts) # :nodoc:
+          self.neural_values += [opts]
+          self
+        end
+
+        def hybrid(opts)
+          spawn.hybrid!(opts)
+        end
+
+        def hybrid!(opts) # :nodoc:
+          self.hybrid_values += [opts]
+          self
+        end
+      end
       # Adds a regexp condition to the query.
       # 
       # @param field [Hash] the field to filter by and the Regexp to match

data/lib/stretchy/shared_scopes.rb

@@ -4,7 +4,12 @@ module Stretchy
 
         included do
 
-            scope :between, ->(range, range_field = "created_at") { filter_query(:range, range_field => {gte: range.begin, lte: range.end}) }
+            scope :between, ->(range, range_field = "created_at") do
+                range_options = {gte: range.begin}
+                upper_bound = range.exclude_end? ? :lt : :lte
+                range_options[upper_bound] = range.end
+                filter_query(:range, range_field => range_options)
+            end
             scope :using_time_based_indices, lambda { |range| search_options(index: time_based_indices(range)) }
 
         end

data/lib/stretchy/version.rb

@@ -1,3 +1,3 @@
 module Stretchy
-  VERSION = '0.6.0'
+  VERSION = '0.6.5'
 end

data/lib/stretchy.rb

@@ -9,8 +9,8 @@ require 'active_model'
 require 'active_support/all'
 
 require_relative "stretchy/version"
-require_relative "rails/instrumentation/railtie" if defined?(Rails)
-
+require_relative "stretchy/rails/instrumentation/railtie" if defined?(Rails)
+require_relative 'elasticsearch/api/namespace/machine_learning/model'
 
 module Stretchy
 
@@ -27,6 +27,7 @@ module Stretchy
 
   def self.boot!
     loader.setup
+    Elasticsearch::API.send(:include, Elasticsearch::API::MachineLearning::Models)
     Stretchy::Attributes.register!
   end
 
@@ -53,6 +54,10 @@ module Stretchy
         self.opensearch = true if @client.class.name =~ /OpenSearch/
     end
 
+    def search_backend_const
+        @opensearch ? OpenSearch : Elasticsearch
+    end
+
     def opensearch=(bool)
         @opensearch = bool
         OpenSearchCompatibility.opensearch_patch! if bool

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stretchy-model
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.6.5
 platform: ruby
 authors:
 - Spencer Markowski
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-03-15 00:00:00.000000000 Z
+date: 2024-03-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: zeitwerk
@@ -206,6 +206,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.5.0
+- !ruby/object:Gem::Dependency
+  name: faker
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.18'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.18'
 description: Provides a familiar ActiveRecord-like interface for working with Elasticsearch
 email:
 - spencer.markowski@theablefew.com
@@ -224,8 +238,24 @@ files:
 - containers/Dockerfile.elasticsearch
 - containers/Dockerfile.opensearch
 - docker-compose.yml
-- lib/rails/instrumentation/publishers.rb
-- lib/rails/instrumentation/railtie.rb
+- lib/elasticsearch/api/actions/machine_learning/models/delete_model.rb
+- lib/elasticsearch/api/actions/machine_learning/models/deploy.rb
+- lib/elasticsearch/api/actions/machine_learning/models/get_model.rb
+- lib/elasticsearch/api/actions/machine_learning/models/get_status.rb
+- lib/elasticsearch/api/actions/machine_learning/models/params_registry.rb
+- lib/elasticsearch/api/actions/machine_learning/models/register.rb
+- lib/elasticsearch/api/actions/machine_learning/models/undeploy.rb
+- lib/elasticsearch/api/actions/machine_learning/models/update_model.rb
+- lib/elasticsearch/api/namespace/machine_learning/model.rb
+- lib/opensearch/api/actions/machine_learning/models/delete_model.rb
+- lib/opensearch/api/actions/machine_learning/models/deploy.rb
+- lib/opensearch/api/actions/machine_learning/models/get_model.rb
+- lib/opensearch/api/actions/machine_learning/models/get_status.rb
+- lib/opensearch/api/actions/machine_learning/models/params_registry.rb
+- lib/opensearch/api/actions/machine_learning/models/register.rb
+- lib/opensearch/api/actions/machine_learning/models/undeploy.rb
+- lib/opensearch/api/actions/machine_learning/models/update_model.rb
+- lib/opensearch/api/namespace/machine_learning/model.rb
 - lib/stretchy.rb
 - lib/stretchy/associations.rb
 - lib/stretchy/associations/associated_validator.rb
@@ -283,12 +313,17 @@ files:
 - lib/stretchy/delegation/delegate_cache.rb
 - lib/stretchy/delegation/gateway_delegation.rb
 - lib/stretchy/indexing/bulk.rb
+- lib/stretchy/machine_learning/model.rb
 - lib/stretchy/model/callbacks.rb
 - lib/stretchy/model/serialization.rb
 - lib/stretchy/null_relation.rb
 - lib/stretchy/open_search_compatibility.rb
 - lib/stretchy/persistence.rb
+- lib/stretchy/pipeline.rb
+- lib/stretchy/pipelines/processor.rb
 - lib/stretchy/querying.rb
+- lib/stretchy/rails/instrumentation/publishers.rb
+- lib/stretchy/rails/instrumentation/railtie.rb
 - lib/stretchy/record.rb
 - lib/stretchy/refreshable.rb
 - lib/stretchy/relation.rb
@@ -334,7 +369,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.3
+rubygems_version: 3.3.7
 signing_key:
 specification_version: 4
 summary: Rails ORM for Elasticsearch

data/lib/rails/instrumentation/publishers.rb

@@ -1,29 +0,0 @@
-module Rails
-    module Instrumentation
-        module Publishers
-
-            module Record 
-
-                extend ActiveSupport::Concern
-
-                included do
-                    unless method_defined?(:search_without_instrumentation!)
-                        alias_method :search_without_instrumentation!, :search
-                        alias_method :search, :search_with_instrumentation!
-                    end
-                end
-
-                def search_with_instrumentation!(query_or_definition, options={})
-                    ActiveSupport::Notifications.instrument "search.stretchy", 
-                      name: "Search", 
-                      klass: self.base_class.to_s,
-                      search: {index: self.index_name, body: query_or_definition }.merge(options) do
-                        search_without_instrumentation!(query_or_definition, options)
-                    end
-                end
-
-            end
-
-        end
-    end
-end