ruby_llm-agents 3.1.0 → 3.3.0

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

@@ -168,7 +168,7 @@ module RubyLLM
         def fallback_provider(provider = nil, **options)
           if provider
             @fallback_providers ||= []
-            @fallback_providers << { provider: provider, **options }
+            @fallback_providers << {provider: provider, **options}
           end
           @fallback_providers || inherited_fallback_providers || []
         end
@@ -201,7 +201,7 @@ module RubyLLM
         #   circuit_breaker errors: 10, within: 60, cooldown: 300
         def circuit_breaker(errors: nil, within: nil, cooldown: nil)
           if errors || within || cooldown
-            @circuit_breaker_config ||= { errors: 10, within: 60, cooldown: 300 }
+            @circuit_breaker_config ||= {errors: 10, within: 60, cooldown: 300}
             @circuit_breaker_config[:errors] = errors if errors
             @circuit_breaker_config[:within] = within if within
             @circuit_breaker_config[:cooldown] = cooldown if cooldown
@@ -297,8 +297,8 @@ module RubyLLM
         # Inner builder class for block-style configuration
         class ReliabilityBuilder
           attr_reader :retries_config, :fallback_models_list, :total_timeout_value,
-                      :circuit_breaker_config, :retryable_patterns_list, :fallback_providers_list,
-                      :non_fallback_errors_list
+            :circuit_breaker_config, :retryable_patterns_list, :fallback_providers_list,
+            :non_fallback_errors_list
 
           def initialize
             @retries_config = nil
@@ -332,7 +332,7 @@ module RubyLLM
           #   fallback_provider :openai, voice: "nova"
           #   fallback_provider :elevenlabs, voice: "Rachel", model: "eleven_multilingual_v2"
           def fallback_provider(provider, **options)
-            @fallback_providers_list << { provider: provider, **options }
+            @fallback_providers_list << {provider: provider, **options}
           end
 
           def total_timeout(seconds)
@@ -366,8 +366,8 @@ module RubyLLM
         #
         class OnFailureBuilder
           attr_reader :retries_config, :fallback_models_list, :total_timeout_value,
-                      :circuit_breaker_config, :retryable_patterns_list, :fallback_providers_list,
-                      :non_fallback_errors_list
+            :circuit_breaker_config, :retryable_patterns_list, :fallback_providers_list,
+            :non_fallback_errors_list
 
           def initialize
             @retries_config = nil
@@ -423,7 +423,7 @@ module RubyLLM
           # @param options [Hash] Provider-specific options
           #
           def fallback_provider(provider, **options)
-            @fallback_providers_list << { provider: provider, **options }
+            @fallback_providers_list << {provider: provider, **options}
           end
 
           # Configure timeout for all retry/fallback attempts
@@ -440,7 +440,7 @@ module RubyLLM
           end
 
           # Also support total_timeout for compatibility
-          alias total_timeout timeout
+          alias_method :total_timeout, :timeout
 
           # Configure circuit breaker
           #

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

@@ -31,7 +31,7 @@ module RubyLLM
         def analysis_type(value = nil)
           if value
             unless VALID_ANALYSIS_TYPES.include?(value)
-              raise ArgumentError, "Analysis type must be one of: #{VALID_ANALYSIS_TYPES.join(', ')}"
+              raise ArgumentError, "Analysis type must be one of: #{VALID_ANALYSIS_TYPES.join(", ")}"
             end
             @analysis_type = value
           else

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

@@ -46,7 +46,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
@@ -138,8 +138,8 @@ module RubyLLM
             caption: "Brief caption string",
             description: "Detailed description string (if applicable)",
             tags: ["array", "of", "tag", "strings"],
-            objects: [{ name: "object name", location: "position", confidence: "high/medium/low" }],
-            colors: [{ hex: "#RRGGBB", name: "color name", percentage: 25 }],
+            objects: [{name: "object name", location: "position", confidence: "high/medium/low"}],
+            colors: [{hex: "#RRGGBB", name: "color name", percentage: 25}],
             text: "Extracted text if any"
           }
 
