llm_cost_tracker 0.6.0 → 0.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fa3f705baf280c2c2239b2dab3522fe7db9b60e26060b00fc08dcc039117da83
-  data.tar.gz: ea34bdad7cb0d7c9fb3233b40b609cea1361ded833ad130c4a7a7ce559b34758
+  metadata.gz: 9de844d3302fda39c5a8b32736d8abff488a5cd6cfd00932f492a1ec701f308d
+  data.tar.gz: a8f942605f9cfbcc77f7e998e3e504150fa820e44829156e901cb06bdcc6d0cf
 SHA512:
-  metadata.gz: 03c55866b522b36b0728f73fd0ae0075e0a42faa6f80c429865f120d2c597e0b0996665b6a82528c566a0f8554d7636dfd03d9ab600441dafd5a4f0233d3f56b
-  data.tar.gz: af01c4554912d80276bf54ed97aa379c2eaed791fa357ca135aa008fdcbdd41e365c236ac74c9ceb1982c0fbbf2538bc569df1690acb4b5758895dd48c4497b5
+  metadata.gz: 4c7d9869224101f85298dab0b4395b0c2b96634f46f24e76487412f0f8418e4270fb15124f6f238499c3205f4ad2ebae6e8bcba1b265f7496e3509b7da7c5ea9
+  data.tar.gz: f07e91cf62de94d7e7c76eff7ec4b798208724df5722374d6a72884445a70a64ef83421b90aa8be51b887e277f3e8bcb7eec3ad14987eec25609483bc11f0667

data/CHANGELOG.md

