llm_memory_pgvector 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aa1b3ab9226fa280c8655446df1fd45cb6a1dfc54779d1c5f71c4f407945fdd6
-  data.tar.gz: 8ae625dae794fbcb87bc915fbb766edbd0a09c5be0afe0c4be3e3ae76e757716
+  metadata.gz: 649ee254ec15284dedeb54afa917347e666472e7719060724d430d27ad2d9767
+  data.tar.gz: 3ed88178232fed65128929c364a3c8359a14487f0fc26f8f4535593173d6b83e
 SHA512:
-  metadata.gz: 65069e24e02fa4dca85c4cdbe0a018337ad511a5cc35a680d9ddb31b59d8c9ee0e409de8934a7fc641a477995ba1176f94bfce8e94a55aeae201653895a8bc22
-  data.tar.gz: bed6685f42c5129f75905a511e4ee78b0fbb546068be6f5fadb5e3ba7df2656333b50517a01c810703b4ee3971dc46475fa248d190ead3702e8ad76bdf6b21cf
+  metadata.gz: 6bb31d3168a85453de4920680bbcd9cf652b74058d6825a87b901aa79fa6af16f929d057e91fc2a9720019f5deae1e4b3dee6f32bf37ea6e5509b864fec70dab
+  data.tar.gz: 1638222ad8b71b281f68ed73d02f47896a8760ec116025738301d7981089c5fff50d4cd854c0ecfd50a51f5143b702a2dfd3c1249535ee0e91dcf1b7c7c35e3a

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    llm_memory_pgvector (0.1.3)
+    llm_memory_pgvector (0.1.5)
       llm_memory (~> 0.1.7)
       pg (~> 1.5.3)
       pgvector (~> 0.2.0)

data/lib/llm_memory_pgvector/pgvector_store.rb

@@ -41,12 +41,13 @@ module LlmMemoryPgvector
       @conn.exec("DROP TABLE IF EXISTS #{@index_name}")
     end
 
-    def list(ids = [])
-      result = if ids.empty?
+    def list(*args)
+      id = (args.length == 1) ? args.first[:id] : []
+      result = if id.empty?
         @conn.exec("SELECT * FROM #{@index_name}")
       else
-        placeholders = ids.map.with_index(1) { |_, i| "$#{i}" }.join(",")
-        @conn.exec_params("SELECT * FROM #{@index_name} WHERE id IN (#{placeholders})", ids)
+        placeholders = id.map.with_index(1) { |_, i| "$#{i}" }.join(",")
+        @conn.exec_params("SELECT * FROM #{@index_name} WHERE id IN (#{placeholders})", id)
       end
       result.entries
     end
@@ -56,6 +57,11 @@ module LlmMemoryPgvector
       result.cmd_tuples > 0
     end
 
+    def get(id)
+      result = @conn.exec_params("SELECT * FROM #{@index_name} WHERE id = $1", [id])
+      result.entries.first
+    end
+
     def delete_all
       @conn.exec("DELETE FROM #{@index_name}")
       true
@@ -65,13 +71,15 @@ module LlmMemoryPgvector
 
     # data = [{ content: "", vector: [], metadata: {} },,]
     def add(data: [])
+      new_ids = []
       data.each do |row|
-        @conn.exec_params(
-          "INSERT INTO #{@index_name} (#{@content_key}, #{@metadata_key}, #{@vector_key}) VALUES ($1, $2, $3)",
+        result = @conn.exec_params(
+          "INSERT INTO #{@index_name} (#{@content_key}, #{@metadata_key}, #{@vector_key}) VALUES ($1, $2, $3) RETURNING id",
           [row[@content_key.to_sym], row[@metadata_key.to_sym].to_json, row[@vector_key.to_sym]]
         )
+        new_ids << result[0]["id"].to_i if result && result[0] && result[0]["id"]
       end
-      list
+      list(id: new_ids)
     end
 
     def search(query: [], k: 3)

data/lib/llm_memory_pgvector/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LlmMemoryPgvector
-  VERSION = "0.1.3"
+  VERSION = "0.1.5"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: llm_memory_pgvector
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - Shohei Kameda
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-07-31 00:00:00.000000000 Z
+date: 2023-08-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: llm_memory