ruby_llm-agents 3.0.0 → 3.2.0

data/lib/ruby_llm/agents/image/generator.rb

@@ -168,25 +168,25 @@ module RubyLLM
 
         def default_image_model
           RubyLLM::Agents.configuration.default_image_model
-        rescue StandardError
+        rescue
           "dall-e-3"
         end
 
         def default_image_size
           RubyLLM::Agents.configuration.default_image_size
-        rescue StandardError
+        rescue
           "1024x1024"
         end
 
         def default_image_quality
           RubyLLM::Agents.configuration.default_image_quality
-        rescue StandardError
+        rescue
           "standard"
         end
 
         def default_image_style
           RubyLLM::Agents.configuration.default_image_style
-        rescue StandardError
+        rescue
           "vivid"
         end
       end
@@ -241,7 +241,7 @@ module RubyLLM
         images = generate_images
 
         execution_completed_at = Time.current
-        duration_ms = ((execution_completed_at - execution_started_at) * 1000).to_i
+        ((execution_completed_at - execution_started_at) * 1000).to_i
 
         # Build result
         result = build_result(
@@ -257,7 +257,7 @@ module RubyLLM
         context.total_cost = result.total_cost
 
         context.output = result
-      rescue StandardError => e
+      rescue => e
         execution_completed_at = Time.current
         context.output = build_error_result(
           e,

data/lib/ruby_llm/agents/image/pipeline/dsl.rb

@@ -76,7 +76,7 @@ module RubyLLM
         #   before_pipeline { |ctx| ctx[:started_at] = Time.current }
         #
         def before_pipeline(method_name = nil, &block)
-          @callbacks ||= { before: [], after: [] }
+          @callbacks ||= {before: [], after: []}
           @callbacks[:before] << (block || method_name)
         end
 
@@ -91,7 +91,7 @@ module RubyLLM
         #   after_pipeline { |result| notify_completion(result) }
         #
         def after_pipeline(method_name = nil, &block)
-          @callbacks ||= { before: [], after: [] }
+          @callbacks ||= {before: [], after: []}
           @callbacks[:after] << (block || method_name)
         end
 
@@ -99,7 +99,7 @@ module RubyLLM
         #
         # @return [Hash] Hash with :before and :after arrays
         def callbacks
-          @callbacks ||= { before: [], after: [] }
+          @callbacks ||= {before: [], after: []}
         end
 
         # Set or get the description
@@ -148,7 +148,7 @@ module RubyLLM
           end
         end
 
-        alias stop_on_error? stop_on_error
+        alias_method :stop_on_error?, :stop_on_error
 
         private
 
@@ -163,11 +163,11 @@ module RubyLLM
           step_keys = config.keys & valid_keys
 
           if step_keys.empty?
-            raise ArgumentError, "Step :#{name} must specify one of: #{valid_keys.join(', ')}"
+            raise ArgumentError, "Step :#{name} must specify one of: #{valid_keys.join(", ")}"
           end
 
           if step_keys.size > 1
-            raise ArgumentError, "Step :#{name} can only specify one step type, got: #{step_keys.join(', ')}"
+            raise ArgumentError, "Step :#{name} can only specify one step type, got: #{step_keys.join(", ")}"
           end
 
           # Validate the class responds to call

data/lib/ruby_llm/agents/image/pipeline/execution.rb

@@ -37,7 +37,7 @@ module RubyLLM
             next unless should_run_step?(step_def)
 
             result = execute_step(step_def, current_image)
-            @step_results << { name: step_def[:name], type: step_def[:type], result: result }
+            @step_results << {name: step_def[:name], type: step_def[:type], result: result}
 
             # Update context with result
             @context[step_def[:name]] = result
@@ -63,7 +63,7 @@ module RubyLLM
           record_execution(result) if execution_tracking_enabled?
 
           result
-        rescue StandardError => e
+        rescue => e
           record_failed_execution(e) if execution_tracking_enabled?
           build_error_result(e)
         end
@@ -218,11 +218,11 @@ module RubyLLM
           return unless tenant
 
           @tenant_id = case tenant
-                       when Hash then tenant[:id]
-                       when Integer, String then tenant
-                       else
-                         tenant.try(:llm_tenant_id) || tenant.try(:id)
-                       end
+          when Hash then tenant[:id]
+          when Integer, String then tenant
+          else
+            tenant.try(:llm_tenant_id) || tenant.try(:id)
+          end
         end
 
         # Budget tracking
@@ -327,7 +327,7 @@ module RubyLLM
           else
             RubyLLM::Agents::Execution.create!(execution_data)
           end
-        rescue StandardError => e
+        rescue => e
           Rails.logger.error("[RubyLLM::Agents] Failed to record pipeline execution: #{e.message}") if defined?(Rails)
         end
 
@@ -359,7 +359,7 @@ module RubyLLM
           else
             RubyLLM::Agents::Execution.create!(execution_data)
           end
-        rescue StandardError => e
+        rescue => e
           Rails.logger.error("[RubyLLM::Agents] Failed to record failed pipeline execution: #{e.message}") if defined?(Rails)
         end
 

data/lib/ruby_llm/agents/image/pipeline.rb

@@ -63,7 +63,7 @@ module RubyLLM
           super
           # Copy steps to subclass
           subclass.instance_variable_set(:@steps, @steps&.dup || [])
-          subclass.instance_variable_set(:@callbacks, @callbacks&.dup || { before: [], after: [] })
+          subclass.instance_variable_set(:@callbacks, @callbacks&.dup || {before: [], after: []})
           subclass.instance_variable_set(:@version, @version)
           subclass.instance_variable_set(:@description, @description)
           subclass.instance_variable_set(:@cache_ttl, @cache_ttl)

data/lib/ruby_llm/agents/image/transformer/execution.rb

@@ -45,7 +45,7 @@ module RubyLLM
           record_execution(result) if execution_tracking_enabled?
 
           result
-        rescue StandardError => e
+        rescue => e
           record_failed_execution(e, started_at) if execution_tracking_enabled?
           build_error_result(e, started_at)
         end

data/lib/ruby_llm/agents/image/upscaler/dsl.rb

@@ -29,7 +29,7 @@ module RubyLLM
         def scale(value = nil)
           if value
             unless VALID_SCALES.include?(value)
-              raise ArgumentError, "Scale must be one of: #{VALID_SCALES.join(', ')}"
+              raise ArgumentError, "Scale must be one of: #{VALID_SCALES.join(", ")}"
             end
             @scale = value
           else

data/lib/ruby_llm/agents/image/upscaler/execution.rb

@@ -45,7 +45,7 @@ module RubyLLM
           record_execution(result) if execution_tracking_enabled?
 
           result
-        rescue StandardError => e
+        rescue => e
           record_failed_execution(e, started_at) if execution_tracking_enabled?
           build_error_result(e, started_at)
         end
@@ -126,7 +126,7 @@ module RubyLLM
         def build_error_result(error, started_at)
           ImageUpscaleResult.new(
             image: nil,
-            source_image: self.image,
+            source_image: image,
             model_id: resolve_model,
             scale: resolve_scale,
             output_size: nil,
@@ -160,10 +160,8 @@ module RubyLLM
           elsif defined?(Vips)
             img = Vips::Image.new_from_file(image)
             [img.width, img.height]
-          else
-            nil
           end
-        rescue StandardError
+        rescue
           nil
         end
 

data/lib/ruby_llm/agents/image/variator/execution.rb

@@ -45,7 +45,7 @@ module RubyLLM
           record_execution(result) if execution_tracking_enabled?
 
           result
-        rescue StandardError => e
+        rescue => e
           record_failed_execution(e, started_at) if execution_tracking_enabled?
           build_error_result(e, started_at)
         end

data/lib/ruby_llm/agents/infrastructure/alert_manager.rb

@@ -43,7 +43,7 @@ module RubyLLM
 
           # Store in cache for dashboard display
           store_for_dashboard(event, full_payload)
-        rescue StandardError => e
+        rescue => e
           Rails.logger.error("[RubyLLM::Agents::AlertManager] Failed: #{e.message}")
         end
 
@@ -72,7 +72,7 @@ module RubyLLM
           return unless handler.respond_to?(:call)
 
           handler.call(event, payload)
-        rescue StandardError => e
+        rescue => e
           Rails.logger.warn("[RubyLLM::Agents::AlertManager] Handler failed: #{e.message}")
         end
 
@@ -83,7 +83,7 @@ module RubyLLM
         # @return [void]
         def emit_notification(event, payload)
           ActiveSupport::Notifications.instrument("ruby_llm_agents.alert.#{event}", payload)
-        rescue StandardError
+        rescue
           # Ignore notification failures
         end
 
@@ -106,7 +106,7 @@ module RubyLLM
           alerts = alerts.first(50)
 
           cache.write(key, alerts, expires_in: 24.hours)
-        rescue StandardError
+        rescue
           # Ignore cache failures
         end
 

data/lib/ruby_llm/agents/infrastructure/attempt_tracker.rb

@@ -235,7 +235,7 @@ module RubyLLM
       def safe_value(response, method, default = nil)
         return default unless response.respond_to?(method)
         response.public_send(method)
-      rescue StandardError
+      rescue
         default
       end
 
@@ -249,7 +249,7 @@ module RubyLLM
           model_id: attempt[:model_id],
           attempt_index: @attempts.length
         )
-      rescue StandardError
+      rescue
         # Ignore notification failures
       end
 
@@ -275,7 +275,7 @@ module RubyLLM
         end
 
         ActiveSupport::Notifications.instrument(event, payload)
-      rescue StandardError
+      rescue
         # Ignore notification failures
       end
 
@@ -288,7 +288,7 @@ module RubyLLM
           "ruby_llm_agents.attempt.short_circuit",
           model_id: attempt[:model_id]
         )
-      rescue StandardError
+      rescue
         # Ignore notification failures
       end
     end

data/lib/ruby_llm/agents/infrastructure/budget/budget_query.rb

@@ -58,9 +58,9 @@ module RubyLLM
           # @return [Float] Total spend in USD
           def current_global_spend(period)
             total = RubyLLM::Agents::Execution
-                      .where("created_at >= ?", period_start(period))
-                      .where(tenant_id: nil)
-                      .sum(:total_cost)
+              .where("created_at >= ?", period_start(period))
+              .where(tenant_id: nil)
+              .sum(:total_cost)
             key = SpendRecorder.budget_cache_key(:global, period)
             BudgetQuery.cache_write(key, total, expires_in: period_ttl(period))
             total
@@ -72,9 +72,9 @@ module RubyLLM
           # @return [Integer] Total tokens used
           def current_global_tokens(period)
             total = RubyLLM::Agents::Execution
-                      .where("created_at >= ?", period_start(period))
-                      .where(tenant_id: nil)
-                      .sum(:total_tokens)
+              .where("created_at >= ?", period_start(period))
+              .where(tenant_id: nil)
+              .sum(:total_tokens)
             key = SpendRecorder.token_cache_key(period)
             BudgetQuery.cache_write(key, total, expires_in: period_ttl(period))
             total
@@ -106,7 +106,7 @@ module RubyLLM
           # @param tenant_id [String, nil] The tenant identifier
           # @param budget_config [Hash] Budget configuration
           # @return [Float, nil] Remaining budget in USD, or nil if no limit configured
-          def remaining_budget(scope, period, agent_type: nil, tenant_id: nil, budget_config:)
+          def remaining_budget(scope, period, budget_config:, agent_type: nil, tenant_id: nil)
             limit = case [scope, period]
             when [:global, :daily]
               budget_config[:global_daily]
@@ -129,7 +129,7 @@ module RubyLLM
           # @param tenant_id [String, nil] The tenant identifier
           # @param budget_config [Hash] Budget configuration
           # @return [Integer, nil] Remaining token budget, or nil if no limit configured
-          def remaining_token_budget(period, tenant_id: nil, budget_config:)
+          def remaining_token_budget(period, budget_config:, tenant_id: nil)
             limit = case period
             when :daily
               budget_config[:global_daily_tokens]
@@ -148,7 +148,7 @@ module RubyLLM
           # @param tenant_id [String, nil] The tenant identifier
           # @param budget_config [Hash] Budget configuration
           # @return [Hash] Budget status information
-          def status(agent_type: nil, tenant_id: nil, budget_config:)
+          def status(budget_config:, agent_type: nil, tenant_id: nil)
             {
               tenant_id: tenant_id,
               enabled: budget_config[:enabled],

data/lib/ruby_llm/agents/infrastructure/budget/config_resolver.rb

@@ -120,7 +120,7 @@ module RubyLLM
             return nil unless tenant_budget_table_exists?
 
             TenantBudget.for_tenant(tenant_id)
-          rescue StandardError => e
+          rescue => e
             Rails.logger.warn("[RubyLLM::Agents] Failed to lookup tenant budget: #{e.message}")
             nil
           end
@@ -133,8 +133,8 @@ module RubyLLM
 
             # Check for new table name (tenants) or old table name (tenant_budgets) for backward compatibility
             @tenant_budget_table_exists = ::ActiveRecord::Base.connection.table_exists?(:ruby_llm_agents_tenants) ||
-                                          ::ActiveRecord::Base.connection.table_exists?(:ruby_llm_agents_tenant_budgets)
-          rescue StandardError
+              ::ActiveRecord::Base.connection.table_exists?(:ruby_llm_agents_tenant_budgets)
+          rescue
             @tenant_budget_table_exists = false
           end
 

data/lib/ruby_llm/agents/infrastructure/budget/forecaster.rb

@@ -13,7 +13,7 @@ module RubyLLM
           # @param tenant_id [String, nil] The tenant identifier
           # @param budget_config [Hash] Budget configuration
           # @return [Hash, nil] Forecast information
-          def calculate_forecast(tenant_id: nil, budget_config:)
+          def calculate_forecast(budget_config:, tenant_id: nil)
             return nil unless budget_config[:enabled]
             return nil unless budget_config[:global_daily] || budget_config[:global_monthly]
 

data/lib/ruby_llm/agents/infrastructure/budget/spend_recorder.rb

@@ -61,7 +61,7 @@ module RubyLLM
           # @return [Float] New total
           def increment_spend(scope, period, amount, agent_type: nil, tenant_id: nil)
             key = budget_cache_key(scope, period, agent_type: agent_type, tenant_id: tenant_id)
-            ttl = period == :daily ? 1.day : 31.days
+            ttl = (period == :daily) ? 1.day : 31.days
 
             # Read-modify-write for float values (cache increment is for integers)
             current = (SpendRecorder.cache_read(key) || 0).to_f
@@ -81,7 +81,7 @@ module RubyLLM
           def increment_tokens(scope, period, tokens, agent_type: nil, tenant_id: nil)
             # For now, we only track global token usage (not per-agent)
             key = token_cache_key(period, tenant_id: tenant_id)
-            ttl = period == :daily ? 1.day : 31.days
+            ttl = (period == :daily) ? 1.day : 31.days
 
             current = (SpendRecorder.cache_read(key) || 0).to_i
             new_total = current + tokens
@@ -102,7 +102,7 @@ module RubyLLM
           # @param period [Symbol] :daily or :monthly
           # @return [String] Date string
           def date_key_part(period)
-            period == :daily ? Date.current.to_s : Date.current.strftime("%Y-%m")
+            (period == :daily) ? Date.current.to_s : Date.current.strftime("%Y-%m")
           end
 
           # Generates an alert cache key
@@ -156,28 +156,28 @@ module RubyLLM
           def check_soft_cap_alerts(agent_type, tenant_id, budget_config)
             # Check global daily
             check_budget_alert(:global_daily, budget_config[:global_daily],
-                              BudgetQuery.current_spend(:global, :daily, tenant_id: tenant_id),
-                              agent_type, tenant_id, budget_config)
+              BudgetQuery.current_spend(:global, :daily, tenant_id: tenant_id),
+              agent_type, tenant_id, budget_config)
 
             # Check global monthly
             check_budget_alert(:global_monthly, budget_config[:global_monthly],
-                              BudgetQuery.current_spend(:global, :monthly, tenant_id: tenant_id),
-                              agent_type, tenant_id, budget_config)
+              BudgetQuery.current_spend(:global, :monthly, tenant_id: tenant_id),
+              agent_type, tenant_id, budget_config)
 
             # Check per-agent daily
             agent_daily_limit = budget_config[:per_agent_daily]&.dig(agent_type)
             if agent_daily_limit
               check_budget_alert(:per_agent_daily, agent_daily_limit,
-                                BudgetQuery.current_spend(:agent, :daily, agent_type: agent_type, tenant_id: tenant_id),
-                                agent_type, tenant_id, budget_config)
+                BudgetQuery.current_spend(:agent, :daily, agent_type: agent_type, tenant_id: tenant_id),
+                agent_type, tenant_id, budget_config)
             end
 
             # Check per-agent monthly
             agent_monthly_limit = budget_config[:per_agent_monthly]&.dig(agent_type)
             if agent_monthly_limit
               check_budget_alert(:per_agent_monthly, agent_monthly_limit,
-                                BudgetQuery.current_spend(:agent, :monthly, agent_type: agent_type, tenant_id: tenant_id),
-                                agent_type, tenant_id, budget_config)
+                BudgetQuery.current_spend(:agent, :monthly, agent_type: agent_type, tenant_id: tenant_id),
+                agent_type, tenant_id, budget_config)
             end
           end
 