@@ -4,6 +4,20 @@ Format: [Keep a Changelog](https://keepachangelog.com/en/1.1.0/). Versioning: [S
 
 ## [Unreleased]
 
+## [0.7.0] - 2026-04-29
+
+### Changed
+
+- BREAKING: ActiveRecord is now the only storage path; removed `storage_backend`, `custom_storage`, `Storage.register`, `:log`, and `:custom`.
+- BREAKING: PostgreSQL and MySQL are now the only supported database adapters; SQLite support was removed.
+- Runtime dependencies now include Rails and ActiveRecord.
+
+## [0.6.1] - 2026-04-29
+
+### Fixed
+
+- Exclude repository documentation from the published gem package.
+
 ## [0.6.0] - 2026-04-29
 
 ### Added

data/README.md

@@ -10,7 +10,7 @@ If you have OpenAI, Anthropic, or Gemini in production and someone keeps asking
 
 It is not Langfuse, Helicone, or LiteLLM. It does not capture prompts, score completions, or replay traces. It does one thing: tells you which provider, which model, which feature, and which user burned how much money. That's the entire pitch.
 
-Requires Ruby 3.3+, ActiveSupport 7.1+, Faraday 2.0+. ActiveRecord storage and the dashboard need Rails 7.1+.
+Requires Ruby 3.3+, Rails 7.1+, PostgreSQL or MySQL, and Faraday 2.0+.
 
 ![Dashboard overview](docs/dashboard-overview.png)
 
@@ -35,7 +35,6 @@ Drop this into `config/initializers/llm_cost_tracker.rb`:
 
 ```ruby
 LlmCostTracker.configure do |config|
-  config.storage_backend = :active_record
   config.default_tags    = -> { { environment: Rails.env } }
   config.instrument :openai
 end
@@ -219,7 +218,7 @@ Mount the engine wherever you want — it's plain ERB, no JavaScript bundle, no
 mount LlmCostTracker::Engine => "/llm-costs"
 ```
 
-Pages: overview (spend trend, budget status, anomaly banner), models, calls (filterable, paginated, CSV export), tags, data quality. Reads `llm_api_calls`, so use `:active_record` storage if you want to mount it.
+Pages: overview (spend trend, budget status, anomaly banner), models, calls (filterable, paginated, CSV export), tags, data quality. Reads the ActiveRecord ledger in `llm_api_calls`.
 
 Auth is your job. Examples for basic auth and Devise: [`docs/dashboard.md`](docs/dashboard.md).
 

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

@@ -45,7 +45,7 @@ module LlmCostTracker
         return postgresql_sql if ActiveRecordAdapter.postgresql?(connection)
         return mysql_sql if ActiveRecordAdapter.mysql?(connection)
 
-        sqlite_sql
+        ActiveRecordAdapter.ensure_supported!(connection)
       end
 
       def mysql_sql
@@ -83,22 +83,6 @@ module LlmCostTracker
         SQL
       end
 
-      def sqlite_sql
-        <<~SQL.squish
-          SELECT je.key AS key,
-                 COUNT(*) AS calls_count,
-                 COUNT(DISTINCT je.value) AS distinct_values
-          FROM (#{subquery}) AS sub,
-               json_each(sub.tags) AS je
-          WHERE sub.tags IS NOT NULL
-            AND sub.tags != '{}'
-            AND sub.tags != ''
-          GROUP BY je.key
-          ORDER BY calls_count DESC
-          LIMIT #{limit}
-        SQL
-      end
-
       def normalized_limit(value)
         value = value.to_i
         value.positive? ? [value, DEFAULT_LIMIT].min : DEFAULT_LIMIT

data/config/routes.rb

@@ -8,6 +8,6 @@ LlmCostTracker::Engine.routes.draw do
   get "tags/:key", to: "tags#show",   as: :tag, format: false
   get "data_quality", to: "data_quality#index", as: :data_quality
 
-  get "assets/#{LlmCostTracker::Assets.stylesheet_filename}",
+  get "assets/#{LlmCostTracker::Assets::STYLESHEET_FILENAME}",
       to: "assets#stylesheet", as: :stylesheet
 end

data/lib/llm_cost_tracker/active_record_adapter.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "errors"
+
 module LlmCostTracker
   module ActiveRecordAdapter
     MYSQL_ADAPTERS = %w[
@@ -10,12 +12,8 @@ module LlmCostTracker
     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)
@@ -24,7 +22,13 @@ module LlmCostTracker
         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)
+      def supported?(value) = mysql?(value) || postgresql?(value)
+
+      def ensure_supported!(value)
+        return if supported?(value)
+
+        raise Error, "Unsupported database adapter: #{adapter_name(value)}. Use PostgreSQL or MySQL."
+      end
 
       private
 

data/lib/llm_cost_tracker/assets.rb

@@ -9,11 +9,5 @@ module LlmCostTracker
     STYLESHEET_PATH = File.join(ROOT, STYLESHEET).freeze
     STYLESHEET_FINGERPRINT = Digest::SHA256.file(STYLESHEET_PATH).hexdigest[0, 12].freeze
     STYLESHEET_FILENAME = "application-#{STYLESHEET_FINGERPRINT}.css".freeze
-
-    class << self
-      def root = ROOT
-      def stylesheet_fingerprint = STYLESHEET_FINGERPRINT
-      def stylesheet_filename = STYLESHEET_FILENAME
-    end
   end
 end

data/lib/llm_cost_tracker/budget.rb

@@ -8,7 +8,6 @@ module LlmCostTracker
       def enforce!
         config = LlmCostTracker.configuration
         return unless config.budget_exceeded_behavior == :block_requests
-        return unless config.active_record?
 
         budgets = enforce_period_budgets(config)
         return if budgets.empty?
@@ -28,7 +27,7 @@ module LlmCostTracker
 
         check_per_call_budget(event, config)
         budgets = check_period_budgets(config)
-        totals = totals_for_check(event, config, budgets)
+        totals = totals_for_check(event, budgets)
 
         budgets.each do |period, budget|
           total = totals.fetch(period)
@@ -63,11 +62,10 @@ module LlmCostTracker
         }.compact
       end
 
-      def totals_for_check(event, config, budgets)
+      def totals_for_check(event, budgets)
         return {} if budgets.empty?
-        return active_record_totals(budgets.keys, time: event.tracked_at) if config.active_record?
 
-        budgets.to_h { |period, _budget| [period, event.cost.total_cost] }
+        active_record_totals(budgets.keys, time: event.tracked_at)
       end
 
       def active_record_totals(periods, time:)

data/lib/llm_cost_tracker/capture_verifier.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require_relative "storage/dispatcher"
+require_relative "storage/active_record_backend"
 
 module LlmCostTracker
   class CaptureVerifier
@@ -48,20 +48,13 @@ module LlmCostTracker
         ]
       end
 
-      LlmCostTracker::Integrations.checks.map do |check|
+      LlmCostTracker::Integrations::Registry.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|
+      LlmCostTracker::Storage::ActiveRecordBackend.verify.map do |check|
         Check.new(check.status, check.name, check.message)
       end
     rescue LlmCostTracker::Error => e

