llm_memory 0.1.10 → 0.1.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5117e4a12dc1c9a3848817dc949038d6c80aa3cfb62b2c10709331fd19baaea8
-  data.tar.gz: 3117e77cd3f0741e94c419cfbc1cb7390aaf84f63aea5df181fdf29df77eb7a4
+  metadata.gz: c72d46a390a534d27ea5ee90bfb2950b75fd16f5320af2a431a314bcee13be30
+  data.tar.gz: 974f529b0158cf5b4a8d4330dca59fe2fd3c2ee37cdca6465d98abee0f84db18
 SHA512:
-  metadata.gz: 05ea60dbd58ea1aabe974e32b22253febe355701940e16f04da685ecefadcdf58323bd8ab28354af6675cc57f17a9a3cc30e64595684deb1be79d3c8251e1eb0
-  data.tar.gz: da8cf3c76bb436e9665241a135ba1154b45581de08a6cb77a3022066c6c3898ea6beb749d561e7264e19cd2bf900e21c908980c4f03ab0dcdbf52b36036ccb26
+  metadata.gz: 697d7a93bc637ec00004e68cd83dac1077e9705706e6b0123131acfaac201ce7979eeef63772bae35c53fa1b95ea7123f23399e5ee0ea1b454c689b6aa0f6a33
+  data.tar.gz: ab67fd9ec6c1297b5308dfc28b9475bab46e28aa3b4f2af1694a9053ed7203c80e478320c11663a49583e43762814b85e0bdc52d2ea7002274a282609b1db9bd

data/Gemfile.lock

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

data/lib/llm_memory/broca.rb

@@ -43,6 +43,46 @@ module LlmMemory
       end
     end
 
+    def respond_with_schema(context: {}, schema: {})
+      response_content = respond(context)
+      begin
+        response = client.chat(
+          parameters: {
+            model: "gpt-3.5-turbo-0613", # as of July 3, 2023
+            messages: [
+              {
+                role: "user",
+                content: response_content
+              }
+            ],
+            functions: [
+              {
+                name: "broca",
+                description: "Formating the content with the specified schema",
+                parameters: schema
+              }
+            ]
+          }
+        )
+        LlmMemory.logger.debug(response)
+        message = response.dig("choices", 0, "message")
+        if message["role"] == "assistant" && message["function_call"]
+          function_name = message.dig("function_call", "name")
+          args =
+            JSON.parse(
+              message.dig("function_call", "arguments"),
+              {symbolize_names: true}
+            )
+          if function_name == "broca"
+            args
+          end
+        end
+      rescue => e
+        LlmMemory.logger.info(e.inspect)
+        nil
+      end
+    end
+
     def generate_prompt(args)
       erb = ERB.new(@prompt)
       erb.result_with_hash(args)

data/lib/llm_memory/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: llm_memory
 version: !ruby/object:Gem::Version
-  version: 0.1.10
+  version: 0.1.11
 platform: ruby
 authors:
 - Shohei Kameda
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-06-02 00:00:00.000000000 Z
+date: 2023-07-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: tokenizers