@@ -194,7 +194,7 @@ module RubyLLM
             return unless limit
             return if current <= limit
 
-            event = budget_config[:enforcement] == :hard ? :budget_hard_cap : :budget_soft_cap
+            event = (budget_config[:enforcement] == :hard) ? :budget_hard_cap : :budget_soft_cap
 
             # Prevent duplicate alerts by using a cache key (include tenant for isolation)
             key = alert_cache_key("budget_alert", scope, tenant_id)
@@ -221,13 +221,13 @@ module RubyLLM
           def check_soft_token_alerts(agent_type, tenant_id, budget_config)
             # Check global daily tokens
             check_token_alert(:global_daily_tokens, budget_config[:global_daily_tokens],
-                             BudgetQuery.current_tokens(:daily, tenant_id: tenant_id),
-                             agent_type, tenant_id, budget_config)
+              BudgetQuery.current_tokens(:daily, tenant_id: tenant_id),
+              agent_type, tenant_id, budget_config)
 
             # Check global monthly tokens
             check_token_alert(:global_monthly_tokens, budget_config[:global_monthly_tokens],
-                             BudgetQuery.current_tokens(:monthly, tenant_id: tenant_id),
-                             agent_type, tenant_id, budget_config)
+              BudgetQuery.current_tokens(:monthly, tenant_id: tenant_id),
+              agent_type, tenant_id, budget_config)
           end
 
           # Checks if a token alert should be fired