data/lib/llm_cost_tracker/configuration.rb

@@ -4,21 +4,18 @@ require_relative "errors"
 require_relative "tag_key"
 require_relative "value_helpers"
 require_relative "configuration/instrumentation"
-require_relative "configuration/storage_backend"
 
 module LlmCostTracker
   class Configuration
     include ConfigurationInstrumentation
-    include ConfigurationStorageBackend
 
     OPENAI_COMPATIBLE_PROVIDERS = { "openrouter.ai" => "openrouter", "api.deepseek.com" => "deepseek" }.freeze
 
     BUDGET_EXCEEDED_BEHAVIORS = %i[notify raise block_requests].freeze
     STORAGE_ERROR_BEHAVIORS = %i[ignore warn raise].freeze
-    STORAGE_BACKENDS = %i[log active_record custom].freeze
     UNKNOWN_PRICING_BEHAVIORS = %i[ignore warn raise].freeze
-    SHARED_SCALAR_ATTRIBUTES = %i[enabled custom_storage on_budget_exceeded monthly_budget daily_budget per_call_budget
-                                  log_level prices_file max_tag_count max_tag_value_bytesize].freeze
+    SHARED_SCALAR_ATTRIBUTES = %i[enabled on_budget_exceeded monthly_budget daily_budget per_call_budget log_level
+                                  prices_file max_tag_count max_tag_value_bytesize].freeze
     SHARED_ENUM_ATTRIBUTES = {
       budget_exceeded_behavior: [BUDGET_EXCEEDED_BEHAVIORS, :notify],
       storage_error_behavior: [STORAGE_ERROR_BEHAVIORS, :warn],
@@ -34,7 +31,6 @@ module LlmCostTracker
       :instrumented_integrations,
       :report_tag_breakdowns,
       :redacted_tag_keys,
-      :storage_backend,
       :storage_error_behavior,
       :unknown_pricing_behavior,
       :openai_compatible_providers
@@ -42,8 +38,6 @@ module LlmCostTracker
 
     def initialize
       @enabled = true
-      self.storage_backend = :log
-      @custom_storage     = nil
       @default_tags       = {}
       @on_budget_exceeded = nil
       @monthly_budget     = nil
@@ -134,8 +128,6 @@ module LlmCostTracker
       copy
     end
 
-    def active_record? = storage_backend == :active_record
-
     private
 
     def normalize_enum(name, value, allowed, default:)

data/lib/llm_cost_tracker/doctor/ingestion_check.rb

@@ -16,7 +16,7 @@ module LlmCostTracker
       end
 
       def call
-        return unless active_record_storage? && llm_api_calls_table?
+        return unless llm_api_calls_table?
 
         missing = missing_parts
         if missing.empty?
@@ -56,8 +56,6 @@ module LlmCostTracker
         ].compact
       end
 
-      def active_record_storage? = LlmCostTracker.configuration.storage_backend == :active_record
-
       def llm_api_calls_table? = table_exists?("llm_api_calls")
 
       def table_exists?(name)

data/lib/llm_cost_tracker/doctor.rb

