llm_cost_tracker 0.7.0 → 0.7.1

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: llm_cost_tracker
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  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
@@ -248,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
@@ -279,26 +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/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/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
@@ -306,82 +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/writer.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

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,53 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "errors"
-
-module LlmCostTracker
-  module ActiveRecordAdapter
-    MYSQL_ADAPTERS = %w[
-      ActiveRecord::ConnectionAdapters::AbstractMysqlAdapter
-      ActiveRecord::ConnectionAdapters::Mysql2Adapter
-      ActiveRecord::ConnectionAdapters::TrilogyAdapter
-    ].freeze
-    POSTGRESQL_ADAPTERS = %w[
-      ActiveRecord::ConnectionAdapters::PostgreSQLAdapter
-    ].freeze
-    MYSQL_PATTERN = /mysql|trilogy|mariadb/i
-    POSTGRESQL_PATTERN = /postgres/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 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
-
-      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,64 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "storage/active_record_backend"
-
-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::Registry.checks.map do |check|
-        Check.new(check.status, "sdk integration #{check.name}", check.message)
-      end
-    end
-
-    def storage_checks
-      LlmCostTracker::Storage::ActiveRecordBackend.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/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/event_metadata.rb

@@ -1,52 +0,0 @@
-# frozen_string_literal: true
-
-module LlmCostTracker
-  module EventMetadata
-    INTERNAL_TAG_KEYS = %w[
-      cache_read_input_tokens
-      cache_write_input_tokens
-      hidden_output_tokens
-      input_tokens
-      output_tokens
-      pricing_mode
-      provider_response_id
-      total_tokens
-    ].freeze
-
-    class << self
-      def usage_data(input_tokens, output_tokens, metadata)
-        metadata = metadata.to_h.symbolize_keys
-        cache_read = first_integer(metadata, :cache_read_input_tokens)
-        cache_write = first_integer(metadata, :cache_write_input_tokens)
-        hidden_output = first_integer(metadata, :hidden_output_tokens)
-        breakdown = UsageBreakdown.build(
-          input_tokens: input_tokens,
-          output_tokens: output_tokens,
-          cache_read_input_tokens: cache_read,
-          cache_write_input_tokens: cache_write,
-          hidden_output_tokens: hidden_output
-        )
-
-        breakdown.to_h.merge(pricing_mode: normalized_pricing_mode(metadata[:pricing_mode])).compact
-      end
-
-      def tags(metadata)
-        metadata.reject { |key, _value| INTERNAL_TAG_KEYS.include?(key.to_s) }
-      end
-
-      private
-
-      def first_integer(metadata, *keys)
-        keys.each { |key| return metadata[key].to_i unless metadata[key].nil? }
-        0
-      end
-
-      def normalized_pricing_mode(value)
-        return nil if value.nil?
-
-        mode = value.to_s.strip
-        mode.empty? || mode == "standard" ? nil : mode
-      end
-    end
-  end
-end

data/lib/llm_cost_tracker/generators/llm_cost_tracker/add_usage_breakdown_generator.rb

@@ -1,29 +0,0 @@
-# frozen_string_literal: true
-
-require "rails/generators"
-require "rails/generators/active_record"
-
-module LlmCostTracker
-  module Generators
-    class AddUsageBreakdownGenerator < Rails::Generators::Base
-      include ActiveRecord::Generators::Migration
-
-      source_root File.expand_path("templates", __dir__)
-
-      desc "Creates a migration to add usage and cost breakdown columns to llm_api_calls"
-
-      def create_migration_file
-        migration_template(
-          "add_usage_breakdown_to_llm_api_calls.rb.erb",
-          "db/migrate/add_usage_breakdown_to_llm_api_calls.rb"
-        )
-      end
-
-      private
-
-      def migration_version
-        "[#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}]"
-      end
-    end
-  end
-end

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

@@ -1,29 +0,0 @@
-class AddUsageBreakdownToLlmApiCalls < ActiveRecord::Migration<%= migration_version %>
-  def up
-    unless column_exists?(:llm_api_calls, :cache_read_input_tokens)
-      add_column :llm_api_calls, :cache_read_input_tokens, :integer, null: false, default: 0
-    end
-    unless column_exists?(:llm_api_calls, :cache_write_input_tokens)
-      add_column :llm_api_calls, :cache_write_input_tokens, :integer, null: false, default: 0
-    end
-    unless column_exists?(:llm_api_calls, :hidden_output_tokens)
-      add_column :llm_api_calls, :hidden_output_tokens, :integer, null: false, default: 0
-    end
-    unless column_exists?(:llm_api_calls, :cache_read_input_cost)
-      add_column :llm_api_calls, :cache_read_input_cost, :decimal, precision: 20, scale: 8
-    end
-    unless column_exists?(:llm_api_calls, :cache_write_input_cost)
-      add_column :llm_api_calls, :cache_write_input_cost, :decimal, precision: 20, scale: 8
-    end
-    add_column :llm_api_calls, :pricing_mode, :string unless column_exists?(:llm_api_calls, :pricing_mode)
-  end
-
-  def down
-    remove_column :llm_api_calls, :pricing_mode if column_exists?(:llm_api_calls, :pricing_mode)
-    remove_column :llm_api_calls, :cache_write_input_cost if column_exists?(:llm_api_calls, :cache_write_input_cost)
-    remove_column :llm_api_calls, :cache_read_input_cost if column_exists?(:llm_api_calls, :cache_read_input_cost)
-    remove_column :llm_api_calls, :hidden_output_tokens if column_exists?(:llm_api_calls, :hidden_output_tokens)
-    remove_column :llm_api_calls, :cache_write_input_tokens if column_exists?(:llm_api_calls, :cache_write_input_tokens)
-    remove_column :llm_api_calls, :cache_read_input_tokens if column_exists?(:llm_api_calls, :cache_read_input_tokens)
-  end
-end

data/lib/llm_cost_tracker/inbox_event.rb

@@ -1,9 +0,0 @@
-# frozen_string_literal: true
-
-require "active_record"
-
-module LlmCostTracker
-  class InboxEvent < ActiveRecord::Base
-    self.table_name = "llm_cost_tracker_inbox_events"
-  end
-end

data/lib/llm_cost_tracker/ingestor_lease.rb

@@ -1,9 +0,0 @@
-# frozen_string_literal: true
-
-require "active_record"
-
-module LlmCostTracker
-  class IngestorLease < ActiveRecord::Base
-    self.table_name = "llm_cost_tracker_ingestor_leases"
-  end
-end