Added NormalizedAttribute

lib/neighbor.rb

@@ -48,6 +48,7 @@ def register_vector_type(m)
 ActiveSupport.on_load(:active_record) do
   require_relative "neighbor/attribute"
   require_relative "neighbor/model"
+  require_relative "neighbor/normalized_attribute"
   require_relative "neighbor/type/cube"
   require_relative "neighbor/type/halfvec"
   require_relative "neighbor/type/mysql_vector"

lib/neighbor/model.rb

@@ -47,7 +47,7 @@ def self.neighbor_attributes
           else
             attribute_names.each do |attribute_name|
               attribute attribute_name do |cast_type|
-                raise "todo"
+                Neighbor::NormalizedAttribute.new(cast_type: cast_type, model: self, attribute_name: attribute_name)
               end
             end
           end

lib/neighbor/normalized_attribute.rb

@@ -0,0 +1,21 @@
+module Neighbor
+  class NormalizedAttribute < ActiveRecord::Type::Value
+    delegate :type, :serialize, :deserialize, to: :@cast_type
+
+    def initialize(cast_type:, model:, attribute_name:)
+      @cast_type = cast_type
+      @model = model
+      @attribute_name = attribute_name.to_s
+    end
+
+    def cast(value)
+      Neighbor::Utils.normalize(@cast_type.cast(value), column_info: @model.columns_hash[@attribute_name])
+    end
+
+    private
+
+    def cast_value(...)
+      @cast_type.send(:cast_value, ...)
+    end
+  end
+end

test/cube_test.rb

@@ -113,11 +113,19 @@ def test_normalize
 
   def test_insert
     CosineItem.insert!({cube_embedding: [0, 3, 4]})
-    assert_elements_in_delta [0, 0.6, 0.8], Item.last.cube_embedding
+    if supports_normalizes?
+      assert_elements_in_delta [0, 0.6, 0.8], Item.last.cube_embedding
+    else
+      assert_elements_in_delta [0, 3, 4], Item.last.cube_embedding
+    end
   end
 
   def test_insert_all
     CosineItem.insert_all!([{cube_embedding: [0, 3, 4]}])
-    assert_elements_in_delta [0, 0.6, 0.8], Item.last.cube_embedding
+    if supports_normalizes?
+      assert_elements_in_delta [0, 0.6, 0.8], Item.last.cube_embedding
+    else
+      assert_elements_in_delta [0, 3, 4], Item.last.cube_embedding
+    end
   end
 end

test/mariadb_test.rb

@@ -85,11 +85,19 @@ def test_normalize
 
   def test_insert
     MariadbCosineItem.insert!({embedding: [0, 3, 4]})
-    assert_elements_in_delta [0, 0.6, 0.8], MariadbItem.last.embedding
+    if supports_normalizes?
+      assert_elements_in_delta [0, 0.6, 0.8], MariadbItem.last.embedding
+    else
+      assert_elements_in_delta [0, 3, 4], MariadbItem.last.embedding
+    end
   end
 
   def test_insert_all
     MariadbCosineItem.insert_all!([{embedding: [0, 3, 4]}])
-    assert_elements_in_delta [0, 0.6, 0.8], MariadbItem.last.embedding
+    if supports_normalizes?
+      assert_elements_in_delta [0, 0.6, 0.8], MariadbItem.last.embedding
+    else
+      assert_elements_in_delta [0, 3, 4], MariadbItem.last.embedding
+    end
   end
 end

test/test_helper.rb

@@ -34,4 +34,8 @@ def assert_index_scan(relation)
       assert_match "Index Scan", relation.limit(5).explain.inspect
     end
   end
+
+  def supports_normalizes?
+    ActiveRecord::VERSION::STRING.to_f >= 7.1
+  end
 end