ruby_llm-agents 0.4.0 → 1.0.0.beta.1

data/lib/ruby_llm/agents.rb

@@ -3,23 +3,92 @@
 require "csv"
 require "ruby_llm"
 
-require_relative "agents/version"
-require_relative "agents/configuration"
-require_relative "agents/deprecations"
-require_relative "agents/reliability"
-require_relative "agents/reliability/retry_strategy"
-require_relative "agents/reliability/fallback_routing"
-require_relative "agents/reliability/breaker_manager"
-require_relative "agents/reliability/execution_constraints"
-require_relative "agents/reliability/executor"
-require_relative "agents/redactor"
-require_relative "agents/circuit_breaker"
-require_relative "agents/budget_tracker"
-require_relative "agents/alert_manager"
-require_relative "agents/attempt_tracker"
-require_relative "agents/result"
-require_relative "agents/inflections" if defined?(Rails)
-require_relative "agents/engine" if defined?(Rails::Engine)
+# Core
+require_relative "agents/core/version"
+require_relative "agents/core/configuration"
+require_relative "agents/core/deprecations"
+require_relative "agents/core/errors"
+require_relative "agents/core/resolved_config"
+require_relative "agents/core/llm_tenant"
+
+# Infrastructure - Reliability
+require_relative "agents/infrastructure/reliability"
+require_relative "agents/infrastructure/reliability/retry_strategy"
+require_relative "agents/infrastructure/reliability/fallback_routing"
+require_relative "agents/infrastructure/reliability/breaker_manager"
+require_relative "agents/infrastructure/reliability/execution_constraints"
+require_relative "agents/infrastructure/reliability/executor"
+
+# Pipeline infrastructure (middleware-based execution)
+require_relative "agents/pipeline"
+
+# DSL modules for agent configuration
+require_relative "agents/dsl"
+
+# BaseAgent - new middleware-based agent architecture
+require_relative "agents/base_agent"
+
+# Infrastructure - Budget & Utilities
+require_relative "agents/infrastructure/redactor"
+require_relative "agents/infrastructure/circuit_breaker"
+require_relative "agents/infrastructure/budget_tracker"
+require_relative "agents/infrastructure/alert_manager"
+require_relative "agents/infrastructure/attempt_tracker"
+require_relative "agents/infrastructure/cache_helper"
+require_relative "agents/infrastructure/budget/budget_query"
+require_relative "agents/infrastructure/budget/config_resolver"
+require_relative "agents/infrastructure/budget/forecaster"
+require_relative "agents/infrastructure/budget/spend_recorder"
+
+# Results
+require_relative "agents/results/base"
+require_relative "agents/results/embedding_result"
+require_relative "agents/results/moderation_result"
+require_relative "agents/results/transcription_result"
+require_relative "agents/results/speech_result"
+require_relative "agents/results/image_generation_result"
+require_relative "agents/results/image_variation_result"
+require_relative "agents/results/image_edit_result"
+require_relative "agents/results/image_transform_result"
+require_relative "agents/results/image_upscale_result"
+require_relative "agents/results/image_analysis_result"
+require_relative "agents/results/background_removal_result"
+require_relative "agents/results/image_pipeline_result"
+
+# Image concerns (shared DSL/execution for image operations)
+require_relative "agents/image/concerns/image_operation_dsl"
+require_relative "agents/image/concerns/image_operation_execution"
+
+# Text agents
+require_relative "agents/text/embedder"
+require_relative "agents/text/moderator"
+
+# Audio agents
+require_relative "agents/audio/transcriber"
+require_relative "agents/audio/speaker"
+
+# Image agents
+require_relative "agents/image/generator"
+require_relative "agents/image/variator"
+require_relative "agents/image/editor"
+require_relative "agents/image/transformer"
+require_relative "agents/image/upscaler"
+require_relative "agents/image/analyzer"
+require_relative "agents/image/background_remover"
+require_relative "agents/image/pipeline"
+
+# Workflow
+require_relative "agents/workflow/async"
+require_relative "agents/workflow/orchestrator"
+require_relative "agents/workflow/async_executor"
+
+# Rails integration
+if defined?(Rails)
+  require_relative "agents/core/inflections"
+  require_relative "agents/core/instrumentation"
+  require_relative "agents/infrastructure/execution_logger_job"
+end
+require_relative "agents/rails/engine" if defined?(Rails::Engine)
 
 module RubyLLM
   # Agent framework for building LLM-powered agents with observability
