llm_memory_pgvector 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 02a8d906cb751a0dffa89839cc7f293c970c618285c917ca5521ed9091f4cad0
-  data.tar.gz: 7eeced97f49923746900caf68269041f93621f588cf743195aa4acdcdb725d8f
+  metadata.gz: aa1b3ab9226fa280c8655446df1fd45cb6a1dfc54779d1c5f71c4f407945fdd6
+  data.tar.gz: 8ae625dae794fbcb87bc915fbb766edbd0a09c5be0afe0c4be3e3ae76e757716
 SHA512:
-  metadata.gz: c92ec2de76562dca9b5b3a8d0c0414858d58a73fc1cf43a575e96dd446016eb780f42ef4ae04354eec3bb63e615bef7f2efb4b6aae3a4faffc507e0a78efa781
-  data.tar.gz: 8989283b6f74ffb336b479114aa3b5a43790442924f51582c5ab3e6402e7460fe3cabd466591877dfc6c9aaf0f95037a926c24fa8b506aa5ecfd2ce9b2ae3a35
+  metadata.gz: 65069e24e02fa4dca85c4cdbe0a018337ad511a5cc35a680d9ddb31b59d8c9ee0e409de8934a7fc641a477995ba1176f94bfce8e94a55aeae201653895a8bc22
+  data.tar.gz: bed6685f42c5129f75905a511e4ee78b0fbb546068be6f5fadb5e3ba7df2656333b50517a01c810703b4ee3971dc46475fa248d190ead3702e8ad76bdf6b21cf

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    llm_memory_pgvector (0.1.1)
+    llm_memory_pgvector (0.1.3)
       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,6 +41,28 @@ module LlmMemoryPgvector
       @conn.exec("DROP TABLE IF EXISTS #{@index_name}")
     end
 
+    def list(ids = [])
+      result = if ids.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)
+      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 delete_all
+      @conn.exec("DELETE FROM #{@index_name}")
+      true
+    rescue PG::Error
+      false
+    end
+
     # data = [{ content: "", vector: [], metadata: {} },,]
     def add(data: [])
       data.each do |row|
@@ -49,6 +71,7 @@ module LlmMemoryPgvector
           [row[@content_key.to_sym], row[@metadata_key.to_sym].to_json, row[@vector_key.to_sym]]
         )
       end
+      list
     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.3"
 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.3
 platform: ruby
 authors:
 - Shohei Kameda
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-05-18 00:00:00.000000000 Z
+date: 2023-07-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: llm_memory