llm_cost_tracker 0.6.1 → 0.7.1

metadata

@@ -1,15 +1,35 @@
 --- !ruby/object:Gem::Specification
 name: llm_cost_tracker
 version: !ruby/object:Gem::Version
-  version: 0.6.1
+  version: 0.7.1
 platform: ruby
 authors:
 - Sergii Khomenko
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-04-29 00:00:00.000000000 Z
+date: 2026-04-30 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '9.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '9.0'
 - !ruby/object:Gem::Dependency
   name: activesupport
   requirement: !ruby/object:Gem::Requirement
@@ -65,7 +85,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '3.0'
 - !ruby/object:Gem::Dependency
-  name: activerecord
+  name: railties
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -74,7 +94,7 @@ dependencies:
     - - "<"
       - !ruby/object:Gem::Version
         version: '9.0'
-  type: :development
+  type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
@@ -99,25 +119,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '1.16'
 - !ruby/object:Gem::Dependency
-  name: railties
+  name: pg
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '7.1'
-    - - "<"
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '9.0'
+        version: '1.6'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '7.1'
-    - - "<"
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '9.0'
+        version: '1.6'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -188,26 +202,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.8'
-- !ruby/object:Gem::Dependency
-  name: sqlite3
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '1.4'
-    - - "<"
-      - !ruby/object:Gem::Version
-        version: '3.0'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '1.4'
-    - - "<"
-      - !ruby/object:Gem::Version
-        version: '3.0'
 - !ruby/object:Gem::Dependency
   name: webmock
   requirement: !ruby/object:Gem::Requirement
@@ -224,8 +218,9 @@ dependencies:
         version: '3.0'
 description: Tracks token usage, latency, and estimated costs for RubyLLM, OpenAI,
   Anthropic, Google Gemini, OpenRouter, DeepSeek, and OpenAI-compatible APIs. Works
-  through Faraday middleware or explicit track/track_stream helpers, with ActiveRecord
-  storage, tag-based attribution, price sync tasks, and budget guardrails.
+  through Rails SDK integrations, Faraday middleware, or explicit track/track_stream
+  helpers, with ActiveRecord storage, tag-based attribution, price sync tasks, and
+  budget guardrails.
 email:
 - sergey@mm.st
 executables: []
@@ -253,18 +248,26 @@ files:
 - app/helpers/llm_cost_tracker/dashboard_filter_options_helper.rb
 - app/helpers/llm_cost_tracker/dashboard_query_helper.rb
 - app/helpers/llm_cost_tracker/pagination_helper.rb
+- app/helpers/llm_cost_tracker/token_usage_helper.rb
+- app/models/llm_cost_tracker/ingestion/event.rb
+- app/models/llm_cost_tracker/ingestion/lease.rb
+- app/models/llm_cost_tracker/ledger/call.rb
+- app/models/llm_cost_tracker/ledger/call_metrics.rb
+- app/models/llm_cost_tracker/ledger/period/grouping.rb
+- app/models/llm_cost_tracker/ledger/period/total.rb
+- app/models/llm_cost_tracker/ledger/tags/accessors.rb
 - app/services/llm_cost_tracker/dashboard/data_quality.rb
-- app/services/llm_cost_tracker/dashboard/data_quality_aggregate.rb
 - app/services/llm_cost_tracker/dashboard/date_range.rb
 - app/services/llm_cost_tracker/dashboard/filter.rb
 - app/services/llm_cost_tracker/dashboard/overview_stats.rb
+- app/services/llm_cost_tracker/dashboard/pagination.rb
+- app/services/llm_cost_tracker/dashboard/params.rb
 - app/services/llm_cost_tracker/dashboard/provider_breakdown.rb
 - app/services/llm_cost_tracker/dashboard/spend_anomaly.rb
 - app/services/llm_cost_tracker/dashboard/tag_breakdown.rb
 - app/services/llm_cost_tracker/dashboard/tag_key_explorer.rb
 - app/services/llm_cost_tracker/dashboard/time_series.rb
 - app/services/llm_cost_tracker/dashboard/top_models.rb
-- app/services/llm_cost_tracker/pagination.rb
 - app/views/layouts/llm_cost_tracker/application.html.erb
 - app/views/llm_cost_tracker/calls/index.html.erb
 - app/views/llm_cost_tracker/calls/show.html.erb
@@ -284,28 +287,27 @@ files:
 - app/views/llm_cost_tracker/tags/show.html.erb
 - config/routes.rb
 - lib/llm_cost_tracker.rb
-- lib/llm_cost_tracker/active_record_adapter.rb
 - lib/llm_cost_tracker/assets.rb
 - lib/llm_cost_tracker/budget.rb
-- lib/llm_cost_tracker/capture_verifier.rb
+- lib/llm_cost_tracker/capture/stream.rb
+- lib/llm_cost_tracker/capture/stream_collector.rb
+- lib/llm_cost_tracker/capture/stream_tracker.rb
 - lib/llm_cost_tracker/configuration.rb
 - lib/llm_cost_tracker/configuration/instrumentation.rb
-- lib/llm_cost_tracker/configuration/storage_backend.rb
-- lib/llm_cost_tracker/cost.rb
 - lib/llm_cost_tracker/doctor.rb
-- lib/llm_cost_tracker/doctor/capture_check.rb
+- lib/llm_cost_tracker/doctor/capture_verifier.rb
+- lib/llm_cost_tracker/doctor/check.rb
 - lib/llm_cost_tracker/doctor/ingestion_check.rb
+- lib/llm_cost_tracker/doctor/price_check.rb
 - lib/llm_cost_tracker/engine.rb
-- lib/llm_cost_tracker/engine_compatibility.rb
 - lib/llm_cost_tracker/errors.rb
 - lib/llm_cost_tracker/event.rb
-- lib/llm_cost_tracker/event_metadata.rb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/add_ingestion_generator.rb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/add_latency_ms_generator.rb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/add_period_totals_generator.rb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/add_provider_response_id_generator.rb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/add_streaming_generator.rb
-- lib/llm_cost_tracker/generators/llm_cost_tracker/add_usage_breakdown_generator.rb
+- lib/llm_cost_tracker/generators/llm_cost_tracker/add_token_usage_generator.rb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/install_generator.rb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/prices_generator.rb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/templates/add_ingestion_to_llm_cost_tracker.rb.erb
@@ -313,85 +315,69 @@ files:
 - lib/llm_cost_tracker/generators/llm_cost_tracker/templates/add_period_totals_to_llm_cost_tracker.rb.erb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/templates/add_provider_response_id_to_llm_api_calls.rb.erb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/templates/add_streaming_to_llm_api_calls.rb.erb
-- lib/llm_cost_tracker/generators/llm_cost_tracker/templates/add_usage_breakdown_to_llm_api_calls.rb.erb
+- lib/llm_cost_tracker/generators/llm_cost_tracker/templates/add_token_usage_to_llm_api_calls.rb.erb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/templates/create_llm_api_calls.rb.erb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/templates/initializer.rb.erb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/templates/upgrade_llm_api_call_cost_precision.rb.erb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/templates/upgrade_llm_api_call_tags_to_jsonb.rb.erb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/upgrade_cost_precision_generator.rb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/upgrade_tags_to_jsonb_generator.rb
-- lib/llm_cost_tracker/inbox_event.rb
-- lib/llm_cost_tracker/ingestor_lease.rb
+- lib/llm_cost_tracker/ingestion.rb
+- lib/llm_cost_tracker/ingestion/batch.rb
+- lib/llm_cost_tracker/ingestion/inbox.rb
+- lib/llm_cost_tracker/ingestion/lease_claim.rb
+- lib/llm_cost_tracker/ingestion/worker.rb
+- lib/llm_cost_tracker/integrations.rb
 - lib/llm_cost_tracker/integrations/anthropic.rb
 - lib/llm_cost_tracker/integrations/base.rb
-- lib/llm_cost_tracker/integrations/object_reader.rb
 - lib/llm_cost_tracker/integrations/openai.rb
-- lib/llm_cost_tracker/integrations/registry.rb
 - lib/llm_cost_tracker/integrations/ruby_llm.rb
-- lib/llm_cost_tracker/integrations/stream_tracker.rb
-- lib/llm_cost_tracker/llm_api_call.rb
-- lib/llm_cost_tracker/llm_api_call_metrics.rb
+- lib/llm_cost_tracker/ledger.rb
+- lib/llm_cost_tracker/ledger/period.rb
+- lib/llm_cost_tracker/ledger/period/totals.rb
+- lib/llm_cost_tracker/ledger/rollups.rb
+- lib/llm_cost_tracker/ledger/rollups/batch.rb
+- lib/llm_cost_tracker/ledger/rollups/upsert_sql.rb
+- lib/llm_cost_tracker/ledger/schema/adapter.rb
+- lib/llm_cost_tracker/ledger/schema/calls.rb
+- lib/llm_cost_tracker/ledger/schema/period_totals.rb
+- lib/llm_cost_tracker/ledger/store.rb
+- lib/llm_cost_tracker/ledger/tags/query.rb
+- lib/llm_cost_tracker/ledger/tags/sql.rb
 - lib/llm_cost_tracker/logging.rb
 - lib/llm_cost_tracker/middleware/faraday.rb