@@ -49,9 +118,6 @@ module RubyLLM
   # @see RubyLLM::Agents::Base
   # @see RubyLLM::Agents::Configuration
   module Agents
-    # Base error class for agent-related exceptions
-    class Error < StandardError; end
-
     class << self
       # Returns the global configuration instance
       #

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm-agents
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 1.0.0.beta.1
 platform: ruby
 authors:
 - adham90
@@ -29,14 +29,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: 1.11.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: 1.11.0
 - !ruby/object:Gem::Dependency
   name: csv
   requirement: !ruby/object:Gem::Requirement
@@ -51,6 +51,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: ostruct
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: A Rails engine for creating, managing, and monitoring LLM-powered agents.
   Includes a DSL for agent configuration, execution tracking, cost analytics, and
   a mountable dashboard UI.
@@ -65,10 +79,14 @@ files:
 - app/controllers/concerns/ruby_llm/agents/filterable.rb
 - app/controllers/concerns/ruby_llm/agents/paginatable.rb
 - app/controllers/ruby_llm/agents/agents_controller.rb
+- app/controllers/ruby_llm/agents/api_configurations_controller.rb
 - app/controllers/ruby_llm/agents/dashboard_controller.rb
 - app/controllers/ruby_llm/agents/executions_controller.rb
-- app/controllers/ruby_llm/agents/settings_controller.rb
+- app/controllers/ruby_llm/agents/system_config_controller.rb
+- app/controllers/ruby_llm/agents/tenants_controller.rb
+- app/controllers/ruby_llm/agents/workflows_controller.rb
 - app/helpers/ruby_llm/agents/application_helper.rb
+- app/models/ruby_llm/agents/api_configuration.rb
 - app/models/ruby_llm/agents/execution.rb
 - app/models/ruby_llm/agents/execution/analytics.rb
 - app/models/ruby_llm/agents/execution/metrics.rb
@@ -78,11 +96,23 @@ files:
 - app/services/ruby_llm/agents/agent_registry.rb
 - app/views/layouts/ruby_llm/agents/application.html.erb
 - app/views/ruby_llm/agents/agents/_agent.html.erb
+- app/views/ruby_llm/agents/agents/_config_agent.html.erb
+- app/views/ruby_llm/agents/agents/_config_embedder.html.erb
+- app/views/ruby_llm/agents/agents/_config_image_generator.html.erb
+- app/views/ruby_llm/agents/agents/_config_moderator.html.erb
+- app/views/ruby_llm/agents/agents/_config_speaker.html.erb
+- app/views/ruby_llm/agents/agents/_config_transcriber.html.erb
 - app/views/ruby_llm/agents/agents/_empty_state.html.erb
 - app/views/ruby_llm/agents/agents/_version_comparison.html.erb
 - app/views/ruby_llm/agents/agents/_workflow.html.erb
 - app/views/ruby_llm/agents/agents/index.html.erb
 - app/views/ruby_llm/agents/agents/show.html.erb
+- app/views/ruby_llm/agents/api_configurations/_api_key_field.html.erb
+- app/views/ruby_llm/agents/api_configurations/_form.html.erb
+- app/views/ruby_llm/agents/api_configurations/edit.html.erb
+- app/views/ruby_llm/agents/api_configurations/edit_tenant.html.erb
+- app/views/ruby_llm/agents/api_configurations/show.html.erb
+- app/views/ruby_llm/agents/api_configurations/tenant.html.erb
 - app/views/ruby_llm/agents/dashboard/_action_center.html.erb
 - app/views/ruby_llm/agents/dashboard/_agent_comparison.html.erb
 - app/views/ruby_llm/agents/dashboard/_alerts_feed.html.erb
