speaky 0.1.5 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 06f3d5c76d75de0c05238c26dfc83d3707d7cc5448f83ef474249b0bd5df185b
-  data.tar.gz: 64cac48f6f8fcdd88512940ae3888d736f7b52890dd6cf92a7f91f6716ffee48
+  metadata.gz: 978cd92d282523846c4b7b592776656f3d351162a361e28460de95e4af338378
+  data.tar.gz: dac048c47d89c41ef1b96dc31d3a7f600ee2ef393d59bffd203af1b7f97c17cd
 SHA512:
-  metadata.gz: 104878c920e7de1c268e727b2ff774f8a6f8c0e240be6cc5b252ee054beeb3f9ae37b36bf7f57ae5491e8d26b3af670b229304885d21f7b0bd63c3686eea5cf4
-  data.tar.gz: 20a27ca58c2f0184d153e5652e9a71058335a9ba1ebd1fd9281f542d95df1c38a8ac5345976f93dd9a824f89b58bdec35440eb441dbf7b443b0a7347db1ad142
+  metadata.gz: 06e1d13344593cce42a57f730d44e4f838660a7581810a5357e023b42a6f7a00c866e3fadcbb98d6a4398f202a0c693c02202c69415f7eac16ca7f9190903b93
+  data.tar.gz: 3afc6a7bd45fd9fff233a5e753a6778421d8664405ccc33c1e7189988770056dcd6a45dd04c60052aa2d24b1c7eb61ecb85ee508959c9360e53e1b26ca70f211

data/lib/speaky/concern.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require 'rails'
 require 'active_support'
 
 module Speaky

data/lib/speaky/vectorstore_qdrant.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require 'rails'
 require 'qdrant'
 
 module Speaky
@@ -29,7 +30,7 @@ module Speaky
         collection_name: @config[:collection_name],
         points: [
           {
-            id: id,
+            id: string_id_to_unique_int_id(id),
             vector: embeddings,
             payload: {
               content: data
@@ -40,6 +41,7 @@ module Speaky
       )
 
       if !points_upsert || points_upsert.dig('status') != 'ok'
+        Rails.logger.error(points_upsert)
         raise 'Failed to add vector'
       end
 
@@ -49,10 +51,11 @@ module Speaky
     def remove(id)
       points_delete = @client.points.delete(
         collection_name: @config[:collection_name],
-        points: [id],
+        points: [string_id_to_unique_int_id(id)],
       )
 
       if !points_delete || points_delete.dig('status') != 'ok'
+        Rails.logger.error(points_delete)
         raise 'Failed to remove vector'
       end
 
@@ -71,6 +74,7 @@ module Speaky
       )
 
       if !points_search || points_search.dig('status') != 'ok'
+        Rails.logger.error(points_search)
         raise 'Failed to search vectors'
       end
 
@@ -80,6 +84,7 @@ module Speaky
     def reset
       collections_delete = @client.collections.delete(collection_name: @config[:collection_name])
       if !collections_delete || collections_delete.dig('status') != 'ok'
+        Rails.logger.error(collections_delete)
         raise 'Failed to delete collection'
       end
 
@@ -99,6 +104,7 @@ module Speaky
           }
         )
         if !collections_create || collections_create.dig('status') != 'ok'
+          Rails.logger.error(collections_create)
           raise 'Failed to create collection'
         end
       end
@@ -106,10 +112,17 @@ module Speaky
       # create index for field "id" in collection
       collections_create_index = @client.collections.create_index(collection_name: @config[:collection_name], field_name: 'id', field_schema: 'keyword')
       if !collections_create_index || collections_create_index.dig('status') != 'ok'
+        Rails.logger.error(collections_create_index)
         raise 'Failed to create index for field "id" on collection'
       end
 
       true
     end
+
+    # This method is used to convert a string ID to a unique integer ID
+    # that can be used by the Qdrant API.
+    def string_id_to_unique_int_id(string_id)
+      string_id.to_s.hash.abs
+    end
   end
 end

data/lib/speaky/version.rb

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

metadata

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