@@ -243,7 +243,7 @@ module RubyLLM
             return unless limit
             return if current <= limit
 
-            event = budget_config[:enforcement] == :hard ? :token_hard_cap : :token_soft_cap
+            event = (budget_config[:enforcement] == :hard) ? :token_hard_cap : :token_soft_cap
 
             # Prevent duplicate alerts
             key = alert_cache_key("token_alert", scope, tenant_id)

data/lib/ruby_llm/agents/infrastructure/circuit_breaker.rb

@@ -30,6 +30,7 @@ module RubyLLM
     # @api public
     class CircuitBreaker
       include CacheHelper
+
       attr_reader :agent_type, :model_id, :tenant_id, :errors_threshold, :window_seconds, :cooldown_seconds
 
       # @param agent_type [String] The agent class name

data/lib/ruby_llm/agents/infrastructure/execution_logger_job.rb

@@ -33,7 +33,7 @@ module RubyLLM
         execution = Execution.create!(filtered_data)
 
         # Create detail record if present
-        if detail_data && detail_data.values.any? { |v| v.present? && v != {} && v != [] }
+        if detail_data&.values&.any? { |v| v.present? && v != {} && v != [] }
           execution.create_detail!(detail_data)
         end
 

data/lib/ruby_llm/agents/infrastructure/reliability.rb