@@ -100,7 +130,7 @@ files:
 - app/views/ruby_llm/agents/executions/dry_run.html.erb
 - app/views/ruby_llm/agents/executions/index.html.erb
 - app/views/ruby_llm/agents/executions/show.html.erb
-- app/views/ruby_llm/agents/settings/show.html.erb
+- app/views/ruby_llm/agents/shared/_agent_type_badge.html.erb
 - app/views/ruby_llm/agents/shared/_breadcrumbs.html.erb
 - app/views/ruby_llm/agents/shared/_executions_table.html.erb
 - app/views/ruby_llm/agents/shared/_filter_dropdown.html.erb
@@ -111,12 +141,35 @@ files:
 - app/views/ruby_llm/agents/shared/_status_dot.html.erb
 - app/views/ruby_llm/agents/shared/_tenant_filter.html.erb
 - app/views/ruby_llm/agents/shared/_workflow_type_badge.html.erb
+- app/views/ruby_llm/agents/system_config/show.html.erb
+- app/views/ruby_llm/agents/tenants/_form.html.erb
+- app/views/ruby_llm/agents/tenants/edit.html.erb
+- app/views/ruby_llm/agents/tenants/index.html.erb
+- app/views/ruby_llm/agents/tenants/show.html.erb
+- app/views/ruby_llm/agents/workflows/_step_performance.html.erb
+- app/views/ruby_llm/agents/workflows/_structure_parallel.html.erb
+- app/views/ruby_llm/agents/workflows/_structure_pipeline.html.erb
+- app/views/ruby_llm/agents/workflows/_structure_router.html.erb
+- app/views/ruby_llm/agents/workflows/show.html.erb
 - config/routes.rb
 - lib/generators/ruby_llm_agents/agent_generator.rb
+- lib/generators/ruby_llm_agents/api_configuration_generator.rb
+- lib/generators/ruby_llm_agents/background_remover_generator.rb
+- lib/generators/ruby_llm_agents/embedder_generator.rb
+- lib/generators/ruby_llm_agents/image_analyzer_generator.rb
+- lib/generators/ruby_llm_agents/image_editor_generator.rb
+- lib/generators/ruby_llm_agents/image_generator_generator.rb
+- lib/generators/ruby_llm_agents/image_pipeline_generator.rb
+- lib/generators/ruby_llm_agents/image_transformer_generator.rb
+- lib/generators/ruby_llm_agents/image_upscaler_generator.rb
+- lib/generators/ruby_llm_agents/image_variator_generator.rb
 - lib/generators/ruby_llm_agents/install_generator.rb
 - lib/generators/ruby_llm_agents/multi_tenancy_generator.rb
+- lib/generators/ruby_llm_agents/restructure_generator.rb
+- lib/generators/ruby_llm_agents/speaker_generator.rb
 - lib/generators/ruby_llm_agents/templates/add_attempts_migration.rb.tt
 - lib/generators/ruby_llm_agents/templates/add_caching_migration.rb.tt
+- lib/generators/ruby_llm_agents/templates/add_execution_type_migration.rb.tt
 - lib/generators/ruby_llm_agents/templates/add_finish_reason_migration.rb.tt
 - lib/generators/ruby_llm_agents/templates/add_prompts_migration.rb.tt
 - lib/generators/ruby_llm_agents/templates/add_routing_migration.rb.tt
@@ -127,47 +180,148 @@ files:
 - lib/generators/ruby_llm_agents/templates/add_workflow_migration.rb.tt
 - lib/generators/ruby_llm_agents/templates/agent.rb.tt
 - lib/generators/ruby_llm_agents/templates/application_agent.rb.tt