-- lib/llm_cost_tracker/parameter_hash.rb
-- lib/llm_cost_tracker/parsed_usage.rb
+- lib/llm_cost_tracker/parsers.rb
 - lib/llm_cost_tracker/parsers/anthropic.rb
 - lib/llm_cost_tracker/parsers/base.rb
 - lib/llm_cost_tracker/parsers/gemini.rb
 - lib/llm_cost_tracker/parsers/openai.rb
 - lib/llm_cost_tracker/parsers/openai_compatible.rb
 - lib/llm_cost_tracker/parsers/openai_usage.rb
-- lib/llm_cost_tracker/parsers/registry.rb
 - lib/llm_cost_tracker/parsers/sse.rb
-- lib/llm_cost_tracker/period_grouping.rb
-- lib/llm_cost_tracker/period_total.rb
-- lib/llm_cost_tracker/price_freshness.rb
-- lib/llm_cost_tracker/price_registry.rb
-- lib/llm_cost_tracker/price_sync.rb
-- lib/llm_cost_tracker/price_sync/fetcher.rb
-- lib/llm_cost_tracker/price_sync/registry_diff.rb
-- lib/llm_cost_tracker/price_sync/registry_loader.rb
-- lib/llm_cost_tracker/price_sync/registry_writer.rb
 - lib/llm_cost_tracker/prices.json
 - lib/llm_cost_tracker/pricing.rb
+- lib/llm_cost_tracker/pricing/components.rb
 - lib/llm_cost_tracker/pricing/effective_prices.rb
 - lib/llm_cost_tracker/pricing/explainer.rb
 - lib/llm_cost_tracker/pricing/lookup.rb
+- lib/llm_cost_tracker/pricing/registry.rb
+- lib/llm_cost_tracker/pricing/sync.rb
+- lib/llm_cost_tracker/pricing/sync/fetcher.rb
+- lib/llm_cost_tracker/pricing/sync/registry_diff.rb
+- lib/llm_cost_tracker/pricing/sync/registry_loader.rb
+- lib/llm_cost_tracker/pricing/sync/registry_writer.rb
+- lib/llm_cost_tracker/pricing/unknown.rb
 - lib/llm_cost_tracker/railtie.rb
 - lib/llm_cost_tracker/report.rb
-- lib/llm_cost_tracker/report_data.rb
-- lib/llm_cost_tracker/report_formatter.rb
-- lib/llm_cost_tracker/request_url.rb
+- lib/llm_cost_tracker/report/data.rb
+- lib/llm_cost_tracker/report/formatter.rb
 - lib/llm_cost_tracker/retention.rb
-- lib/llm_cost_tracker/storage/active_record_backend.rb
-- lib/llm_cost_tracker/storage/active_record_connection_cleanup.rb
-- lib/llm_cost_tracker/storage/active_record_inbox.rb
-- lib/llm_cost_tracker/storage/active_record_inbox_batch.rb
-- lib/llm_cost_tracker/storage/active_record_ingestor.rb
-- lib/llm_cost_tracker/storage/active_record_ingestor_lease.rb
-- lib/llm_cost_tracker/storage/active_record_period_totals.rb
-- lib/llm_cost_tracker/storage/active_record_periods.rb
-- lib/llm_cost_tracker/storage/active_record_rollup_batch.rb
-- lib/llm_cost_tracker/storage/active_record_rollup_upsert_sql.rb
-- lib/llm_cost_tracker/storage/active_record_rollups.rb
-- lib/llm_cost_tracker/storage/active_record_store.rb
-- lib/llm_cost_tracker/storage/custom_backend.rb
-- lib/llm_cost_tracker/storage/dispatcher.rb
-- lib/llm_cost_tracker/storage/log_backend.rb
-- lib/llm_cost_tracker/storage/registry.rb
-- lib/llm_cost_tracker/stream_capture.rb
-- lib/llm_cost_tracker/stream_collector.rb
-- lib/llm_cost_tracker/tag_accessors.rb
-- lib/llm_cost_tracker/tag_context.rb
-- lib/llm_cost_tracker/tag_key.rb
-- lib/llm_cost_tracker/tag_query.rb
-- lib/llm_cost_tracker/tag_sanitizer.rb
-- lib/llm_cost_tracker/tag_sql.rb
-- lib/llm_cost_tracker/tags_column.rb
+- lib/llm_cost_tracker/tags/context.rb
+- lib/llm_cost_tracker/tags/key.rb
+- lib/llm_cost_tracker/tags/sanitizer.rb
+- lib/llm_cost_tracker/token_usage.rb
 - lib/llm_cost_tracker/tracker.rb
