llm_memory 0.1.5 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aea28ca9fb65d35a8d98964ac433cb7c445d8137a2672bcf542d35fa31935582
-  data.tar.gz: 02d4d7a619eb3031df0310c2fa9bef9105035f732bc1d9e5381a3dd2d8ded836
+  metadata.gz: c20af963efe2764e2360a0e7c5149afc18b6c757b7c24dc178b3efe190330c20
+  data.tar.gz: 90d3ee12fffc237846e9bd6342ad9863e257bed7ae573c378673cb6c1b4a42f5
 SHA512:
-  metadata.gz: d30618749a0b4016a2ca9cd2815cb6b7b4971a46c50ea83fa3e5b30d1e0813127053b576882587634f166cff7312f3898c6183f71a32ea724b53deea5d676936
-  data.tar.gz: 4957d9857a4a5b05cd725b45e44c44b9ecfb695034e6c2e6c45ed80c9c819d0b581fdc4f9977d4af9180b15af983f544df62507e5d25001f0e508ff90acfa8b6
+  metadata.gz: 2240230ab5c96a4f8642f01114da452207d086e8a35432bf69c8652a170307b200fdacb8f48713c3027074d712bb33628da905a5aec1d5afb2c6db9659723d8f
+  data.tar.gz: 907c03f50e230d7fccbd60244ff6a9c412c5dc7962b1202585c3e56619203c7e3b5dc57be2881d49228167d8296827958315b6e823b69a3e35c32dd08db3be36

data/Gemfile.lock

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

data/README.md

@@ -99,7 +99,7 @@ answer the question: <%= query_str %>
 TEMPLATE
 
 broca = LlmMemory::Broca.new(prompt: prompt, model: 'gpt-3.5-turbo')
-messages = broca.respond(query_str: query_str, related_docs: related_docs)
+message = broca.respond(query_str: query_str, related_docs: related_docs)
 
 ...
 query_str2 = "How are you?"

data/lib/llm_memory/broca.rb

@@ -32,11 +32,12 @@ module LlmMemory
             temperature: @temperature
           }
         )
+        LlmMemory.logger.debug(response)
         response_content = response.dig("choices", 0, "message", "content")
-        @messages.push({role: "system", content: response_content})
+        @messages.push({role: "system", content: response_content}) unless response_content.nil?
         response_content
       rescue => e
-        puts e.inspect
+        LlmMemory.logger.info(e.inspect)
         # @messages = []
         nil
       end
@@ -52,8 +53,9 @@ module LlmMemory
       new_messages = []
       @messages.reverse_each do |message|
         encoded = tokenizer.encode(message[:content], add_special_tokens: true)
-        if count < @max_token
-          count += encoded.tokens.length
+        token_count = encoded.tokens.length
+        count += token_count
+        if count <= @max_token
           new_messages.push(message)
         else
           break

data/lib/llm_memory/hippocampus.rb

@@ -57,12 +57,13 @@ module LlmMemory
         v.each_slice(2).to_h.transform_keys(&:to_sym)
       }
       result.each do |item|
-        item[:metadata] = JSON.parse(item[:metadata])
+        hash = JSON.parse(item[:metadata])
+        item[:metadata] = hash.transform_keys(&:to_sym)
       end
       result
     end
 
-    def forgot_all
+    def forget_all
       @store.drop_index
     end
 

data/lib/llm_memory/version.rb

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

data/lib/llm_memory.rb

@@ -1,19 +1,24 @@
 # frozen_string_literal: true
 
+require "logger"
 # config
 require_relative "llm_memory/configuration"
-
 require_relative "llm_memory/hippocampus"
 require_relative "llm_memory/broca"
 require_relative "llm_memory/wernicke"
-
 require_relative "llm_memory/version"
 
 module LlmMemory
   class Error < StandardError; end
 
   class << self
-    attr_accessor :configuration
+    attr_accessor :configuration, :log_level
+
+    def logger
+      @logger ||= Logger.new($stdout).tap do |logger|
+        logger.level = log_level || Logger::INFO
+      end
+    end
   end
 
   def self.configure

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm_memory
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.7
 platform: ruby
 authors:
 - Shohei Kameda