rails_ai_kit 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 01b1f21a6cc3de948b270cbd23c95911803accd689598315f655cc320fcdcc4f
-  data.tar.gz: ec76e4cedde5863f5022bd9842e0360ecbe0e51d39d8287c83c5b83415376023
+  metadata.gz: 5f5b6bc37b278c3ccd429b8b37a24256b89675b56b2d0ffad092d55a6b2a93ba
+  data.tar.gz: f876cc97ad7180007ad9e57481474232c3695e4538e59d820b94b365cc52def9
 SHA512:
-  metadata.gz: 3bc4d41ca7050bc75b1f92276d1826ae43020d10b48bcd07d0fad1838c4f6dadba8f0dd262fe6a3188d2c18514a92f2dd35540ec40a99583d304fff45d585803
-  data.tar.gz: 1ef19893af1c1a7bf6ea5be4d15da6243cb00629ff79f1bc3c6254f2bebcef1d957a7c1f9a81be5eac3cb16a2b90a2f0894c9158df0922a0fc154760d414d987
+  metadata.gz: fddc5b46a839156459abc6ba82419329d66aeca703b76151663a9b66296e0d9867467422ce71dd067f998e99c22639eb8fabc558e5ea07f34c4e697faf8694ca
+  data.tar.gz: 7a1420f049a63341d5783b20e08c7948ac4347c51a65ebfe71c6691f5f88ae9507174b16ccf1e42648c9ebd2dcbfc37aa7235ef378a99a63da525977bae016be

data/CHANGELOG.md

@@ -2,6 +2,21 @@
 
 ## [Unreleased]
 
+## [0.1.3] - 2026-03-07
+
+### Fixed
+
+- Ensure Neighbor gem is required in `on_load(:active_record)` when app loads rails_ai_kit before neighbor (Gemfile order)
+- Install generator: use `ActiveRecord::Generators::Migration` and correct `source_root` (`install/templates`) so migration and initializer are found
+- Install generator: rename task to `create_labels_migration` to avoid shadowing Migration’s `create_migration`
+
+## [0.1.2] - 2026-03-07
+
+### Fixed
+
+- Compatibility with Neighbor 0.6: load label_record, classifier, and vector_classify in `ActiveSupport.on_load(:active_record)` so `has_neighbors` is available
+- Use integer for `dimensions` in LabelRecord (Neighbor 0.6 does not support proc)
+
 ## [0.1.1] - 2026-03-07
 
 ### Fixed

data/lib/generators/rails_ai_kit/install_generator.rb

@@ -1,28 +1,25 @@
 # frozen_string_literal: true
 
 require "rails/generators"
+require "rails/generators/active_record/migration"
 
 module RailsAiKit
   module Generators
     class InstallGenerator < Rails::Generators::Base
-      source_root File.expand_path("templates", __dir__)
+      include ActiveRecord::Generators::Migration
+
+      source_root File.expand_path("install/templates", __dir__)
 
       desc "Creates a migration to enable pgvector and add rails_ai_kit_labels table for label embeddings"
 
-      def create_migration
+      def create_labels_migration
         migration_template "create_rails_ai_kit_labels.rb",
-                          "db/migrate/#{migration_timestamp}_create_rails_ai_kit_labels.rb"
+                          "db/migrate/create_rails_ai_kit_labels.rb"
       end
 
       def create_initializer
         copy_file "rails_ai_kit.rb", "config/initializers/rails_ai_kit.rb"
       end
-
-      private
-
-      def migration_timestamp
-        Time.now.utc.strftime("%Y%m%d%H%M%S")
-      end
     end
   end
 end

data/lib/rails_ai_kit/label_record.rb

@@ -7,7 +7,7 @@ module RailsAiKit
   class LabelRecord < ActiveRecord::Base
     self.table_name = "rails_ai_kit_labels"
 
-    has_neighbors :embedding, dimensions: -> { RailsAiKit.configuration.embedding_dimensions }
+    has_neighbors :embedding, dimensions: RailsAiKit.configuration.embedding_dimensions
 
     validates :classifier_name, presence: true
     validates :label_name, presence: true

data/lib/rails_ai_kit/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RailsAiKit
-  VERSION = "0.1.1"
+  VERSION = "0.1.3"
 end

data/lib/rails_ai_kit.rb

@@ -11,9 +11,25 @@ require_relative "rails_ai_kit/embedding_providers/base"
 require_relative "rails_ai_kit/embedding_providers/openai"
 require_relative "rails_ai_kit/embedding_providers/cohere"
 require_relative "rails_ai_kit/embedding_service"
-require_relative "rails_ai_kit/label_record"
-require_relative "rails_ai_kit/classifier"
-require_relative "rails_ai_kit/vector_classify"
+
+# Load models and AR integration only after ActiveRecord is ready.
+# Neighbor 0.6+ adds has_neighbors in its own on_load(:active_record).
+# Gemfile order can make our hook run before Neighbor's (or before neighbor is required).
+# So we ensure the neighbor gem is loaded and has_neighbors is on Base before our models.
+ActiveSupport.on_load(:active_record) do
+  unless ActiveRecord::Base.respond_to?(:has_neighbors)
+    require "neighbor" # load gem so Neighbor and its lib are available
+    require "neighbor/attribute"
+    require "neighbor/model"
+    require "neighbor/normalized_attribute"
+    ActiveRecord::Base.extend(Neighbor::Model)
+  end
+
+  require_relative "rails_ai_kit/label_record"
+  require_relative "rails_ai_kit/classifier"
+  require_relative "rails_ai_kit/vector_classify"
+  ActiveRecord::Base.include RailsAiKit::VectorClassify
+end
 
 module RailsAiKit
   # Top-level classifier using default classifier name. For custom name use Classifier.new(classifier_name: "MyClassifier").
@@ -21,8 +37,3 @@ module RailsAiKit
     Classifier.new(classifier_name: classifier_name)
   end
 end
-
-# Optional Rails integration: extend ActiveRecord so vector_classify is available.
-if defined?(ActiveRecord::Base)
-  ActiveRecord::Base.include RailsAiKit::VectorClassify
-end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails_ai_kit
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Rails AI Kit Contributors