Fixed previous commit for Active Record < 7.2

ankane · Oct 7, 2024 · 8a8c95a · 8a8c95a
1 parent 29684d8
commit 8a8c95a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/neighbor/model.rb b/lib/neighbor/model.rb
@@ -88,7 +88,7 @@ def self.neighbor_attributes
 
           distance = distance.to_s
 
-          quoted_attribute = with_connection { |c| "#{c.quote_table_name(table_name)}.#{c.quote_column_name(attribute_name)}" }
+          quoted_attribute = connection_pool.with_connection { |c| "#{c.quote_table_name(table_name)}.#{c.quote_column_name(attribute_name)}" }
 
           column_info = columns_hash[attribute_name.to_s]
           column_type = column_info&.type
@@ -113,7 +113,7 @@ def self.neighbor_attributes
           Neighbor::Utils.validate(vector, dimensions: dimensions, type: type || Utils.type(adapter, column_info&.type), adapter: adapter)
           vector = Neighbor::Utils.normalize(vector, column_info: column_info) if normalize
 
-          query = with_connection { |c| c.quote(column_attribute.serialize(vector)) }
+          query = connection_pool.with_connection { |c| c.quote(column_attribute.serialize(vector)) }
 
           if !precision.nil?
             if adapter != :postgresql || column_type != :vector
@@ -124,7 +124,7 @@ def self.neighbor_attributes
             when "half"
               cast_dimensions = dimensions || column_info&.limit
               raise ArgumentError, "Unknown dimensions" unless cast_dimensions
-              quoted_attribute += "::halfvec(#{with_connection { |c| c.quote(cast_dimensions.to_i) }})"
+              quoted_attribute += "::halfvec(#{connection_pool.with_connection { |c| c.quote(cast_dimensions.to_i) }})"
             else
               raise ArgumentError, "Invalid precision"
             end