+- lib/generators/ruby_llm_agents/templates/application_background_remover.rb.tt
+- lib/generators/ruby_llm_agents/templates/application_embedder.rb.tt
+- lib/generators/ruby_llm_agents/templates/application_image_analyzer.rb.tt
+- lib/generators/ruby_llm_agents/templates/application_image_editor.rb.tt
+- lib/generators/ruby_llm_agents/templates/application_image_generator.rb.tt
+- lib/generators/ruby_llm_agents/templates/application_image_pipeline.rb.tt
+- lib/generators/ruby_llm_agents/templates/application_image_transformer.rb.tt
+- lib/generators/ruby_llm_agents/templates/application_image_upscaler.rb.tt
+- lib/generators/ruby_llm_agents/templates/application_image_variator.rb.tt
+- lib/generators/ruby_llm_agents/templates/application_speaker.rb.tt
+- lib/generators/ruby_llm_agents/templates/application_transcriber.rb.tt
+- lib/generators/ruby_llm_agents/templates/background_remover.rb.tt
+- lib/generators/ruby_llm_agents/templates/create_api_configurations_migration.rb.tt
 - lib/generators/ruby_llm_agents/templates/create_tenant_budgets_migration.rb.tt
+- lib/generators/ruby_llm_agents/templates/embedder.rb.tt
+- lib/generators/ruby_llm_agents/templates/image_analyzer.rb.tt
+- lib/generators/ruby_llm_agents/templates/image_editor.rb.tt
+- lib/generators/ruby_llm_agents/templates/image_generator.rb.tt
+- lib/generators/ruby_llm_agents/templates/image_pipeline.rb.tt
+- lib/generators/ruby_llm_agents/templates/image_transformer.rb.tt
+- lib/generators/ruby_llm_agents/templates/image_upscaler.rb.tt
+- lib/generators/ruby_llm_agents/templates/image_variator.rb.tt
 - lib/generators/ruby_llm_agents/templates/initializer.rb.tt
 - lib/generators/ruby_llm_agents/templates/migration.rb.tt
+- lib/generators/ruby_llm_agents/templates/skills/AGENTS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/BACKGROUND_REMOVERS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/EMBEDDERS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/IMAGE_ANALYZERS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/IMAGE_EDITORS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/IMAGE_GENERATORS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/IMAGE_PIPELINES.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/IMAGE_TRANSFORMERS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/IMAGE_UPSCALERS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/IMAGE_VARIATORS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/SPEAKERS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/TOOLS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/TRANSCRIBERS.md.tt
+- lib/generators/ruby_llm_agents/templates/skills/WORKFLOWS.md.tt
+- lib/generators/ruby_llm_agents/templates/speaker.rb.tt
+- lib/generators/ruby_llm_agents/templates/transcriber.rb.tt
+- lib/generators/ruby_llm_agents/transcriber_generator.rb
 - lib/generators/ruby_llm_agents/upgrade_generator.rb
 - lib/ruby_llm-agents.rb
 - lib/ruby_llm/agents.rb