@@ -154,7 +154,7 @@ module RubyLLM
 
           # Use RubyLLM chat with vision
           chat = RubyLLM.chat(model: model)
-          chat.ask(prompt, with: { image: image_content })
+          chat.ask(prompt, with: {image: image_content})
         end
 
         def prepare_image_content

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

@@ -30,7 +30,7 @@ module RubyLLM
         def output_format(value = nil)
           if value
             unless VALID_OUTPUT_FORMATS.include?(value)
-              raise ArgumentError, "Output format must be one of: #{VALID_OUTPUT_FORMATS.join(', ')}"
+              raise ArgumentError, "Output format must be one of: #{VALID_OUTPUT_FORMATS.join(", ")}"
             end
             @output_format = value
           else

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

@@ -46,7 +46,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
@@ -112,13 +112,13 @@ module RubyLLM
                 mode: "segmentation_mask",
                 **build_removal_options
               )
-            rescue StandardError
+            rescue
               # Mask generation failed, continue without it
               mask = nil
             end
           end
 
-          { foreground: foreground, mask: mask }
+          {foreground: foreground, mask: mask}
         end
 
         def build_removal_options

data/lib/ruby_llm/agents/image/concerns/image_operation_execution.rb

@@ -20,11 +20,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
 
         # Check budget before execution
@@ -87,7 +87,7 @@ module RubyLLM
           else
             RubyLLM::Agents::Execution.create!(execution_data)
           end
-        rescue StandardError => e
+        rescue => e
           Rails.logger.error("[RubyLLM::Agents] Failed to record #{execution_type} execution: #{e.message}") if defined?(Rails)
         end
 
@@ -101,7 +101,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 #{execution_type} execution: #{e.message}") if defined?(Rails)
         end
 
@@ -142,7 +142,7 @@ module RubyLLM
         end
 
         def build_metadata(result)
-          { count: result.count }
+          {count: result.count}
         end
 
         def budget_tracking_enabled?

data/lib/ruby_llm/agents/image/editor/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/generator/pricing.rb

@@ -158,7 +158,7 @@ module RubyLLM
           else
             fetch_from_url
           end
-        rescue StandardError => e
+        rescue => e
           warn "[RubyLLM::Agents] Failed to fetch LiteLLM pricing: #{e.message}"
           {}
         end
@@ -178,7 +178,7 @@ module RubyLLM
           else
             {}
           end
-        rescue StandardError => e
+        rescue => e
           warn "[RubyLLM::Agents] HTTP error fetching LiteLLM pricing: #{e.message}"
           {}
         end
@@ -314,9 +314,9 @@ module RubyLLM
 
         def fallback_pricing_table
           {
-            "gpt-image-1" => { standard: 0.04, hd: 0.08, large_hd: 0.12 },
-            "dall-e-3" => { standard: 0.04, hd: 0.08, large_hd: 0.12 },
-            "dall-e-2" => { "1024x1024" => 0.02, "512x512" => 0.018, "256x256" => 0.016 },
+            "gpt-image-1" => {standard: 0.04, hd: 0.08, large_hd: 0.12},
+            "dall-e-3" => {standard: 0.04, hd: 0.08, large_hd: 0.12},
+            "dall-e-2" => {"1024x1024" => 0.02, "512x512" => 0.018, "256x256" => 0.016},
             "flux-pro" => 0.05,
             "flux-dev" => 0.025,
             "flux-schnell" => 0.003,
@@ -332,16 +332,15 @@ module RubyLLM
           width, height = size.to_s.split("x").map(&:to_i)
           return nil if width.zero? || height.zero?
           width * height
-        rescue StandardError
+        rescue
           nil
         end
 
         def normalize_model_id(model_id)
           model_id.to_s
-                  .downcase
-                  .gsub(/[^a-z0-9.-]/, "-")
-                  .gsub(/-+/, "-")
-                  .gsub(/^-|-$/, "")
+            .downcase
+            .gsub(/[^a-z0-9.-]/, "-").squeeze("-")
+            .gsub(/^-|-$/, "")
         end
 
         def config

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]