llm_memory 0.1.5 → 0.1.6
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/Gemfile.lock +1 -1
- data/lib/llm_memory/broca.rb +6 -4
- data/lib/llm_memory/version.rb +1 -1
- data/lib/llm_memory.rb +8 -3
- metadata +1 -1
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: 72e01db010504ae3ec7877d5122ecba01bf7c1f15140e7ed94d9e66c62e2ea40
|
4
|
+
data.tar.gz: 8d170f4f7fda8c1af72fe62bb7873f3758fba9960babb1dc562b048bb4201f09
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: be7ac41f195ae267c60b917f5640be23bf3a458fea0d6fd9de1084367db6b89bc517a6285fc4459ad5f0a36fb82257bf7cb0dc925233910a567f15fb74e1fb7e
|
7
|
+
data.tar.gz: '0888eea1d469818a6ea2272b164218d934b3a1591e7e7179d74d5748e4e3765082a0ea40a3177793a8be60b82f702341f8eb4ac387639c8d219cbb09de263908'
|
data/Gemfile.lock
CHANGED
data/lib/llm_memory/broca.rb
CHANGED
@@ -32,11 +32,12 @@ module LlmMemory
|
|
32
32
|
temperature: @temperature
|
33
33
|
}
|
34
34
|
)
|
35
|
+
LlmMemory.logger.debug(response)
|
35
36
|
response_content = response.dig("choices", 0, "message", "content")
|
36
|
-
@messages.push({role: "system", content: response_content})
|
37
|
+
@messages.push({role: "system", content: response_content}) unless response_content.nil?
|
37
38
|
response_content
|
38
39
|
rescue => e
|
39
|
-
|
40
|
+
LlmMemory.logger.info(e.inspect)
|
40
41
|
# @messages = []
|
41
42
|
nil
|
42
43
|
end
|
@@ -52,8 +53,9 @@ module LlmMemory
|
|
52
53
|
new_messages = []
|
53
54
|
@messages.reverse_each do |message|
|
54
55
|
encoded = tokenizer.encode(message[:content], add_special_tokens: true)
|
55
|
-
|
56
|
-
|
56
|
+
token_count = encoded.tokens.length
|
57
|
+
count += token_count
|
58
|
+
if count <= @max_token
|
57
59
|
new_messages.push(message)
|
58
60
|
else
|
59
61
|
break
|
data/lib/llm_memory/version.rb
CHANGED
data/lib/llm_memory.rb
CHANGED
@@ -1,19 +1,24 @@
|
|
1
1
|
# frozen_string_literal: true
|
2
2
|
|
3
|
+
require "logger"
|
3
4
|
# config
|
4
5
|
require_relative "llm_memory/configuration"
|
5
|
-
|
6
6
|
require_relative "llm_memory/hippocampus"
|
7
7
|
require_relative "llm_memory/broca"
|
8
8
|
require_relative "llm_memory/wernicke"
|
9
|
-
|
10
9
|
require_relative "llm_memory/version"
|
11
10
|
|
12
11
|
module LlmMemory
|
13
12
|
class Error < StandardError; end
|
14
13
|
|
15
14
|
class << self
|
16
|
-
attr_accessor :configuration
|
15
|
+
attr_accessor :configuration, :log_level
|
16
|
+
|
17
|
+
def logger
|
18
|
+
@logger ||= Logger.new($stdout).tap do |logger|
|
19
|
+
logger.level = log_level || Logger::INFO
|
20
|
+
end
|
21
|
+
end
|
17
22
|
end
|
18
23
|
|
19
24
|
def self.configure
|