-- lib/ruby_llm/agents/alert_manager.rb
-- lib/ruby_llm/agents/attempt_tracker.rb
-- lib/ruby_llm/agents/base.rb
-- lib/ruby_llm/agents/base/caching.rb
-- lib/ruby_llm/agents/base/cost_calculation.rb
-- lib/ruby_llm/agents/base/dsl.rb
-- lib/ruby_llm/agents/base/execution.rb
-- lib/ruby_llm/agents/base/reliability_dsl.rb
-- lib/ruby_llm/agents/base/reliability_execution.rb
-- lib/ruby_llm/agents/base/response_building.rb
-- lib/ruby_llm/agents/base/tool_tracking.rb
-- lib/ruby_llm/agents/budget_tracker.rb
-- lib/ruby_llm/agents/cache_helper.rb
-- lib/ruby_llm/agents/circuit_breaker.rb
-- lib/ruby_llm/agents/configuration.rb
-- lib/ruby_llm/agents/deprecations.rb
-- lib/ruby_llm/agents/engine.rb
-- lib/ruby_llm/agents/execution_logger_job.rb
-- lib/ruby_llm/agents/inflections.rb
-- lib/ruby_llm/agents/instrumentation.rb
-- lib/ruby_llm/agents/redactor.rb
-- lib/ruby_llm/agents/reliability.rb
-- lib/ruby_llm/agents/reliability/breaker_manager.rb
-- lib/ruby_llm/agents/reliability/execution_constraints.rb
-- lib/ruby_llm/agents/reliability/executor.rb
-- lib/ruby_llm/agents/reliability/fallback_routing.rb
-- lib/ruby_llm/agents/reliability/retry_strategy.rb
-- lib/ruby_llm/agents/result.rb
-- lib/ruby_llm/agents/version.rb
-- lib/ruby_llm/agents/workflow.rb
+- lib/ruby_llm/agents/audio/speaker.rb
+- lib/ruby_llm/agents/audio/transcriber.rb
+- lib/ruby_llm/agents/base_agent.rb
+- lib/ruby_llm/agents/core/base.rb
+- lib/ruby_llm/agents/core/base/moderation_dsl.rb
+- lib/ruby_llm/agents/core/base/moderation_execution.rb
+- lib/ruby_llm/agents/core/configuration.rb
+- lib/ruby_llm/agents/core/deprecations.rb
+- lib/ruby_llm/agents/core/errors.rb
+- lib/ruby_llm/agents/core/inflections.rb
+- lib/ruby_llm/agents/core/instrumentation.rb
+- lib/ruby_llm/agents/core/llm_tenant.rb
+- lib/ruby_llm/agents/core/resolved_config.rb
+- lib/ruby_llm/agents/core/version.rb
+- lib/ruby_llm/agents/dsl.rb
+- lib/ruby_llm/agents/dsl/base.rb
+- lib/ruby_llm/agents/dsl/caching.rb
+- lib/ruby_llm/agents/dsl/reliability.rb
+- lib/ruby_llm/agents/image/analyzer.rb
+- lib/ruby_llm/agents/image/analyzer/dsl.rb
+- lib/ruby_llm/agents/image/analyzer/execution.rb
+- lib/ruby_llm/agents/image/background_remover.rb
+- lib/ruby_llm/agents/image/background_remover/dsl.rb
+- lib/ruby_llm/agents/image/background_remover/execution.rb
+- lib/ruby_llm/agents/image/concerns/image_operation_dsl.rb
+- lib/ruby_llm/agents/image/concerns/image_operation_execution.rb
+- lib/ruby_llm/agents/image/editor.rb
+- lib/ruby_llm/agents/image/editor/dsl.rb
+- lib/ruby_llm/agents/image/editor/execution.rb
+- lib/ruby_llm/agents/image/generator.rb
+- lib/ruby_llm/agents/image/generator/active_storage_support.rb
+- lib/ruby_llm/agents/image/generator/content_policy.rb
+- lib/ruby_llm/agents/image/generator/pricing.rb
+- lib/ruby_llm/agents/image/generator/templates.rb
+- lib/ruby_llm/agents/image/pipeline.rb
+- lib/ruby_llm/agents/image/pipeline/dsl.rb
+- lib/ruby_llm/agents/image/pipeline/execution.rb
+- lib/ruby_llm/agents/image/transformer.rb
+- lib/ruby_llm/agents/image/transformer/dsl.rb
+- lib/ruby_llm/agents/image/transformer/execution.rb
+- lib/ruby_llm/agents/image/upscaler.rb
+- lib/ruby_llm/agents/image/upscaler/dsl.rb
+- lib/ruby_llm/agents/image/upscaler/execution.rb
+- lib/ruby_llm/agents/image/variator.rb
+- lib/ruby_llm/agents/image/variator/dsl.rb
+- lib/ruby_llm/agents/image/variator/execution.rb
+- lib/ruby_llm/agents/infrastructure/alert_manager.rb
+- lib/ruby_llm/agents/infrastructure/attempt_tracker.rb
+- lib/ruby_llm/agents/infrastructure/budget/budget_query.rb
+- lib/ruby_llm/agents/infrastructure/budget/config_resolver.rb
+- lib/ruby_llm/agents/infrastructure/budget/forecaster.rb
+- lib/ruby_llm/agents/infrastructure/budget/spend_recorder.rb
+- lib/ruby_llm/agents/infrastructure/budget_tracker.rb
+- lib/ruby_llm/agents/infrastructure/cache_helper.rb
+- lib/ruby_llm/agents/infrastructure/circuit_breaker.rb
+- lib/ruby_llm/agents/infrastructure/execution_logger_job.rb
+- lib/ruby_llm/agents/infrastructure/redactor.rb
+- lib/ruby_llm/agents/infrastructure/reliability.rb
+- lib/ruby_llm/agents/infrastructure/reliability/breaker_manager.rb
+- lib/ruby_llm/agents/infrastructure/reliability/execution_constraints.rb
+- lib/ruby_llm/agents/infrastructure/reliability/executor.rb
+- lib/ruby_llm/agents/infrastructure/reliability/fallback_routing.rb
+- lib/ruby_llm/agents/infrastructure/reliability/retry_strategy.rb
+- lib/ruby_llm/agents/pipeline.rb
+- lib/ruby_llm/agents/pipeline/builder.rb
+- lib/ruby_llm/agents/pipeline/context.rb
+- lib/ruby_llm/agents/pipeline/executor.rb
+- lib/ruby_llm/agents/pipeline/middleware/base.rb
+- lib/ruby_llm/agents/pipeline/middleware/budget.rb
+- lib/ruby_llm/agents/pipeline/middleware/cache.rb
+- lib/ruby_llm/agents/pipeline/middleware/instrumentation.rb
+- lib/ruby_llm/agents/pipeline/middleware/reliability.rb
+- lib/ruby_llm/agents/pipeline/middleware/tenant.rb
+- lib/ruby_llm/agents/rails/engine.rb
+- lib/ruby_llm/agents/results/background_removal_result.rb
+- lib/ruby_llm/agents/results/base.rb
+- lib/ruby_llm/agents/results/embedding_result.rb
+- lib/ruby_llm/agents/results/image_analysis_result.rb
+- lib/ruby_llm/agents/results/image_edit_result.rb
+- lib/ruby_llm/agents/results/image_generation_result.rb
+- lib/ruby_llm/agents/results/image_pipeline_result.rb
+- lib/ruby_llm/agents/results/image_transform_result.rb
+- lib/ruby_llm/agents/results/image_upscale_result.rb
+- lib/ruby_llm/agents/results/image_variation_result.rb
+- lib/ruby_llm/agents/results/moderation_result.rb
+- lib/ruby_llm/agents/results/speech_result.rb
+- lib/ruby_llm/agents/results/transcription_result.rb
+- lib/ruby_llm/agents/text/embedder.rb
+- lib/ruby_llm/agents/text/moderator.rb
+- lib/ruby_llm/agents/workflow/async.rb
+- lib/ruby_llm/agents/workflow/async_executor.rb
 - lib/ruby_llm/agents/workflow/instrumentation.rb
