llm_memory 0.1.12 → 0.1.14

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7d5876cf6c293ca2371eab3021704be2710c8b3d558e501543bde4acf6898595
-  data.tar.gz: 232d45abb20fb82b09b0f8ba6b22c12168d498cd8e23110d56446ae5265991aa
+  metadata.gz: 1ddc205359564a6863d027d91243d6da9cc2079a8965a89c3e0dde0fe13c32cb
+  data.tar.gz: 23a4980aa97684abcf9893c5598446243ec5b9c198c4c083a11d7cd25884c34c
 SHA512:
-  metadata.gz: 128b6f275d904d2fafb8c7e0c1a26d33462edc015a33b1a3da6b4f1dc6f73ba51ae15f76f25d00221baa167956af06f1d5cd081fe8ab54e54b44c783c7380a98
-  data.tar.gz: 693ec575ca7b4add6b64afb6f3608a3ca6c989274c4f896aaf7520df5e82a0c051d062dd24bd6975a12c585bfaed91b6c4c1a96195e5f1831b05ee3c3679659a
+  metadata.gz: 03df070d8832990b091c5c4814632e23c832ba595eceaaaa769a6cd06823d302783bd5b1865b6b5531cd13d07287abeb9ed7a31296d73efd6a3031738502074b
+  data.tar.gz: 9b46d3197283516e44aec1de34799a0ae50b9232f00b6c55f462b8a529c8de9e38d2b6971b0eb9d09056a03cb4f6978787783921deca018863b8437accf37dc2

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    llm_memory (0.1.12)
+    llm_memory (0.1.14)
       redis (~> 4.6.0)
       ruby-openai (~> 3.7.0)
       tokenizers (~> 0.3.3)
@@ -24,7 +24,7 @@ GEM
     json (2.6.3)
     language_server-protocol (3.17.0.3)
     lint_roller (1.0.0)
-    llm_memory_pgvector (0.1.2)
+    llm_memory_pgvector (0.1.5)
       llm_memory (~> 0.1.7)
       pg (~> 1.5.3)
       pgvector (~> 0.2.0)
@@ -35,7 +35,7 @@ GEM
     parser (3.2.2.1)
       ast (~> 2.4.1)
     pg (1.5.3)
-    pgvector (0.2.0)
+    pgvector (0.2.1)
     pry (0.14.2)
       coderay (~> 1.1)
       method_source (~> 1.0)

data/lib/llm_memory/hippocampus.rb

@@ -57,6 +57,18 @@ module LlmMemory
       @store.drop_index if @store.index_exists?
     end
 
+    def forget(key)
+      @store.delete(key)
+    end
+
+    def list(*args)
+      @store.list(*args)
+    end
+
+    def get(key)
+      @store.get(key)
+    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,23 @@ 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 get(key)
+      @client.hgetall(key)
+    end
+
+    def list(*args)
+      pattern = "#{@index_name}:#{args.first || "*"}"
+      @client.keys(pattern)
     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.14"
 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.14
 platform: ruby
 authors:
 - Shohei Kameda
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-07-27 00:00:00.000000000 Z
+date: 2023-08-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: tokenizers