@@ -108,11 +108,11 @@ module RubyLLM
 
           @attempts.filter_map do |attempt|
             if attempt["short_circuited"]
-              { model: attempt["model_id"], error_class: "CircuitBreakerOpen", error_message: "circuit breaker open",
-                error_backtrace: nil }
+              {model: attempt["model_id"], error_class: "CircuitBreakerOpen", error_message: "circuit breaker open",
+               error_backtrace: nil}
             elsif attempt["error_class"]
-              { model: attempt["model_id"], error_class: attempt["error_class"], error_message: attempt["error_message"],
-                error_backtrace: attempt["error_backtrace"] }
+              {model: attempt["model_id"], error_class: attempt["error_class"], error_message: attempt["error_message"],
+               error_backtrace: attempt["error_backtrace"]}
             end
           end
         end
@@ -125,11 +125,11 @@ module RubyLLM
               if attempt["short_circuited"]
                 parts << "  #{model}: circuit breaker open"
               elsif attempt["error_class"]
-                parts << "  #{model}: #{attempt['error_class']} - #{attempt['error_message']}"
+                parts << "  #{model}: #{attempt["error_class"]} - #{attempt["error_message"]}"
               end
             end
           else
-            parts << "  Models: #{@models_tried.join(', ')}"
+            parts << "  Models: #{@models_tried.join(", ")}"
             parts << "  Last error: #{@last_error.message}"
           end
           parts.join("\n")

