llm_memory 0.1.13 → 0.1.14

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 783f2bf830c8e3bf6ac8f033c200bcdc068ac51e23c444a15a28ce4c3fdb3faf
-  data.tar.gz: 0705cd834c8877f368bb8827e7916f72ec81d1e70839b2455fd27f030b1d9346
+  metadata.gz: 1ddc205359564a6863d027d91243d6da9cc2079a8965a89c3e0dde0fe13c32cb
+  data.tar.gz: 23a4980aa97684abcf9893c5598446243ec5b9c198c4c083a11d7cd25884c34c
 SHA512:
-  metadata.gz: 6adf9915342427085ec647c1475f4169159d8670731b76100820bd13780d43ab7e74e31e36f8eed48453ea7f3a63dd1d4da11e784ab6124a16dfe0a853c17199
-  data.tar.gz: fecd053b1881b4d52a6f587469b7ffb50501d1e75b100eedc27a33039805dfdccb33b2904cb0482754bcbce9c5d034369364db28b7749bcd511eac51176c01a2
+  metadata.gz: 03df070d8832990b091c5c4814632e23c832ba595eceaaaa769a6cd06823d302783bd5b1865b6b5531cd13d07287abeb9ed7a31296d73efd6a3031738502074b
+  data.tar.gz: 9b46d3197283516e44aec1de34799a0ae50b9232f00b6c55f462b8a529c8de9e38d2b6971b0eb9d09056a03cb4f6978787783921deca018863b8437accf37dc2

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    llm_memory (0.1.13)
+    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

@@ -61,8 +61,12 @@ module LlmMemory
       @store.delete(key)
     end
 
-    def list(array = [])
-      @store.list(array)
+    def list(*args)
+      @store.list(*args)
+    end
+
+    def get(key)
+      @store.get(key)
     end
 
     def add_vectors(docs)

data/lib/llm_memory/stores/redis_store.rb

@@ -133,23 +133,13 @@ module LlmMemory
       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
+    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.13"
+  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.13
+  version: 0.1.14
 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: tokenizers