0.1.14

shohey1226 · Aug 1, 2023 · 4cdcdad · 4cdcdad
1 parent e1d4dab
commit 4cdcdad
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/Gemfile.lock b/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.4)
+    llm_memory_pgvector (0.1.5)
       llm_memory (~> 0.1.7)
       pg (~> 1.5.3)
       pgvector (~> 0.2.0)

diff --git a/lib/llm_memory/version.rb b/lib/llm_memory/version.rb
@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LlmMemory
-  VERSION = "0.1.13"
+  VERSION = "0.1.14"
 end
diff --git a/spec/llm_memory/hippocampus_spec.rb b/spec/llm_memory/hippocampus_spec.rb
@@ -106,6 +106,8 @@
         {content: "I like music", metadata: {info: "hobby"}}
       ]
       hippocampus.memorize(docs)
+      expect(hippocampus.list.map { |s| s["id"] }).to eq([1, 2, 3])
+      expect(hippocampus.list(id: [1, 2]).map { |s| s["id"] }).to eq([1, 2])
       res = hippocampus.query("What is my name?", limit: 2)
       expect(res.first[:content]).to eq("Hello, I'm Shohei.")
       expect(res.first[:metadata][:info]).to eq("name")