+- lib/ruby_llm/agents/workflow/orchestrator.rb
 - lib/ruby_llm/agents/workflow/parallel.rb
 - lib/ruby_llm/agents/workflow/pipeline.rb
 - lib/ruby_llm/agents/workflow/result.rb
 - lib/ruby_llm/agents/workflow/router.rb
+- lib/ruby_llm/agents/workflow/thread_pool.rb
 homepage: https://github.com/adham90/ruby_llm-agents
 licenses:
 - MIT

data/lib/ruby_llm/agents/base/caching.rb

@@ -1,40 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "../cache_helper"
-
-module RubyLLM
-  module Agents
-    class Base
-      # Cache management for agent responses
-      #
-      # Handles cache key generation and store access for
-      # caching agent execution results.
-      module Caching
-        include CacheHelper
-
-        # Generates the full cache key for this agent invocation
-        #
-        # @return [String] Cache key in format "ruby_llm_agent/ClassName/version/hash"
-        def agent_cache_key
-          ["ruby_llm_agent", self.class.name, self.class.version, cache_key_hash].join("/")
-        end
-
-        # Generates a hash of the cache key data
-        #
-        # @return [String] SHA256 hex digest of the cache key data
-        def cache_key_hash
-          Digest::SHA256.hexdigest(cache_key_data.to_json)
-        end
-
-        # Returns data to include in cache key generation
-        #
-        # Override to customize what parameters affect cache invalidation.
-        #
-        # @return [Hash] Data to hash for cache key
-        def cache_key_data
-          @options.except(:skip_cache, :dry_run, :with)
-        end
-      end
-    end
-  end
-end

data/lib/ruby_llm/agents/base/cost_calculation.rb

