chromable 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5695e34925255a187cf2cdca8337c8cb408e688b905984f28af08f43a12b8ffc
-  data.tar.gz: 5cbb179745647c0fd4fb982a4890570f6dbdf7e9d269f3af6b468949829cf83b
+  metadata.gz: 5b0b02fff83da824d9195113352c1d68dfaaa06bde568eba6448fb4cb79e0919
+  data.tar.gz: 879ac8e61ba6be8a9490174f13517c509409c0d271fd93e9fd06d6a22cf3f1fe
 SHA512:
-  metadata.gz: e281b9fd661de160e7c4c27a0333f7db04b57971a87cba74b8aaa4ff2eb1d5e12f4cb7683692b6ad26e8f048cb2e78ace7113e16f07c420fb3962e2259f3b8db
-  data.tar.gz: 73531acad9fecd1eea5da410e3191994bea0a7ff149da71f1f630b0c3770b8594af900c4efd86d64f0db02be2d8193d07d8c31f60790ae92233b578fc2595933
+  metadata.gz: a5a97a3c29f46ee311eafa1d2ea9d51d032568a2bafa64877d22590403152b623318a93d650328af3db1db005e5315b4a90002185a9bfa0c328fcb81dd374276
+  data.tar.gz: cc14d192452c5e4f72ee090f5423c7b78afe1449e1cd1b25009fcb7d4db29394fe50a2093e107c438f9d4b0d60a46ff8f3230f4a6fd9a1aab5f5b33cdc99798e

data/README.md

@@ -4,11 +4,11 @@ Ruby on Rails integration for ChromaDB based on `chroma-db` gem.
 
 ## Installation
 
-Install the gem and add to the application's Gemfile by executing:
+Install `chromable` and add it to the application's Gemfile by executing:
 
     $ bundle add chromable
 
-If bundler is not being used to manage dependencies, install the gem by executing:
+Or, if bundler is not being used to manage dependencies, install `chromable` by executing:
 
     $ gem install chromable
 
@@ -24,19 +24,31 @@ Chroma.logger = Logger.new($stdout)
 Chroma.log_level = Chroma::LEVEL_ERROR
 ```
 
-Then, add the following line to your Rails model:
+Then, include `Chromable` module in your model and initialize it:
 
 ```ruby
-chromable document: :content, metadata: %i[author category], embedder: :embed
+class Post < ApplicationRecord
+  include Chromable
+
+  chromable document: :content, metadata: %i[author category], embedder: :embed
+
+  def embed
+    # Call OpenAI now please :)
+  end
+end
 ```
 
 Where:
 - `document:` is a callable represents the text content you want to embed and store in ChromaDB.
-- `metadata:` is the list of attributes to be passed to ChromaDB as metadata.
-- `embedder:` is the method that returns the embedding representation for the current instance.
+- `metadata:` is the list of attributes to be passed to ChromaDB as metadata to be used to filter.
+- `embedder:` is a callable returns the embedding representation for the current instance.
 
 Optionaly you can pass `collection_name:`. If not passed, the plural form of the model name will be used.
 
+All `chromable` method arguments are optional.
+
+At this point, `chromable` will create, update, and destroy the ChromaDB embeddings for your objects based on Rails callbacks.
+
 To interact with the ChromaDB collection, `chromable` provides `Model.collection` method to retrieve the collection instance.
 Also, `chromable` provides the following methods for each model instance:
 

data/lib/chromable/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Chromable
-  VERSION = '0.1.2'
+  VERSION = '0.1.4'
 end

data/lib/chromable.rb

@@ -19,7 +19,7 @@ module Chromable
   end
 
   class_methods do
-    def chromable(document:, metadata:, embedder:, collection_name: nil)
+    def chromable(collection_name: nil, document: nil, metadata: nil, embedder: nil)
       self.collection_name = (collection_name.presence || name.underscore.pluralize)
       self.document = document
       self.metadata = metadata
@@ -41,9 +41,9 @@ module Chromable
     self.class.chroma_collection.upsert(
       Chroma::Resources::Embedding.new(
         id: id,
-        document: send(self.class.document),
-        embedding: send(self.class.embedder),
-        metadata: self.class.metadata.index_with { |attribute| send(attribute) }
+        document: self.class.document ? send(self.class.document) : nil,
+        embedding: self.class.embedder ? send(self.class.embedder) : nil,
+        metadata: self.class.metadata ? self.class.metadata.index_with { |attribute| send(attribute) } : nil
       )
     )
   end
@@ -53,6 +53,6 @@ module Chromable
   end
 
   alias embedding chroma_embedding unless method_defined? :embedding
-  alias upsert_embedding chroma_embedding unless method_defined? :upsert_embedding
-  alias destroy_embedding chroma_embedding unless method_defined? :destroy_embedding
+  alias upsert_embedding chroma_upsert_embedding unless method_defined? :upsert_embedding
+  alias destroy_embedding chroma_destroy_embedding unless method_defined? :destroy_embedding
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: chromable
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Ali Hamdi Ali Fadel
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-12-07 00:00:00.000000000 Z
+date: 2023-12-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport