llm_memory_pgvector 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 02a8d906cb751a0dffa89839cc7f293c970c618285c917ca5521ed9091f4cad0
-  data.tar.gz: 7eeced97f49923746900caf68269041f93621f588cf743195aa4acdcdb725d8f
+  metadata.gz: 429622b6f10a9c12ecc17068730ebdefa18f23c06560347bd97201efd5ef97de
+  data.tar.gz: 7c6b463ccc7e5289a711635befa43ce075dc8169bba678803b5c54f5d05c2571
 SHA512:
-  metadata.gz: c92ec2de76562dca9b5b3a8d0c0414858d58a73fc1cf43a575e96dd446016eb780f42ef4ae04354eec3bb63e615bef7f2efb4b6aae3a4faffc507e0a78efa781
-  data.tar.gz: 8989283b6f74ffb336b479114aa3b5a43790442924f51582c5ab3e6402e7460fe3cabd466591877dfc6c9aaf0f95037a926c24fa8b506aa5ecfd2ce9b2ae3a35
+  metadata.gz: 2ee72e800eafc4ec006a4cc528345fa9d8b4911656ff773305046e65105499a300c7a7cc11ae34f274955600515bbf4e60cc8a77c432adae38fcbae8569fe903
+  data.tar.gz: 637159935cd9f9d33c230fbb2aaf46c64e55b8db35cf282747bc75ccff27742863a96d0d8853e8f4588991fa0bc67a98e7ca0b19312b1455f9167a4cef7d728a

data/Gemfile.lock

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

data/README.md

@@ -4,15 +4,13 @@ This is a Ruby plugin for the [llm_memory](https://github.com/shohey1226/llm_mem
 
 ## Installation
 
-TODO: Replace `UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG` with your gem name right after releasing it to RubyGems.org. Please do not do it earlier due to security reasons. Alternatively, replace this section with instructions to install your gem from git if you don't plan to release to RubyGems.org.
-
 Install the gem and add to the application's Gemfile by executing:
 
-    $ bundle add UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORGg
+    $ bundle add llm_memory_pgvector 
 
 If bundler is not being used to manage dependencies, install the gem by executing:
 
-    $ gem install UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+    $ gem install llm_memory_pgvector 
 
 Plese don't forget to add `llm_memory`
 

data/lib/llm_memory_pgvector/pgvector_store.rb

@@ -41,14 +41,44 @@ module LlmMemoryPgvector
       @conn.exec("DROP TABLE IF EXISTS #{@index_name}")
     end
 
+    def list(id: [])
+      result = if id.empty?
+        @conn.exec("SELECT * FROM #{@index_name}")
+      else
+        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
+
+    def delete(id)
+      result = @conn.exec_params("DELETE FROM #{@index_name} WHERE id = $1", [id])
+      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
+    rescue PG::Error
+      false
+    end
+
     # 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(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.2"
+  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.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Shohei Kameda
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-05-18 00:00:00.000000000 Z
+date: 2023-08-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: llm_memory