data/lib/ruby_llm/agents/pipeline/builder.rb

@@ -164,7 +164,7 @@ module RubyLLM
           # @return [Boolean]
           def budgets_enabled?
             RubyLLM::Agents.configuration.budgets_enabled?
-          rescue StandardError
+          rescue
             false
           end
 
@@ -176,7 +176,7 @@ module RubyLLM
             return false unless agent_class
 
             agent_class.respond_to?(:cache_enabled?) && agent_class.cache_enabled?
-          rescue StandardError
+          rescue
             false
           end
 
@@ -192,20 +192,20 @@ module RubyLLM
             return false unless agent_class
 
             retries = if agent_class.respond_to?(:retries)
-                        agent_class.retries
-                      else
-                        0
-                      end
+              agent_class.retries
+            else
+              0
+            end
 
             fallbacks = if agent_class.respond_to?(:fallback_models)
-                          agent_class.fallback_models
-                        else
-                          []
-                        end
+              agent_class.fallback_models
+            else
+              []
+            end
 
             (retries.is_a?(Integer) && retries.positive?) ||
               (fallbacks.is_a?(Array) && fallbacks.any?)
-          rescue StandardError
+          rescue
             false
           end
         end

data/lib/ruby_llm/agents/pipeline/middleware/budget.rb

