foobara-llm-backed-command 0.0.7 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2d0fd4526fdfafec7f3f36b2734206b6d5a21e9af138081344b7662bb192e0df
-  data.tar.gz: 471ddc74cdd4782c46c74ae5c872eb698b85a73514474a945968776477303137
+  metadata.gz: dd38cd90e786b6c0da9b5050bc524db798c11b5621201eff0510b5264d00c240
+  data.tar.gz: a1529aaac899adda0831d947e982affc9420bbdcfbce53d12677fd09a6fcab4a
 SHA512:
-  metadata.gz: 45197b581cef44bf69b6b75fe158c045d03670902f008e213db2e76d5d148d2bb152ebf7cda5ebe7163856ea490934d09d27a29798c3da86726cd2ded442aaa9
-  data.tar.gz: bfdcfae305ed54a01e78cc50ffcd6addd3bac578d8201638790d4b7b28b0a27e24f1edd78f2aa72b9586b5ff15ae45f766cf7e8514e596ff9f162fee089eccfb
+  metadata.gz: 38df4ecf874a19a014329d5ecf3787f33fc0b70bc73c727de70220bda27312187bc43e1d1752177a58076c17fbb0f4003375f988f0b39a32117fd5bc91825344
+  data.tar.gz: 1c1ff92542023066c4ebabd4da9b417bfb715494f8ecf12a988072e84b41ddd9f300484eb9302ee47bf27237e320f4dc88a186ddf0a8fbb8e741d99aaaab08ae

data/CHANGELOG.md

@@ -1,3 +1,11 @@
+## [0.0.9] - 2025-06-28
+
+- Use temperature and replace Ask with GenerateNextMessage
+
+## [0.0.8] - 2025-06-19
+
+- Convert result type entities to their primary key types with a description
+
 ## [0.0.7] - 2025-05-30
 
 - Add a could-not-parse-json command error instead of raising

data/src/llm_backed_execute_method.rb

@@ -10,7 +10,7 @@ module Foobara
     include Concern
 
     on_include do
-      depends_on Ai::AnswerBot::Ask
+      depends_on Ai::AnswerBot::GenerateNextMessage
       possible_error :could_not_parse_result_json,
                      message: "Could not parse answer",
                      context: {
@@ -21,14 +21,14 @@ module Foobara
 
     def execute
       determine_serializer
-      construct_input_json
+      construct_messages
       generate_answer
       parse_answer
 
       parsed_answer
     end
 
-    attr_accessor :serializer, :input_json, :answer, :parsed_answer
+    attr_accessor :serializer, :answer, :parsed_answer, :messages
 
     def determine_serializer
       depth = if respond_to?(:association_depth)
@@ -56,24 +56,48 @@ module Foobara
       self.serializer = serializer.new
     end
 
-    def construct_input_json
-      inputs_without_llm_integration_inputs = inputs.except(:llm_model, :association_depth)
-      input_json = serializer.serialize(inputs_without_llm_integration_inputs)
-
-      self.input_json = JSON.fast_generate(input_json)
-    end
-
     def generate_answer
-      ask_inputs = {
-        instructions: llm_instructions,
-        question: input_json
+      inputs = {
+        chat: Ai::AnswerBot::Types::Chat.new(messages:)
       }
 
+      inputs[:temperature] = if respond_to?(:temperature)
+                               temperature
+                             end || 0
+
       if respond_to?(:llm_model)
-        ask_inputs[:model] = llm_model
+        inputs[:model] = llm_model
+      end
+
+      message = run_subcommand!(Ai::AnswerBot::GenerateNextMessage, inputs)
+
+      self.answer = message.content
+    end
+
+    def construct_messages
+      self.messages = build_messages.map do |message|
+        content = message[:content]
+
+        if content.is_a?(String)
+          message
+        else
+          content = serializer.serialize(content)
+          message.merge(content: JSON.fast_generate(content))
+        end
       end
+    end
 
-      self.answer = run_subcommand!(Ai::AnswerBot::Ask, ask_inputs)
+    def build_messages
+      [
+        {
+          role: :system,
+          content: llm_instructions
+        },
+        {
+          role: :user,
+          content: inputs.except(:llm_model, :association_depth)
+        }
+      ]
     end
 
     def llm_instructions
@@ -161,7 +185,10 @@ module Foobara
       end
 
       def result_json_schema
-        @result_json_schema ||= JsonSchemaGenerator.to_json_schema(result_type)
+        @result_json_schema ||= JsonSchemaGenerator.to_json_schema(
+          result_type,
+          association_depth: JsonSchemaGenerator::AssociationDepth::PRIMARY_KEY_ONLY
+        )
       end
 
       def llm_instructions

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: foobara-llm-backed-command
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.9
 platform: ruby
 authors:
 - Miles Georgi
@@ -15,14 +15,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.0.92
+        version: 0.0.132
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.0.92
+        version: 0.0.132
 - !ruby/object:Gem::Dependency
   name: foobara-ai
   requirement: !ruby/object:Gem::Requirement
@@ -43,14 +43,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.0.1
+        version: 0.0.6
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.0.1
+        version: 0.0.6
 email:
 - azimux@gmail.com
 executables: []
@@ -86,7 +86,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.7
+rubygems_version: 3.6.9
 specification_version: 4
 summary: Provides an easy way to implement a command whose logic is managed by an
   LLM