-- lib/llm_cost_tracker/unknown_pricing.rb
-- lib/llm_cost_tracker/usage_breakdown.rb
-- lib/llm_cost_tracker/value_helpers.rb
+- lib/llm_cost_tracker/usage_capture.rb
 - lib/llm_cost_tracker/version.rb
 - lib/tasks/llm_cost_tracker.rake
 homepage: https://github.com/sergey-homenko/llm_cost_tracker
@@ -421,5 +407,5 @@ requirements: []
 rubygems_version: 3.5.22
 signing_key:
 specification_version: 4
-summary: Self-hosted LLM usage and cost tracking for Ruby and Rails
+summary: Rails-native LLM usage and cost tracking with ActiveRecord storage
 test_files: []

data/app/services/llm_cost_tracker/dashboard/data_quality_aggregate.rb

@@ -1,81 +0,0 @@
-# frozen_string_literal: true
-
-module LlmCostTracker
-  module Dashboard
-    class DataQualityAggregate
-      class << self
-        def call(scope:)
-          model = scope.klass
-          expressions = aggregate_expressions(scope, model:)
-          values = Array(scope.unscope(:order).pick(*expressions.values))
-
-          expressions.keys.zip(values).to_h
-        end
-
-        private
-
-        def aggregate_expressions(scope, model:)
-          usage_breakdown_present = model.usage_breakdown_columns?
-          usage_breakdown_cost_present = model.usage_breakdown_cost_columns?
-
-          expressions = {
-            total_calls: Arel.sql("COUNT(*)"),
-            unknown_pricing_count: conditional_count_expression("total_cost IS NULL"),
-            tagged_calls_count: tagged_calls_expression(model)
-          }
-
-          if model.latency_column?
-            expressions[:missing_latency_count] = conditional_count_expression("latency_ms IS NULL")
-          end
-          expressions[:streaming_count] = conditional_count_expression("stream") if model.stream_column?
-          if model.stream_column? && model.usage_source_column?
-            expressions[:streaming_missing_usage_count] =
-              conditional_count_expression("stream AND (usage_source = 'unknown' OR usage_source IS NULL)")
-          end
-          if model.provider_response_id_column?
-            expressions[:missing_provider_response_id_count] =
-              conditional_count_expression("provider_response_id IS NULL OR provider_response_id = ''")
-          end
-
-          usage_sum_columns(usage_breakdown_present, usage_breakdown_cost_present).each do |column|
-            expressions[column] = sum_expression(scope, column)
-          end
-
-          expressions
-        end
-
-        def usage_sum_columns(usage_breakdown_present, usage_breakdown_cost_present)
-          columns = %i[input_tokens output_tokens input_cost output_cost]
-          if usage_breakdown_present
-            columns += %i[cache_read_input_tokens cache_write_input_tokens hidden_output_tokens]
-          end
-          columns += %i[cache_read_input_cost cache_write_input_cost] if usage_breakdown_cost_present
-          columns
-        end
-
-        def conditional_count_expression(predicate)
-          Arel.sql("COALESCE(SUM(CASE WHEN #{predicate} THEN 1 ELSE 0 END), 0)")
-        end
-
-        def tagged_calls_expression(model)
-          table = model.quoted_table_name
-          column = "#{table}.#{model.connection.quote_column_name('tags')}"
-
-          Arel.sql(case
-                   when model.tags_jsonb_column?
-                     "COALESCE(SUM(CASE WHEN #{column} <> '{}'::jsonb THEN 1 ELSE 0 END), 0)"
-                   when model.tags_mysql_json_column?
-                     "COALESCE(SUM(CASE WHEN JSON_LENGTH(#{column}) > 0 THEN 1 ELSE 0 END), 0)"
-                   else
-                     "COALESCE(SUM(CASE WHEN #{column} IS NOT NULL AND #{column} <> '' " \
-                     "AND #{column} <> '{}' THEN 1 ELSE 0 END), 0)"
-                   end)
-        end
-
-        def sum_expression(scope, column)
-          Arel.sql("COALESCE(SUM(#{scope.connection.quote_column_name(column)}), 0)")
-        end
-      end
-    end
-  end
-end