@@ -40,7 +40,7 @@ module LlmCostTracker
     def checks
       [
         configuration_check,
-        capture_check,
+        CaptureCheck.call(Check),
         *integration_checks,
         active_record_check,
         table_check,
@@ -56,26 +56,23 @@ module LlmCostTracker
 
     def configuration_check
       config = LlmCostTracker.configuration
-      Check.new(:ok, "configuration", "storage_backend=#{config.storage_backend.inspect}, enabled=#{config.enabled}")
+      Check.new(:ok, "configuration", "active_record ledger enabled=#{config.enabled}")
     end
 
-    def capture_check = CaptureCheck.call(Check)
-
     def integration_checks
-      LlmCostTracker::Integrations.checks.map do |check|
+      LlmCostTracker::Integrations::Registry.checks.map do |check|
         Check.new(check.status, check.name.to_s, check.message)
       end
     end
 
     def active_record_check
-      return Check.new(:ok, "storage", "ActiveRecord storage is disabled") unless active_record_storage?
       return Check.new(:ok, "active_record", "available") if active_record_available?
 
-      Check.new(:error, "active_record", "unavailable; add ActiveRecord/Rails or change storage_backend")
+      Check.new(:error, "active_record", "unavailable")
     end
 
     def table_check
-      return unless active_record_storage? && active_record_available?
+      return unless active_record_available?
       return Check.new(:ok, "llm_api_calls", "table exists") if llm_api_calls_table?
 
       Check.new(
@@ -86,7 +83,7 @@ module LlmCostTracker
     end
 
     def column_check
-      return unless active_record_storage? && llm_api_calls_table?
+      return unless llm_api_calls_table?
 
       columns = column_names("llm_api_calls")
       missing_core = CORE_COLUMNS - columns
@@ -106,7 +103,7 @@ module LlmCostTracker
     end
 
     def period_totals_check
-      return unless active_record_storage? && llm_api_calls_table?
+      return unless llm_api_calls_table?
       if table_exists?("llm_cost_tracker_period_totals")
         return Check.new(:ok, "period totals", "llm_cost_tracker_period_totals exists")
       end
@@ -133,7 +130,7 @@ module LlmCostTracker
     end
 
     def calls_check
-      return unless active_record_storage? && llm_api_calls_table?
+      return unless llm_api_calls_table?
 
       count = LlmCostTracker::LlmApiCall.count
       return Check.new(:warn, "tracked calls", "none recorded yet") if count.zero?
@@ -142,8 +139,6 @@ module LlmCostTracker
       Check.new(:ok, "tracked calls", "#{count} recorded; latest #{latest}")
     end
 
-    def active_record_storage? = LlmCostTracker.configuration.storage_backend == :active_record
-
     def active_record_available?
       require_relative "llm_api_call" unless defined?(LlmCostTracker::LlmApiCall)
       LlmCostTracker::LlmApiCall.connection

data/lib/llm_cost_tracker/engine.rb

@@ -2,12 +2,9 @@
 
 require "rails"
 require_relative "../llm_cost_tracker"
-require_relative "engine_compatibility"
 require_relative "assets"
 require "rack/files"
 
-LlmCostTracker::EngineCompatibility.check_rails_version!(Rails.version)
-
 module LlmCostTracker
   class Engine < ::Rails::Engine
     isolate_namespace LlmCostTracker

data/lib/llm_cost_tracker/generators/llm_cost_tracker/templates/add_period_totals_to_llm_cost_tracker.rb.erb

@@ -80,7 +80,7 @@ class AddPeriodTotalsToLlmCostTracker < ActiveRecord::Migration<%= migration_ver
     elsif mysql?
       "DATE(tracked_at)"
     else
-      "date(tracked_at)"
+      raise "LLM Cost Tracker supports PostgreSQL and MySQL only"
     end
   end
 
@@ -90,7 +90,7 @@ class AddPeriodTotalsToLlmCostTracker < ActiveRecord::Migration<%= migration_ver
     elsif mysql?
       "DATE_FORMAT(tracked_at, '%Y-%m-01')"
     else
-      "strftime('%Y-%m-01', tracked_at)"
+      raise "LLM Cost Tracker supports PostgreSQL and MySQL only"
     end
   end
 

data/lib/llm_cost_tracker/generators/llm_cost_tracker/templates/create_llm_api_calls.rb.erb

@@ -24,8 +24,10 @@ class CreateLlmApiCalls < ActiveRecord::Migration<%= migration_version %>
       t.string  :pricing_mode
       if postgresql?
         t.jsonb :tags, null: false, default: {}
+      elsif mysql?
+        t.json :tags, null: false
       else
-        t.text :tags
+        raise "LLM Cost Tracker supports PostgreSQL and MySQL only"
       end
       t.datetime :tracked_at,   null: false
 
@@ -79,4 +81,8 @@ class CreateLlmApiCalls < ActiveRecord::Migration<%= migration_version %>
   def postgresql?
     LlmCostTracker::ActiveRecordAdapter.postgresql?(connection)
   end
+
+  def mysql?
+    LlmCostTracker::ActiveRecordAdapter.mysql?(connection)
+  end
 end

data/lib/llm_cost_tracker/generators/llm_cost_tracker/templates/initializer.rb.erb

@@ -4,10 +4,6 @@ LlmCostTracker.configure do |config|
   # Set to false to temporarily disable tracking without removing middleware.
   config.enabled = true
 
-  # :active_record captures events into a durable inbox and ingests them into llm_api_calls.
-  # Other options: :log for local logging, :custom for your own storage callable.
-  config.storage_backend = :active_record
-
   # Tags are merged into every event. Use a callable for request/job-time context.
   config.default_tags = -> { { environment: Rails.env } }
 
@@ -34,7 +30,7 @@ LlmCostTracker.configure do |config|
   # every model must have known pricing before it can be used.
   config.unknown_pricing_behavior = :warn
 
-  # Used only by the :log storage backend.
+  # LLM Cost Tracker logs warnings through Rails.logger when available.
   config.log_level = :info
 <% if options[:prices] -%>
 
@@ -65,16 +61,4 @@ LlmCostTracker.configure do |config|
   # for bin/rails llm_cost_tracker:report.
   # config.openai_compatible_providers["llm.my-company.com"] = "internal_gateway"
   # config.report_tag_breakdowns = %w[feature user_id]
-
-  # Use :custom when you want to send events to your own sink instead of ActiveRecord.
-  # Return false from custom_storage to skip budget checks for that event.
-  # config.storage_backend = :custom
-  # config.custom_storage = ->(event) {
-  #   Rails.logger.info(
-  #     provider: event.provider,
-  #     model: event.model,
-  #     total_cost: event.cost&.total_cost,
-  #     tags: event.tags
-  #   )
-  # }
 end

data/lib/llm_cost_tracker/integrations/registry.rb

@@ -67,7 +67,5 @@ module LlmCostTracker
     end
 
     def self.register(name, integration) = Registry.register(name, integration)
-    def self.install! = Registry.install!
-    def self.checks = Registry.checks
   end
 end

data/lib/llm_cost_tracker/period_grouping.rb

@@ -7,13 +7,11 @@ module LlmCostTracker
     PERIOD_FORMATS = {
       day: {
         postgres: "YYYY-MM-DD",
-        mysql: "%Y-%m-%d",
-        sqlite: "%Y-%m-%d"
+        mysql: "%Y-%m-%d"
       },
       month: {
         postgres: "YYYY-MM",
-        mysql: "%Y-%m",
-        sqlite: "%Y-%m"
+        mysql: "%Y-%m"
       }
     }.freeze
 
@@ -41,7 +39,7 @@ module LlmCostTracker
       elsif ActiveRecordAdapter.mysql?(connection)
         "DATE_FORMAT(#{column}, #{connection.quote(formats.fetch(:mysql))})"
       else
-        "strftime(#{connection.quote(formats.fetch(:sqlite))}, #{column})"
+        ActiveRecordAdapter.ensure_supported!(connection)
       end
     end
 

data/lib/llm_cost_tracker/railtie.rb

@@ -21,10 +21,8 @@ module LlmCostTracker
 
     initializer "llm_cost_tracker.configure" do
       ActiveSupport.on_load(:active_record) do
-        if LlmCostTracker.configuration.active_record?
-          require_relative "llm_api_call"
-          require_relative "storage/active_record_store"
-        end
+        require_relative "llm_api_call"
+        require_relative "storage/active_record_store"
       end
     end
   end

data/lib/llm_cost_tracker/report.rb

@@ -5,10 +5,8 @@ require_relative "report_formatter"
 
 module LlmCostTracker
   class Report
-    DEFAULT_DAYS = ReportData::DEFAULT_DAYS
-
     class << self
-      def generate(days: DEFAULT_DAYS, now: Time.now.utc, tag_breakdowns: nil)
+      def generate(days: ReportData::DEFAULT_DAYS, now: Time.now.utc, tag_breakdowns: nil)
         report_data = ReportData.build(
           days: days,
           now: now,
@@ -23,7 +21,7 @@ module LlmCostTracker
         "Unable to build LLM cost report: #{e.class}: #{e.message}"
       end
 
-      def data(days: DEFAULT_DAYS, now: Time.now.utc, tag_breakdowns: nil)
+      def data(days: ReportData::DEFAULT_DAYS, now: Time.now.utc, tag_breakdowns: nil)
         ReportData.build(days: days, now: now, tag_breakdowns: tag_breakdowns)
       end
     end

data/lib/llm_cost_tracker/storage/active_record_backend.rb

@@ -2,13 +2,14 @@
 
 require "securerandom"
 
-require_relative "registry"
 require_relative "active_record_inbox"
 require_relative "active_record_ingestor"
 require_relative "active_record_store"
 
 module LlmCostTracker
   module Storage
+    VerificationResult = Data.define(:status, :name, :message)
+
     class ActiveRecordBackend
       VERIFY_TAG = "llm_cost_tracker_verify"
 

data/lib/llm_cost_tracker/storage/active_record_inbox.rb

@@ -3,7 +3,6 @@
 require "json"
 require "time"
 
-require_relative "../active_record_adapter"
 require_relative "../cost"
 require_relative "../event"
 require_relative "../inbox_event"
@@ -110,7 +109,7 @@ module LlmCostTracker
 
         def insert_row(row)
           connection = LlmCostTracker::LlmApiCall.connection
-          if connection.transaction_open? && !sqlite_database?(connection)
+          if connection.transaction_open?
             insert_with_separate_connection(row)
           else
             execute_insert(connection, row)
@@ -155,10 +154,6 @@ module LlmCostTracker
         def symbolize_keys(hash)
           hash.transform_keys(&:to_sym)
         end
-
-        def sqlite_database?(connection)
-          ActiveRecordAdapter.sqlite?(connection)
-        end
       end
     end
   end

data/lib/llm_cost_tracker/storage/active_record_rollup_upsert_sql.rb

@@ -17,7 +17,7 @@ module LlmCostTracker
         return Arel.sql(mysql_sql) if ActiveRecordAdapter.mysql?(connection)
         return Arel.sql(postgres_sql) if ActiveRecordAdapter.postgresql?(connection)
 
-        Arel.sql("total_cost = total_cost + excluded.total_cost, updated_at = excluded.updated_at")
+        ActiveRecordAdapter.ensure_supported!(connection)
       end
 
       private

data/lib/llm_cost_tracker/storage/{dispatcher.rb → writer.rb}

@@ -1,17 +1,15 @@
 # frozen_string_literal: true
 
+require_relative "../errors"
 require_relative "../logging"
-require_relative "registry"
 require_relative "active_record_backend"
-require_relative "custom_backend"
-require_relative "log_backend"
 
 module LlmCostTracker
   module Storage
-    class Dispatcher
+    class Writer
       class << self
         def save(event)
-          backend.save(event)
+          ActiveRecordBackend.save(event)
         rescue LlmCostTracker::BudgetExceededError, LlmCostTracker::UnknownPricingError
           raise
         rescue StandardError => e
@@ -21,25 +19,17 @@ module LlmCostTracker
 
         private
 
-        def backend
-          Registry.fetch(LlmCostTracker.configuration.storage_backend)
-        end
-
         def handle_error(error)
           case LlmCostTracker.configuration.storage_error_behavior
           when :ignore
             nil
           when :warn
-            Logging.warn("Storage failed; tracking event was not persisted: #{error.class}: #{error.message}")
+            Logging.warn("ActiveRecord ledger write failed: #{error.class}: #{error.message}")
           when :raise
             raise StorageError, error
           end
         end
       end
     end
-
-    Registry.register(:log, LogBackend)
-    Registry.register(:active_record, ActiveRecordBackend)
-    Registry.register(:custom, CustomBackend)
   end
 end

data/lib/llm_cost_tracker/tag_sql.rb

@@ -16,7 +16,7 @@ module LlmCostTracker
         elsif ActiveRecordAdapter.mysql?(model.connection)
           "JSON_UNQUOTE(JSON_EXTRACT(#{column}, #{model.connection.quote(json_path(key))}))"
         else
-          "json_extract(#{column}, #{model.connection.quote(json_path(key))})"
+          ActiveRecordAdapter.ensure_supported!(model.connection)
         end
       end
 

data/lib/llm_cost_tracker/tags_column.rb

@@ -79,6 +79,8 @@ module LlmCostTracker
     end
 
     def build_lct_schema_capabilities(columns, adapter_name)
+      ActiveRecordAdapter.ensure_supported!(adapter_name)
+
       tag_column = columns["tags"]
       tags_jsonb = tag_column && (tag_column.type == :jsonb || tag_column.sql_type.to_s.downcase == "jsonb")
       tags_mysql_json =

data/lib/llm_cost_tracker/tracker.rb

@@ -2,7 +2,7 @@
 
 require "securerandom"
 
-require_relative "storage/dispatcher"
+require_relative "storage/writer"
 
 module LlmCostTracker
   class Tracker
@@ -41,7 +41,7 @@ module LlmCostTracker
 
         ActiveSupport::Notifications.instrument(EVENT_NAME, event.to_h)
 
-        stored = Storage::Dispatcher.save(event)
+        stored = Storage::Writer.save(event)
         Budget.check!(event) unless stored == false
 
         event