speaky 0.1.9 → 0.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 564d48d19c0f2c99ef2266064911bc1010523dd52d127c7aca4262c8b329700f
-  data.tar.gz: 38c4365d3a09e26523c79da531211e276f53aa4a2ebff0a7337af8ff6e47a307
+  metadata.gz: d4c6b9331e1d4d69bf0e8e66f22fc9e43620874d22400552a418af290d286969
+  data.tar.gz: 1878fe018b91f0caede735903334a2e86a9a1ecbfd3814ebccfe6857f58ff606
 SHA512:
-  metadata.gz: 565cba4dfa83cd596d3fafbf45db271ffc99225726a7af0ad7a899f859c8157cca0da0308f79957f0d7307ad15e9c34ef31e009e960dc4cadd35b77cd171dbed
-  data.tar.gz: 81edbf263c317fe847fe6befbaf6d8bef329114ab8e1f87fefc440b6499d9218ec0090e39e0cad414f61e421f98a97a74c0b499247834a4d9ab3fd8c28bae9b6
+  metadata.gz: 3dbb0b0cd00be46b2ab61ff749474623c182c067ffdc3e51b6b20c5f1d87587f78d1290577cc61e6276db5574b81cb8116afe5db9f7a2e9aa0e4e078c220ae0e
+  data.tar.gz: 628ae505b3b8b6dbce97d08dcf953c568116d21725a2d4f3c82ee04c49815ac0c8dc10621af9e17f97aed9c34ee26426b41a3c7d2e31e319a5a04c8253c97880

data/lib/speaky/llm_openai.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-require "openai"
-
 module Speaky
   class LlmOpenai < LlmBase
     def initialize(config)
+      require "openai"
+
       @config = config
 
       # check if the access token is set

data/lib/speaky/vectorstore_faiss.rb

@@ -1,51 +1,57 @@
-# # frozen_string_literal: true
+# frozen_string_literal: true
 
-# require 'faiss'
+module Speaky
+  class VectorstoreFaiss < VectorstoreBase
+    def initialize(config)
+      require 'faiss'
 
-# module Speaky
-#   class VectorstoreFaiss < VectorstoreBase
-#     def initialize(config)
-#       @config = config
+      @config = config
 
-#       # check if the index path is set
-#       raise ArgumentError, 'index_path is required' unless @config[:index_path]
+      # check if the index path is set
+      raise ArgumentError, 'index_path is required' unless @config[:index_path]
 
-#       # load index from index_path if exists
-#       if File.exist?(@config[:index_path])
-#         @index = Faiss::Index.load(@config[:index_path])
-#       else
-#         # create a new index
-#         index = Faiss::IndexFlatL2.new(1536)
-#         @index = Faiss::IndexIDMap.new(index)
-#         @index.save(@config[:index_path])
-#       end
-#     end
+      # load index from index_path if exists
+      if File.exist?(@config[:index_path])
+        @index = Faiss::Index.load(@config[:index_path])
+      else
+        # create a new index
+        index = Faiss::IndexFlatL2.new(1536)
+        @index = Faiss::IndexIDMap.new(index)
+        @index.save(@config[:index_path])
+      end
+    end
 
-#     def add(id, data)
-#       embeddings = Speaky.llm.embed(data)
+    def add(id, data)
+      embeddings = Speaky.llm.embed(data)
 
-#       @index.add_with_ids([embeddings], [string_id_to_unique_int_id(id)])
+      @index.add_with_ids([embeddings], [string_id_to_unique_int_id(id)])
 
-#       true
-#     end
+      true
+    end
 
-#     def remove(id)
-#       # remove is not supported by Faiss
-#       true
-#     end
+    def remove(id)
+      # remove is not supported by Faiss
+      true
+    end
 
-#     def query(question)
-#       # TODO: implement query
+    def query(question)
+      # TODO
 
-#       []
-#     end
+      []
+    end
 
-#     private
+    def reset
+      # TODO
 
-#     # 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
+      true
+    end
+
+    private
+
+    # 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/vectorstore_qdrant.rb

@@ -1,11 +1,12 @@
 # frozen_string_literal: true
 
 require 'rails'
-require 'qdrant'
 
 module Speaky
   class VectorstoreQdrant < VectorstoreBase
     def initialize(config)
+      require 'qdrant'
+
       @config = config
 
       # check if required fields are set

data/lib/speaky/version.rb

@@ -1,3 +1,3 @@
 module Speaky
-  VERSION = '0.1.9'
+  VERSION = '0.1.10'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: speaky
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.1.10
 platform: ruby
 authors:
 - Gregorio Galante
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-03-25 00:00:00.000000000 Z
+date: 2024-03-26 00:00:00.000000000 Z
 dependencies: []
 description: Store activerecord models in vector stores and query them with LLMs!
 email: