llm_memory_pgvector 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aa1b3ab9226fa280c8655446df1fd45cb6a1dfc54779d1c5f71c4f407945fdd6
-  data.tar.gz: 8ae625dae794fbcb87bc915fbb766edbd0a09c5be0afe0c4be3e3ae76e757716
+  metadata.gz: 429622b6f10a9c12ecc17068730ebdefa18f23c06560347bd97201efd5ef97de
+  data.tar.gz: 7c6b463ccc7e5289a711635befa43ce075dc8169bba678803b5c54f5d05c2571
 SHA512:
-  metadata.gz: 65069e24e02fa4dca85c4cdbe0a018337ad511a5cc35a680d9ddb31b59d8c9ee0e409de8934a7fc641a477995ba1176f94bfce8e94a55aeae201653895a8bc22
-  data.tar.gz: bed6685f42c5129f75905a511e4ee78b0fbb546068be6f5fadb5e3ba7df2656333b50517a01c810703b4ee3971dc46475fa248d190ead3702e8ad76bdf6b21cf
+  metadata.gz: 2ee72e800eafc4ec006a4cc528345fa9d8b4911656ff773305046e65105499a300c7a7cc11ae34f274955600515bbf4e60cc8a77c432adae38fcbae8569fe903
+  data.tar.gz: 637159935cd9f9d33c230fbb2aaf46c64e55b8db35cf282747bc75ccff27742863a96d0d8853e8f4588991fa0bc67a98e7ca0b19312b1455f9167a4cef7d728a

data/Gemfile.lock

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

data/lib/llm_memory_pgvector/pgvector_store.rb

@@ -41,12 +41,12 @@ module LlmMemoryPgvector
       @conn.exec("DROP TABLE IF EXISTS #{@index_name}")
     end
 
-    def list(ids = [])
-      result = if ids.empty?
+    def list(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 +56,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 +70,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.4"
 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.4
 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