rails_ai_kit 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d497000abdc4e0a8f02f6167570d5d00888979e31ed5e484fc707647ebdface5
-  data.tar.gz: b721aeaeddabe1c8dce81f6aca4ff94188623c6cdb8fd5b22dd5199a7d1fb3bf
+  metadata.gz: e2037f291c26e83110faa8bf175c1d41e22c504019185b49cedad10a709f61db
+  data.tar.gz: 39a510a0b7788da34a530e24a89ad832c1402fb29b101813a76f30fd290aa05a
 SHA512:
-  metadata.gz: 0f6a1a9ec8f3f9a4d0404f2ef5b7b8dc61ad41bee7a7557d3701a7e958786114430ca4b5c3db4e1581b98bda4b280856eca1a1cd7152987c6d5db54ebc4b1b63
-  data.tar.gz: 43316a2c58575ae72429deb036cd3499b42bd19ec5c7fadf15404b46a864fdcffa3b41a33b24815fef16dd426ace74bac38fe8460e6e722c9639785ee5c4d7ef
+  metadata.gz: ede0b9f2d61943ee974e9b89c5b40d492e2e81ed3b31a5a04031c7fb32906cc746cf6720b930d404eb293a8814caa72ecfc794b5beb8dc2b72567beadf36d8f1
+  data.tar.gz: 95db7112cee6a9922fc93b5fe83b053588a4b46ea7d4d3fb9166e0192873653ef8a20b134b195bb05dd20ca07942e351a9170b1581dcbf45b08af138b568ebc5

data/CHANGELOG.md

@@ -2,6 +2,19 @@
 
 ## [Unreleased]
 
+## [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
+
+- Define `RailsAiKit::Error` before loading embedding providers to fix `uninitialized constant RailsAiKit::Error` when running generators or migrations
+
 ## [0.1.0] - 2025-03-07
 
 ### Added

data/README.md

@@ -2,6 +2,8 @@
 
 A Rails gem that adds a **classification layer on top of [pgvector](https://github.com/pgvector/pgvector)**. Instead of building custom ML or calling LLMs every time, use vector similarity to classify data: support tickets, content moderation, ecommerce categories, document routing, and more.
 
+**Source:** [github.com/imrrohitt/rails_ai_kit](https://github.com/imrrohitt/rails_ai_kit)
+
 - **No ML training** – train labels with example texts and compare embeddings
 - **No LLM cost** – one-time embedding per piece of content; classification is nearest-neighbor in PostgreSQL
 - **Rails-friendly** – `vector_classify`, `Classifier.train`, `Classifier.classify`, `Article.similar_to("query")`
@@ -247,15 +249,6 @@ MIT.
 
 ## How it’s built
 
-- **Configuration** (`lib/rails_ai_kit/configuration.rb`) – Embeding provider, dimensions, and API keys (e.g. `api_keys[:openai]`).
-- **Embedding providers** (`lib/rails_ai_kit/embedding_providers/`) – Base class plus OpenAI and Cohere. Each implements `embed(text)` and `embed_batch(texts)` using the provider API.
-- **EmbeddingService** – Wraps the configured provider and API key so `RailsAiKit.embedding.embed(text)` works without passing keys every time.
-- **LabelRecord** – ActiveRecord model for `rails_ai_kit_labels` (classifier_name, label_name, embedding). Uses Neighbor’s `has_neighbors :embedding` for similarity.
-- **Classifier** – Trains labels by averaging example embeddings and storing them; classifies by nearest-neighbor (cosine) against those label vectors. Supports `classify(text)`, `classify_by_embedding(vector)`, and `batch_classify(records)`.
-- **VectorClassify** – Concern that adds the `vector_classify` macro: `has_neighbors` on the embedding column, a before_save that embeds the source column and runs `classify_by_embedding`, and a `similar_to(query_text)` scope that embeds the query and runs nearest-neighbor search.
-
-## How it’s built
-
 - **Configuration** (`lib/rails_ai_kit/configuration.rb`) – Embedding provider, dimensions, and API keys (e.g. `api_keys[:openai]`).
 - **Embedding providers** (`lib/rails_ai_kit/embedding_providers/`) – Base class plus OpenAI and Cohere. Each implements `embed(text)` and `embed_batch(texts)` using the provider API.
 - **EmbeddingService** – Wraps the configured provider and API key so `RailsAiKit.embedding.embed(text)` works without passing keys every time.
@@ -268,4 +261,3 @@ MIT.
 
 - [pgvector](https://github.com/pgvector/pgvector) – Open-source vector similarity search for Postgres
 - [Neighbor](https://github.com/ankane/neighbor) – Nearest neighbor search for Rails (used by this gem)
-test

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.0"
+  VERSION = "0.1.2"
 end

data/lib/rails_ai_kit.rb

@@ -2,24 +2,28 @@
 
 require_relative "rails_ai_kit/version"
 require_relative "rails_ai_kit/configuration"
+
+module RailsAiKit
+  class Error < StandardError; end
+end
+
 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"
 
-module RailsAiKit
-  class Error < StandardError; end
+# Load models and AR integration only after ActiveRecord (and Neighbor) are ready.
+# Neighbor 0.6+ adds has_neighbors via ActiveSupport.on_load(:active_record).
+ActiveSupport.on_load(:active_record) do
+  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").
   def self.classifier(classifier_name = nil)
     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,13 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rails_ai_kit
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Rails AI Kit Contributors
+autorequire:
 bindir: bin
 cert_chain: []
-date: 1980-01-02 00:00:00.000000000 Z
+date: 2026-03-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -91,12 +92,13 @@ files:
 - lib/rails_ai_kit/label_record.rb
 - lib/rails_ai_kit/vector_classify.rb
 - lib/rails_ai_kit/version.rb
-homepage: https://github.com/your-org/rails_ai_kit
+homepage: https://github.com/imrrohitt/rails_ai_kit
 licenses: []
 metadata:
-  homepage_uri: https://github.com/your-org/rails_ai_kit
-  source_code_uri: https://github.com/your-org/rails_ai_kit
-  changelog_uri: https://github.com/your-org/rails_ai_kit/blob/main/CHANGELOG.md
+  homepage_uri: https://github.com/imrrohitt/rails_ai_kit
+  source_code_uri: https://github.com/imrrohitt/rails_ai_kit
+  changelog_uri: https://github.com/imrrohitt/rails_ai_kit/blob/main/CHANGELOG.md
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -111,7 +113,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.7
+rubygems_version: 3.0.3.1
+signing_key:
 specification_version: 4
 summary: Vector-based classification layer on top of pgvector for Rails
 test_files: []