data/app/services/llm_cost_tracker/pagination.rb

@@ -1,57 +0,0 @@
-# frozen_string_literal: true
-
-module LlmCostTracker
-  class Pagination
-    DEFAULT_PER = 50
-    MAX_PER = 200
-    MIN_PAGE = 1
-
-    attr_reader :page, :per
-
-    def self.call(params)
-      params = LlmCostTracker::ParameterHash.with_indifferent_access(params)
-      new(
-        page: integer_param(params, :page, default: MIN_PAGE, min: MIN_PAGE),
-        per: integer_param(params, :per, default: DEFAULT_PER, min: 1, max: MAX_PER)
-      )
-    end
-
-    def self.integer_param(params, key, default:, min:, max: nil)
-      value = Integer(params[key], 10)
-      value = [value, min].max
-      value = [value, max].min if max
-      value
-    rescue ArgumentError, TypeError
-      default
-    end
-    private_class_method :integer_param
-
-    def initialize(page:, per:)
-      @page = page
-      @per = per
-      freeze
-    end
-
-    def limit
-      per
-    end
-
-    def offset
-      (page - 1) * per
-    end
-
-    def prev_page?
-      page > MIN_PAGE
-    end
-
-    def next_page?(total_count)
-      offset + per < total_count.to_i
-    end
-
-    def total_pages(total_count)
-      return MIN_PAGE if total_count.to_i <= 0
-
-      [(total_count.to_f / per).ceil, MIN_PAGE].max
-    end
-  end
-end

data/lib/llm_cost_tracker/active_record_adapter.rb

@@ -1,49 +0,0 @@
-# frozen_string_literal: true
-
-module LlmCostTracker
-  module ActiveRecordAdapter
-    MYSQL_ADAPTERS = %w[
-      ActiveRecord::ConnectionAdapters::AbstractMysqlAdapter
-      ActiveRecord::ConnectionAdapters::Mysql2Adapter
-      ActiveRecord::ConnectionAdapters::TrilogyAdapter
-    ].freeze
-    POSTGRESQL_ADAPTERS = %w[
-      ActiveRecord::ConnectionAdapters::PostgreSQLAdapter
-    ].freeze
-    SQLITE_ADAPTERS = %w[
-      ActiveRecord::ConnectionAdapters::SQLite3Adapter
-    ].freeze
-    MYSQL_PATTERN = /mysql|trilogy|mariadb/i
-    POSTGRESQL_PATTERN = /postgres/i
-    SQLITE_PATTERN = /sqlite/i
-
-    class << self
-      def mysql?(value) = adapter_instance?(value, MYSQL_ADAPTERS) || adapter_name(value).match?(MYSQL_PATTERN)
-
-      def postgresql?(value)
-        adapter_instance?(value, POSTGRESQL_ADAPTERS) || adapter_name(value).match?(POSTGRESQL_PATTERN)
-      end
-
-      def sqlite?(value) = adapter_instance?(value, SQLITE_ADAPTERS) || adapter_name(value).match?(SQLITE_PATTERN)
-
-      private
-
-      def adapter_instance?(value, class_names)
-        class_names.any? do |class_name|
-          adapter_class = constantize(class_name)
-          adapter_class && value.is_a?(adapter_class)
-        end
-      end
-
-      def constantize(name)
-        name.split("::").reduce(Object) { |namespace, part| namespace.const_get(part, false) }
-      rescue NameError
-        nil
-      end
-
-      def adapter_name(value)
-        value.respond_to?(:adapter_name) ? value.adapter_name.to_s : value.to_s
-      end
-    end
-  end
-end

data/lib/llm_cost_tracker/capture_verifier.rb