@@ -1,105 +0,0 @@
-# frozen_string_literal: true
-
-module RubyLLM
-  module Agents
-    class Base
-      # Cost calculation methods for token and pricing calculations
-      #
-      # Handles input/output cost calculations, model info resolution,
-      # and budget tracking for agent executions.
-      module CostCalculation
-        # Calculates input cost from tokens
-        #
-        # @param input_tokens [Integer, nil] Number of input tokens
-        # @param response_model_id [String, nil] Model that responded
-        # @return [Float, nil] Input cost in USD
-        def result_input_cost(input_tokens, response_model_id)
-          return nil unless input_tokens
-          model_info = result_model_info(response_model_id)
-          return nil unless model_info&.pricing
-          price = model_info.pricing.text_tokens&.input || 0
-          (input_tokens / 1_000_000.0 * price).round(6)
-        end
-
-        # Calculates output cost from tokens
-        #
-        # @param output_tokens [Integer, nil] Number of output tokens
-        # @param response_model_id [String, nil] Model that responded
-        # @return [Float, nil] Output cost in USD
-        def result_output_cost(output_tokens, response_model_id)
-          return nil unless output_tokens
-          model_info = result_model_info(response_model_id)
-          return nil unless model_info&.pricing
-          price = model_info.pricing.text_tokens&.output || 0
-          (output_tokens / 1_000_000.0 * price).round(6)
-        end
-
-        # Calculates total cost from tokens
-        #
-        # @param input_tokens [Integer, nil] Number of input tokens
-        # @param output_tokens [Integer, nil] Number of output tokens
-        # @param response_model_id [String, nil] Model that responded
-        # @return [Float, nil] Total cost in USD
-        def result_total_cost(input_tokens, output_tokens, response_model_id)
-          input_cost = result_input_cost(input_tokens, response_model_id)
-          output_cost = result_output_cost(output_tokens, response_model_id)
-          return nil unless input_cost || output_cost
-          ((input_cost || 0) + (output_cost || 0)).round(6)
-        end
-
-        # Resolves model info for cost calculation
-        #
-        # @param response_model_id [String, nil] Model ID from response
-        # @return [Object, nil] Model info or nil
-        def result_model_info(response_model_id)
-          lookup_id = response_model_id || model
-          return nil unless lookup_id
-          model_obj, _provider = RubyLLM::Models.resolve(lookup_id)
-          model_obj
-        rescue StandardError
-          nil
-        end
-
-        # Resolves model info for cost calculation (alternate method)
-        #
-        # @param model_id [String] The model identifier
-        # @return [Object, nil] Model info or nil
-        def resolve_model_info(model_id)
-          model_obj, _provider = RubyLLM::Models.resolve(model_id)
-          model_obj
-        rescue StandardError
-          nil
-        end
-
-        # Records cost from an attempt to the budget tracker
-        #
-        # @param attempt_tracker [AttemptTracker] The attempt tracker
-        # @param tenant_id [String, nil] Optional tenant identifier for multi-tenant tracking
-        # @return [void]
-        def record_attempt_cost(attempt_tracker, tenant_id: nil)
-          successful = attempt_tracker.successful_attempt
-          return unless successful
-
-          # Calculate cost for this execution
-          # Note: Full cost calculation happens in instrumentation, but we
-          # record the spend here for budget tracking
-          model_info = resolve_model_info(successful[:model_id])
-          return unless model_info&.pricing
-
-          input_tokens = successful[:input_tokens] || 0
-          output_tokens = successful[:output_tokens] || 0
-
-          input_price = model_info.pricing.text_tokens&.input || 0
-          output_price = model_info.pricing.text_tokens&.output || 0
-
-          total_cost = (input_tokens / 1_000_000.0 * input_price) +
-                       (output_tokens / 1_000_000.0 * output_price)
-
-          BudgetTracker.record_spend!(self.class.name, total_cost, tenant_id: tenant_id)
-        rescue StandardError => e
-          Rails.logger.warn("[RubyLLM::Agents] Failed to record budget spend: #{e.message}")
-        end
-      end
-    end
-  end
-end