foobara-agent-backed-command 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 282e1d35799a4c8226d1d381689f21a09450e4533c0ec3cdd1a52daabe2565e1
-  data.tar.gz: d1eeb6e39f5518f5e9800d8e9c010cf5e4a3607aa533f29c9a2949fa52da9eb5
+  metadata.gz: 3f5c7e403d3856677c7e1a8b7d047eaa2632e7d25d77e805ed28e6d59ecfee0f
+  data.tar.gz: 465f51347fa4ea0c28ae0c692716dbe0bfc6fcf993e9b619b414dd8d8d546f79
 SHA512:
-  metadata.gz: f8a4237ef3f621ab8c854a9f9f2f89f3021ccf6ea4679dcbbeba59237e55572e5d1c7221a6c483fd80ddd2556a4534572fbc69d603b25beb51c5f1587d2433e5
-  data.tar.gz: dc5baf5bd4cd96fe08ad0294b45561c954e8cadef331e98b86e426d91986da4a637c1108777d8784e214de7492a6ad4f5ada7295d6cd6d5bd0f801d066efa7df
+  metadata.gz: e83aadf2ce5940b7b51c601b273319cd53d112a44ba673842b40a76cf566aabf29963b4218c56d5287f94896d764711d2af231c41423b9f68fd65a7b2f60c0bd
+  data.tar.gz: d9804222c4a0832d5e747ec0e663bb0caf21668adbf032310fc498db7ec65f1f14b8c9f18082d0cdaf1382249ef4cd304fea3a8a4db73b8bdd80a64aaa8e35d0

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## [0.0.2] - 2025-07-08
+
+- Better handle serialization/pre-commit loading for different inputs/result scenarios
+
 ## [0.0.1] - 2025-07-03
 
 - Initial release

data/src/agent_backed_command.rb

@@ -9,7 +9,8 @@ module Foobara
 
     class << self
       # TODO: does this need to be a Concern for proper inheritance?
-      attr_accessor :is_verbose, :io_out, :io_err, :context, :agent_name, :llm_model, :max_llm_calls_per_minute
+      attr_accessor :is_verbose, :io_out, :io_err, :context, :agent_name, :llm_model, :max_llm_calls_per_minute,
+                    :pass_aggregates_to_llm, :result_entity_depth
 
       def verbose(value = true)
         self.is_verbose = value
@@ -22,7 +23,7 @@ module Foobara
 
     inputs do
       agent_options do
-        verbose :boolean
+        verbose :boolean, :allow_nil
         io_out :duck
         io_err :duck
         context Foobara::Agent::Context, :allow_nil, "The current context of the agent"
@@ -32,6 +33,8 @@ module Foobara
                   one_of: Foobara::Ai::AnswerBot::Types::ModelEnum,
                   description: "The model to use for the LLM"
         max_llm_calls_per_minute :integer, :allow_nil
+        pass_aggregates_to_llm :boolean, :allow_nil
+        result_entity_depth :symbol, :allow_nil, one_of: Foobara::AssociationDepth
       end
     end
 
@@ -97,7 +100,7 @@ module Foobara
       verbose = agent_options&.[](:verbose)
       verbose = self.class.verbose? if verbose.nil?
 
-      self.agent = Foobara::Agent.new(
+      opts = {
         command_classes:,
         include_message_to_user_in_result:,
         result_type: agent_result_type,
@@ -107,8 +110,35 @@ module Foobara
         agent_name:,
         context: agent_options&.[](:context) || self.class.context,
         llm_model: agent_options&.[](:llm_model) || self.class.llm_model,
+        # TODO: eliminate this now that we have backoffs for 529s and 429s
         max_llm_calls_per_minute: agent_options&.[](:max_llm_calls_per_minute) || self.class.max_llm_calls_per_minute
-      )
+      }
+
+      if agent_options&.[](:result_entity_depth).nil?
+        unless self.class.result_entity_depth.nil?
+          opts[:result_entity_depth] = self.class.result_entity_depth
+        end
+      else
+        opts[:result_entity_depth] = agent_options[:result_entity_depth]
+      end
+
+      if agent_options&.[](:pass_aggregates_to_llm).nil?
+        unless self.class.pass_aggregates_to_llm.nil?
+          opts[:pass_aggregates_to_llm] = self.class.pass_aggregates_to_llm
+        end
+      else
+        opts[:pass_aggregates_to_llm] = agent_options[:pass_aggregates_to_llm]
+      end
+
+      self.agent = Foobara::Agent.new(**opts)
+    end
+
+    def pass_aggregates_to_llm?
+      if agent_options&.[](:pass_aggregates_to_llm).nil?
+        self.class.pass_aggregates_to_llm
+      else
+        agent_options[:pass_aggregates_to_llm]
+      end
     end
 
     def construct_goal_if_needed
@@ -132,13 +162,24 @@ module Foobara
             inputs_type = TypeDeclarations::Attributes.reject(inputs_type.declaration_data, :agent_options)
             inputs_type = domain.foobara_type_from_declaration(inputs_type)
 
+            association_depth = if pass_aggregates_to_llm?
+                                  AssociationDepth::AGGREGATE
+                                else
+                                  AssociationDepth::ATOM
+                                end
+
             json_inputs_type = JsonSchemaGenerator.to_json_schema(
               inputs_type,
-              association_depth: JsonSchemaGenerator::AssociationDepth::ATOM
+              association_depth:
             )
             goal += " The inputs to this command have the following type:\n\n#{json_inputs_type}\n\n"
 
-            serializer = Foobara::CommandConnectors::Serializers::AtomicSerializer.new
+            serializer = if pass_aggregates_to_llm?
+                           CommandConnectors::Serializers::AggregateSerializer
+                         else
+                           CommandConnectors::Serializers::AtomicSerializer
+                         end.new
+
             inputs_json = serializer.serialize(inputs.except(:agent_options))
             inputs_json = JSON.fast_generate(inputs_json)
             goal += "You have been ran with the following inputs:\n\n#{inputs_json}"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: foobara-agent-backed-command
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Miles Georgi