chromable 0.3.3 → 0.3.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 30a53a0c8bde8c1d4c091e073058d6273d57cfd191cfeff0f4f48817e2bf14d4
-  data.tar.gz: 4374788d45a3acb8e491c452f862bb20bf5634dcfb483cf194d9c7090b349de7
+  metadata.gz: ed812a621b3f57a8deb5493793874dc01c217b463b68b9b2f44c4b439b0b933e
+  data.tar.gz: 3c70f3a0358049ff04fd62234e6fb2a3107a678673fc85de9af914e6d990a624
 SHA512:
-  metadata.gz: 2488713488c214ab53be05616da9ca3a7ee7e3c619c4627d3b338a1454680add7284b2fef6e84007285acc0f3830d24290f09024b5189241e3657bb9086cce17
-  data.tar.gz: bf63bb358eb179ae5ab7e02fcdeb88d1657d2d3009e28b3e3c5d7354a8bb0a7e387a09020920d8bc88cb03ad10fb18ac4d5f3ac8926a979094a0fa575340af2e
+  metadata.gz: 71591b825092f3f05da1687854ca62766d90b759c9e5e7ad7b1130ba78386bf65b51e18d3c7c3ba12be084710eabb39c0ce207817e4a1f03a558e39fc20a657f
+  data.tar.gz: dc01cf2ebc8f14c513ef45996e2f55557d196f740f2869e2cf36bd70658b0bfa58698e031c5af536a8cc3542f04e638de4673c0c5a2d7fcedf73b83a606ec85f

data/chromable.gemspec

@@ -32,7 +32,7 @@ Gem::Specification.new do |spec|
   spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
   spec.require_paths = ['lib']
 
-  spec.add_dependency 'chroma-db', '>= 0.6.0'
+  spec.add_dependency 'chroma-db', '>= 0.7.0'
 
   # For more information and examples about making a new gem, check out our
   # guide at: https://bundler.io/guides/creating_gem.html

data/lib/chromable/version.rb

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

data/lib/chromable.rb

@@ -23,6 +23,10 @@ module Chromable
       @embedder = embedder
       @keep_document = keep_document
     end
+
+    def collection_name # rubocop:disable Lint/DuplicateMethods
+      @collection_name.is_a?(Proc) ? @collection_name.call : @collection_name
+    end
   end
 
   # Methods to be added to the model class.
@@ -38,7 +42,7 @@ module Chromable
     end
 
     def chromable(**options)
-      options[:collection_name] ||= name.underscore.pluralize
+      options[:collection_name] ||= -> { name.underscore.pluralize }
 
       self.chromable_settings = Settings.new(**options)
     end
@@ -49,6 +53,8 @@ module Chromable
 
     def chroma_delete_collection
       Chroma::Resources::Collection.delete(chromable_settings.collection_name)
+    rescue Chroma::InvalidRequestError
+      Chroma::Util.log_debug("Collection #{chromable_settings.collection_name} does not exist.")
     end
 
     def chroma_query(text:, results: 10, where: {}, where_document: {}, **embedder_options)
@@ -75,7 +81,7 @@ module Chromable
     end
 
     def chroma_embedding
-      self.class.chroma_collection.get(ids: [id])[0]
+      self.class.chroma_collection.get(ids: [id.to_s])[0]
     end
 
     def chroma_upsert_embedding
@@ -83,13 +89,13 @@ module Chromable
     end
 
     def chroma_destroy_embedding
-      self.class.chroma_collection.delete(ids: [id])
+      self.class.chroma_collection.delete(ids: [id.to_s])
     end
 
     def chroma_neighbors(results: 10, where: {}, where_document: {})
       collection = self.class.chroma_collection
 
-      embedding = collection.get(ids: [id], include: [:embeddings])[0].embedding
+      embedding = collection.get(ids: [id.to_s], include: [:embeddings])[0].embedding
 
       self.class.find(collection.query(
         query_embeddings: [embedding],

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: chromable
 version: !ruby/object:Gem::Version
-  version: 0.3.3
+  version: 0.3.5
 platform: ruby
 authors:
 - Ali Hamdi Ali Fadel
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-12-18 00:00:00.000000000 Z
+date: 2025-02-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: chroma-db
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.6.0
+        version: 0.7.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.6.0
+        version: 0.7.0
 description: 
 email:
 - aliosm1997@gmail.com