@@ -1,71 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "storage/dispatcher"
-
-module LlmCostTracker
-  class CaptureVerifier
-    Check = Data.define(:status, :name, :message)
-
-    class << self
-      def call = new.checks
-
-      def report(checks = call)
-        (["LLM Cost Tracker capture verification"] + checks.map { |check| format_check(check) }).join("\n")
-      end
-
-      def healthy?(checks = call)
-        checks.none? { |check| check.status == :error }
-      end
-
-      private
-
-      def format_check(check)
-        "[#{check.status}] #{check.name}: #{check.message}"
-      end
-    end
-
-    def checks
-      [
-        enabled_check,
-        *integration_checks,
-        *storage_checks
-      ].compact
-    end
-
-    private
-
-    def enabled_check
-      return Check.new(:ok, "tracking", "enabled") if LlmCostTracker.configuration.enabled
-
-      Check.new(:error, "tracking", "disabled; set config.enabled = true before verifying capture")
-    end
-
-    def integration_checks
-      enabled = LlmCostTracker.configuration.instrumented_integrations
-      if enabled.empty?
-        return [
-          Check.new(:ok, "sdk integrations", "none enabled; Faraday middleware and manual capture remain available")
-        ]
-      end
-
-      LlmCostTracker::Integrations.checks.map do |check|
-        Check.new(check.status, "sdk integration #{check.name}", check.message)
-      end
-    end
-
-    def storage_checks
-      backend = LlmCostTracker::Storage::Registry.fetch(LlmCostTracker.configuration.storage_backend)
-      unless backend.respond_to?(:verify)
-        return [
-          Check.new(:warn, "storage", "#{LlmCostTracker.configuration.storage_backend} backend has no verifier")
-        ]
-      end
-
-      backend.verify.map do |check|
-        Check.new(check.status, check.name, check.message)
-      end
-    rescue LlmCostTracker::Error => e
-      [Check.new(:error, "storage", e.message)]
-    end
-  end
-end

data/lib/llm_cost_tracker/configuration/storage_backend.rb

@@ -1,26 +0,0 @@
-# frozen_string_literal: true
-
-module LlmCostTracker
-  module ConfigurationStorageBackend
-    def storage_backend=(value)
-      ensure_shared_configuration_mutable!
-      @storage_backend = normalize_storage_backend(value)
-    end
-
-    private
-
-    def normalize_storage_backend(value)
-      value = :log if value.nil?
-      value = value.to_sym
-      return value if self.class::STORAGE_BACKENDS.include?(value)
-      return value if defined?(Storage::Registry) && Storage::Registry.registered?(value)
-
-      names = if defined?(Storage::Registry)
-                Storage::Registry.names
-              else
-                self.class::STORAGE_BACKENDS
-              end
-      raise Error, "Unknown storage_backend: #{value.inspect}. Use one of: #{names.join(', ')}"
-    end
-  end
-end

data/lib/llm_cost_tracker/cost.rb

@@ -1,12 +0,0 @@
-# frozen_string_literal: true
-
-module LlmCostTracker
-  Cost = Data.define(
-    :input_cost,
-    :cache_read_input_cost,
-    :cache_write_input_cost,
-    :output_cost,
-    :total_cost,
-    :currency
-  )
-end

data/lib/llm_cost_tracker/doctor/capture_check.rb

@@ -1,39 +0,0 @@
-# frozen_string_literal: true
-
-module LlmCostTracker
-  class Doctor
-    class CaptureCheck
-      def self.call(check_class)
-        new(check_class).call
-      end
-
-      def initialize(check_class)
-        @check_class = check_class
-      end
-
-      def call
-        config = LlmCostTracker.configuration
-        return disabled_check unless config.enabled
-        return integrations_check(config.instrumented_integrations) if config.instrumented_integrations.any?
-
-        check(:ok, "no SDK integrations enabled; Faraday middleware and manual capture remain available")
-      end
-
-      private
-
-      attr_reader :check_class
-
-      def disabled_check
-        check(:warn, "tracking is disabled; set config.enabled = true to record calls")
-      end
-
-      def integrations_check(integrations)
-        check(:ok, "SDK integrations enabled: #{integrations.join(', ')}")
-      end
-
-      def check(status, message)
-        check_class.new(status, "capture", message)
-      end
-    end
-  end
-end

data/lib/llm_cost_tracker/engine_compatibility.rb

@@ -1,15 +0,0 @@
-# frozen_string_literal: true
-
-module LlmCostTracker
-  module EngineCompatibility
-    REQUIRED_RAILS_VERSION = Gem::Version.new("7.1.0")
-
-    class << self
-      def check_rails_version!(version)
-        return if Gem::Version.new(version) >= REQUIRED_RAILS_VERSION
-
-        raise LlmCostTracker::Error, "LlmCostTracker::Engine requires Rails 7.1+"
-      end
-    end
-  end
-end