@@ -51,7 +51,7 @@ module RubyLLM
           # @return [Boolean]
           def budgets_enabled?
             global_config.budgets_enabled?
-          rescue StandardError
+          rescue
             false
           end
 
@@ -78,7 +78,7 @@ module RubyLLM
             )
           rescue RubyLLM::Agents::Reliability::BudgetExceededError
             raise
-          rescue StandardError => e
+          rescue => e
             error("Budget check failed: #{e.message}")
           end
 
@@ -117,7 +117,7 @@ module RubyLLM
                 tenant_id: context.tenant_id
               )
             end
-          rescue StandardError => e
+          rescue => e
             error("Failed to record spend: #{e.message}")
           end
         end

data/lib/ruby_llm/agents/pipeline/middleware/cache.rb

@@ -71,7 +71,7 @@ module RubyLLM
           # @return [ActiveSupport::Cache::Store, nil]
           def cache_store
             global_config.cache_store
-          rescue StandardError
+          rescue
             nil
           end
 
@@ -130,7 +130,7 @@ module RubyLLM
             else
               input.to_json
             end
-          rescue StandardError
+          rescue
             input.to_s
           end
 
@@ -140,7 +140,7 @@ module RubyLLM
           # @return [Object, nil] Cached value or nil
           def cache_read(key)
             cache_store.read(key)
-          rescue StandardError => e
+          rescue => e
             error("Cache read failed: #{e.message}")
             nil
           end
@@ -154,7 +154,7 @@ module RubyLLM
             options[:expires_in] = cache_ttl if cache_ttl
 
             cache_store.write(key, value, **options)
-          rescue StandardError => e
+          rescue => e
             error("Cache write failed: #{e.message}")
           end
         end