speaky 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6223b310b2223f1af3c8c2c45dc5ebe90aade3ed8cf040b9caad69a03b97e06c
-  data.tar.gz: 77b50bfbf7925c14f78504774c7502c1018c27ef1e80bd6ebebf743310ed511a
+  metadata.gz: 06f3d5c76d75de0c05238c26dfc83d3707d7cc5448f83ef474249b0bd5df185b
+  data.tar.gz: 64cac48f6f8fcdd88512940ae3888d736f7b52890dd6cf92a7f91f6716ffee48
 SHA512:
-  metadata.gz: cd78c665ea63fc51f7ff069e5e7eb1ee41635e9bddcdf1e5f6105762442bb5b7be12ac890742cf9d7e2d62b7aab903a12b11723b45bce6c385fc57c59a23a5eb
-  data.tar.gz: 491775c48104c32a125934f9e8ffbd5f5960aebc357b08f135e1b78cc640409b1df5923cf0b3e0ac855c1f01ebc161f99b2981d8e6c0b886cf9c4e44e3924783
+  metadata.gz: 104878c920e7de1c268e727b2ff774f8a6f8c0e240be6cc5b252ee054beeb3f9ae37b36bf7f57ae5491e8d26b3af670b229304885d21f7b0bd63c3686eea5cf4
+  data.tar.gz: 20a27ca58c2f0184d153e5652e9a71058335a9ba1ebd1fd9281f542d95df1c38a8ac5345976f93dd9a824f89b58bdec35440eb441dbf7b443b0a7347db1ad142

data/README.md

@@ -62,8 +62,6 @@ class MyModel < ApplicationRecord
   end
 
   # Add any callbacks that should update the model instance data in the vector store.
-  after_create :create_for_speaky
-  after_update :update_for_speaky
   after_save :save_for_speaky
   after_destroy :destroy_for_speaky
 end

data/lib/speaky/concern.rb

@@ -11,9 +11,9 @@ module Speaky
       self.to_json
     end
 
-    def create_for_speaky
+    def save_for_speaky
       begin
-        Speaky.vectorstore.add(self.id, self.as_speaky)
+        Speaky.vectorstore.add("#{self.class.name}_#{self.id}", self.as_speaky)
       rescue StandardError => e
         Rails.logger.error(e)
         errors.add(:base, 'Failed to create for speaky')
@@ -21,34 +21,14 @@ module Speaky
       end
     end
 
-    def update_for_speaky
-      begin
-        Speaky.vectorstore.update(self.id, self.as_speaky)
-      rescue StandardError => e
-        Rails.logger.error(e)
-        errors.add(:base, 'Failed to update for speaky')
-        raise ActiveRecord::Rollback
-      end
-    end
-
     def destroy_for_speaky
       begin
-        Speaky.vectorstore.remove(self.id)
+        Speaky.vectorstore.remove("#{self.class.name}_#{self.id}")
       rescue StandardError => e
         Rails.logger.error(e)
         errors.add(:base, 'Failed to destroy for speaky')
         raise ActiveRecord::Rollback
       end
     end
-
-    def save_for_speaky
-      ActiveRecord::Base.transaction do
-        if self.new_record?
-          create_for_speaky
-        else
-          update_for_speaky
-        end
-      end
-    end
   end
 end

data/lib/speaky/vectorstore_base.rb

@@ -17,13 +17,6 @@ module Speaky
       raise NotImplementedError
     end
 
-    # Update a vector in the vectorstore.
-    # NOTE: If the vector does not exist, it will be added.
-    # Returns a boolean.
-    def update(id, data)
-      raise NotImplementedError
-    end
-
     # Remove a vector from the vectorstore.
     # NOTE: If the vector does not exist, it will be ignored.
     # Returns a boolean.

data/lib/speaky/vectorstore_qdrant.rb

@@ -46,10 +46,6 @@ module Speaky
       true
     end
 
-    def update(id, data)
-      add(id, data)
-    end
-
     def remove(id)
       points_delete = @client.points.delete(
         collection_name: @config[:collection_name],

data/lib/speaky/version.rb

@@ -1,3 +1,3 @@
 module Speaky
-  VERSION = '0.1.3'
+  VERSION = '0.1.5'
 end

data/lib/speaky.rb

@@ -111,11 +111,11 @@ module Speaky
       Rails.application.eager_load!
       rails_models = ActiveRecord::Base.descendants
       rails_models.each do |rails_model|
-        next unless rails_model.instance_methods.include?(:create_for_speaky)
+        next unless rails_model.instance_methods.include?(:save_for_speaky)
 
         rails_model.find_in_batches do |batch|
           batch.each do |record|
-            record.create_for_speaky
+            record.save_for_speaky
           end
         end
       end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: speaky
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - Gregorio Galante