llm_memory 0.1.12 → 0.1.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7d5876cf6c293ca2371eab3021704be2710c8b3d558e501543bde4acf6898595
-  data.tar.gz: 232d45abb20fb82b09b0f8ba6b22c12168d498cd8e23110d56446ae5265991aa
+  metadata.gz: 783f2bf830c8e3bf6ac8f033c200bcdc068ac51e23c444a15a28ce4c3fdb3faf
+  data.tar.gz: 0705cd834c8877f368bb8827e7916f72ec81d1e70839b2455fd27f030b1d9346
 SHA512:
-  metadata.gz: 128b6f275d904d2fafb8c7e0c1a26d33462edc015a33b1a3da6b4f1dc6f73ba51ae15f76f25d00221baa167956af06f1d5cd081fe8ab54e54b44c783c7380a98
-  data.tar.gz: 693ec575ca7b4add6b64afb6f3608a3ca6c989274c4f896aaf7520df5e82a0c051d062dd24bd6975a12c585bfaed91b6c4c1a96195e5f1831b05ee3c3679659a
+  metadata.gz: 6adf9915342427085ec647c1475f4169159d8670731b76100820bd13780d43ab7e74e31e36f8eed48453ea7f3a63dd1d4da11e784ab6124a16dfe0a853c17199
+  data.tar.gz: fecd053b1881b4d52a6f587469b7ffb50501d1e75b100eedc27a33039805dfdccb33b2904cb0482754bcbce9c5d034369364db28b7749bcd511eac51176c01a2

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    llm_memory (0.1.12)
+    llm_memory (0.1.13)
       redis (~> 4.6.0)
       ruby-openai (~> 3.7.0)
       tokenizers (~> 0.3.3)

data/lib/llm_memory/hippocampus.rb

@@ -57,6 +57,14 @@ module LlmMemory
       @store.drop_index if @store.index_exists?
     end
 
+    def forget(key)
+      @store.delete(key)
+    end
+
+    def list(array = [])
+      @store.list(array)
+    end
+
     def add_vectors(docs)
       # embed documents and add vector
       result = []

data/lib/llm_memory/store.rb

@@ -27,6 +27,14 @@ module LlmMemory
       raise NotImplementedError, "Each store must implement the 'add' method."
     end
 
+    def list
+      raise NotImplementedError, "Each store must implement the 'list' method."
+    end
+
+    def delete
+      raise NotImplementedError, "Each store must implement the 'delete' method."
+    end
+
     def search
       raise NotImplementedError, "Each store must implement the 'search' method."
     end

data/lib/llm_memory/stores/redis_store.rb

@@ -123,7 +123,33 @@ module LlmMemory
       puts "Unexpected Error: #{e.message}"
     end
 
-    def delete
+    def delete(key)
+      @client.del(key) if @client.exists?(key)
+    end
+
+    def delete_all
+      list.keys.each do |key|
+        delete(key)
+      end
+    end
+
+    def list(patterns = [])
+      patterns = if patterns.empty?
+        ["#{@index_name}:*"]
+      else
+        patterns.map { |pattern| "#{@index_name}:#{pattern}" }
+      end
+      data = {}
+      patterns.each do |pattern|
+        @client.keys(pattern).each do |key|
+          obj = @client.hgetall(key)
+          data[key] = {
+            content: obj.dig("content"),
+            metadata: JSON.parse(obj.dig("metadata"))
+          }
+        end
+      end
+      data
     end
 
     def update

data/lib/llm_memory/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LlmMemory
-  VERSION = "0.1.12"
+  VERSION = "0.1.13"
 end

metadata

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