llm_cost_tracker 0.7.2 → 0.8.0

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: llm_cost_tracker
 version: !ruby/object:Gem::Version
-  version: 0.7.2
+  version: 0.8.0
 platform: ruby
 authors:
 - Sergii Khomenko
-autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-05-01 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -216,11 +215,9 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         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 Rails SDK integrations, Faraday middleware, or explicit track/track_stream
-  helpers, with ActiveRecord storage, tag-based attribution, price sync tasks, and
-  budget guardrails.
+description: 'Logs every call your Rails app makes to OpenAI, Anthropic, Gemini, RubyLLM,
+  or an OpenAI-compatible API: tokens, cost, latency, tags. Calls go straight to the
+  provider — no proxy. Includes price sync, budget guardrails, and a mountable dashboard.'
 email:
 - sergey@mm.st
 executables: []
@@ -228,6 +225,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".rspec"
+- ".ruby-version"
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
 - LICENSE.txt
@@ -249,13 +247,13 @@ files:
 - 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/call.rb
+- app/models/llm_cost_tracker/call_line_item.rb
+- app/models/llm_cost_tracker/call_rollup.rb
+- app/models/llm_cost_tracker/call_tag.rb
+- app/models/llm_cost_tracker/ingestion/inbox_entry.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/models/llm_cost_tracker/provider_invoice.rb
 - app/services/llm_cost_tracker/dashboard/data_quality.rb
 - app/services/llm_cost_tracker/dashboard/date_range.rb
 - app/services/llm_cost_tracker/dashboard/filter.rb
@@ -279,7 +277,9 @@ files:
 - app/views/llm_cost_tracker/models/index.html.erb
 - app/views/llm_cost_tracker/shared/_active_filters.html.erb
 - app/views/llm_cost_tracker/shared/_bar.html.erb
+- app/views/llm_cost_tracker/shared/_filters.html.erb
 - app/views/llm_cost_tracker/shared/_metric_stack.html.erb
+- app/views/llm_cost_tracker/shared/_sort.html.erb
 - app/views/llm_cost_tracker/shared/_spend_chart.html.erb
 - app/views/llm_cost_tracker/shared/_tag_chips.html.erb
 - app/views/llm_cost_tracker/shared/setup_required.html.erb
@@ -288,40 +288,33 @@ files:
 - config/routes.rb
 - lib/llm_cost_tracker.rb
 - lib/llm_cost_tracker/assets.rb
+- lib/llm_cost_tracker/billing/components.rb
+- lib/llm_cost_tracker/billing/components.yml
+- lib/llm_cost_tracker/billing/cost_status.rb
+- lib/llm_cost_tracker/billing/line_item.rb
 - lib/llm_cost_tracker/budget.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/doctor.rb
 - lib/llm_cost_tracker/doctor/capture_verifier.rb
 - lib/llm_cost_tracker/doctor/check.rb
+- lib/llm_cost_tracker/doctor/cost_drift_check.rb
 - lib/llm_cost_tracker/doctor/ingestion_check.rb
+- lib/llm_cost_tracker/doctor/legacy_audit_check.rb
+- lib/llm_cost_tracker/doctor/legacy_billing_status_check.rb
 - lib/llm_cost_tracker/doctor/price_check.rb
+- lib/llm_cost_tracker/doctor/pricing_snapshot_drift_check.rb
+- lib/llm_cost_tracker/doctor/probe.rb
+- lib/llm_cost_tracker/doctor/schema_check.rb
 - lib/llm_cost_tracker/engine.rb
 - lib/llm_cost_tracker/errors.rb
 - lib/llm_cost_tracker/event.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_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
-- lib/llm_cost_tracker/generators/llm_cost_tracker/templates/add_latency_ms_to_llm_api_calls.rb.erb
-- 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_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/create_llm_cost_tracker_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/ingestion.rb
 - lib/llm_cost_tracker/ingestion/batch.rb
 - lib/llm_cost_tracker/ingestion/inbox.rb
@@ -336,11 +329,13 @@ files:
 - 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/call_line_items.rb
+- lib/llm_cost_tracker/ledger/schema/call_rollups.rb
+- lib/llm_cost_tracker/ledger/schema/call_tags.rb
 - lib/llm_cost_tracker/ledger/schema/calls.rb
-- lib/llm_cost_tracker/ledger/schema/period_totals.rb
+- lib/llm_cost_tracker/ledger/schema/provider_invoices.rb
 - lib/llm_cost_tracker/ledger/store.rb
 - lib/llm_cost_tracker/ledger/tags/query.rb
 - lib/llm_cost_tracker/ledger/tags/sql.rb
@@ -352,20 +347,21 @@ files:
 - 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_service_charges.rb
 - lib/llm_cost_tracker/parsers/openai_usage.rb
 - lib/llm_cost_tracker/parsers/sse.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/service_charges.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/sync_change_printer.rb
 - lib/llm_cost_tracker/pricing/unknown.rb
 - lib/llm_cost_tracker/railtie.rb
 - lib/llm_cost_tracker/report.rb
@@ -375,6 +371,7 @@ files:
 - 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/timing.rb
 - lib/llm_cost_tracker/token_usage.rb
 - lib/llm_cost_tracker/tracker.rb
 - lib/llm_cost_tracker/usage_capture.rb
@@ -389,7 +386,6 @@ metadata:
   source_code_uri: https://github.com/sergey-homenko/llm_cost_tracker
   documentation_uri: https://github.com/sergey-homenko/llm_cost_tracker#readme
   rubygems_mfa_required: 'true'
-post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -397,15 +393,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.3.0
+      version: 3.4.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.22
-signing_key:
+rubygems_version: 3.6.9
 specification_version: 4
-summary: Rails-native LLM usage and cost tracking with ActiveRecord storage
+summary: LLM API cost tracking for Rails applications
 test_files: []

data/app/models/llm_cost_tracker/ingestion/event.rb

@@ -1,13 +0,0 @@
-# frozen_string_literal: true
-
-require "active_record"
-
-module LlmCostTracker
-  module Ingestion
-    class Event < ActiveRecord::Base
-      MAX_ATTEMPTS = 5
-
-      self.table_name = "llm_cost_tracker_inbox_events"
-    end
-  end
-end

data/app/models/llm_cost_tracker/ledger/call.rb

@@ -1,45 +0,0 @@
-# frozen_string_literal: true
-
-require "active_record"
-
-module LlmCostTracker
-  module Ledger
-    class Call < ActiveRecord::Base
-      extend Period::Grouping
-      extend Ledger::CallMetrics
-      include Ledger::Tags::Accessors
-
-      self.table_name = "llm_api_calls"
-
-      scope :with_cost, -> { where.not(total_cost: nil) }
-      scope :without_cost, -> { where(total_cost: nil) }
-      scope :unknown_pricing, -> { without_cost }
-      scope :with_latency, -> { where.not(latency_ms: nil) }
-      scope :streaming,     -> { where(stream: true) }
-      scope :non_streaming, -> { where(stream: [false, nil]) }
-      scope :by_usage_source, ->(source) { where(usage_source: source.to_s) }
-      scope :with_provider_response_id, -> { where.not(provider_response_id: [nil, ""]) }
-      scope :missing_provider_response_id, -> { where(provider_response_id: [nil, ""]) }
-      scope :streaming_missing_usage, lambda {
-        where(stream: true).where(usage_source: ["unknown", nil])
-      }
-
-      scope :with_json_tags, lambda {
-        where.not(tags: {})
-      }
-
-      scope :today,       -> { where(tracked_at: Time.now.utc.beginning_of_day..) }
-      scope :this_week,   -> { where(tracked_at: Time.now.utc.beginning_of_week..) }
-      scope :this_month,  -> { where(tracked_at: Time.now.utc.beginning_of_month..) }
-      scope :between,     ->(from, to) { where(tracked_at: from..to) }
-
-      def self.by_tag(key, value)
-        by_tags(key => value)
-      end
-
-      def self.by_tags(tags)
-        Ledger::Tags::Query.apply(self, tags)
-      end
-    end
-  end
-end

data/app/models/llm_cost_tracker/ledger/call_metrics.rb

@@ -1,66 +0,0 @@
-# frozen_string_literal: true
-
-require "llm_cost_tracker/ledger/tags/sql"
-
-module LlmCostTracker
-  module Ledger
-    module CallMetrics
-      def total_cost
-        sum(:total_cost).to_f
-      end
-
-      def total_tokens
-        sum(:total_tokens).to_i
-      end
-
-      def cost_by_model(limit: nil)
-        cost_by_column(:model, limit: limit)
-      end
-
-      def cost_by_provider(limit: nil)
-        cost_by_column(:provider, limit: limit)
-      end
-
-      def group_by_tag(key)
-        group(Arel.sql(tag_value_expression(key)))
-      end
-
-      def cost_by_tag(key, limit: nil)
-        expression = tag_value_expression(key)
-        label_expression = "COALESCE(NULLIF(#{expression}, ''), #{connection.quote('(untagged)')})"
-        relation = select("#{label_expression} AS name, COALESCE(SUM(total_cost), 0) AS total_cost")
-                   .group(Arel.sql(label_expression))
-                   .order(Arel.sql("COALESCE(SUM(total_cost), 0) DESC"))
-        relation = relation.limit(limit) if limit
-        relation
-      end
-
-      def average_latency_ms
-        average(:latency_ms)&.to_f
-      end
-
-      def latency_by_model
-        group(:model).average(:latency_ms).transform_values(&:to_f)
-      end
-
-      def latency_by_provider
-        group(:provider).average(:latency_ms).transform_values(&:to_f)
-      end
-
-      def tag_value_expression(key, table_name: quoted_table_name)
-        Ledger::Tags::Sql.value_expression(self, key, table_name: table_name)
-      end
-
-      private
-
-      def cost_by_column(column, limit:)
-        quoted_column = "#{quoted_table_name}.#{connection.quote_column_name(column)}"
-        relation = select("#{quoted_column} AS name, COALESCE(SUM(total_cost), 0) AS total_cost")
-                   .group(column)
-                   .order(Arel.sql("COALESCE(SUM(total_cost), 0) DESC"))
-        relation = relation.limit(limit) if limit
-        relation
-      end
-    end
-  end
-end

data/app/models/llm_cost_tracker/ledger/period/grouping.rb

@@ -1,71 +0,0 @@
-# frozen_string_literal: true
-
-require "llm_cost_tracker/ledger/schema/adapter"
-
-module LlmCostTracker
-  module Ledger
-    module Period
-      module Grouping
-        PERIOD_FORMATS = {
-          day: {
-            postgres: "YYYY-MM-DD",
-            mysql: "%Y-%m-%d"
-          },
-          month: {
-            postgres: "YYYY-MM",
-            mysql: "%Y-%m"
-          }
-        }.freeze
-
-        private_constant :PERIOD_FORMATS
-
-        def group_by_period(period, column: :tracked_at)
-          group(Arel.sql(period_group_expression(period, column: column)))
-        end
-
-        def daily_costs(days: 30)
-          where(tracked_at: days.days.ago..)
-            .group_by_period(:day)
-            .sum(:total_cost)
-        end
-
-        private
-
-        def period_group_expression(period, column:)
-          period = validated_period(period)
-          column = period_column_expression(column)
-          formats = PERIOD_FORMATS.fetch(period)
-
-          if Ledger::Schema::Adapter.postgresql?(connection)
-            postgres_period_expression(period, column, formats)
-          elsif Ledger::Schema::Adapter.mysql?(connection)
-            "DATE_FORMAT(#{column}, #{connection.quote(formats.fetch(:mysql))})"
-          else
-            Ledger::Schema::Adapter.ensure_supported!(connection)
-          end
-        end
-
-        def postgres_period_expression(period, column, formats)
-          "TO_CHAR(" \
-            "DATE_TRUNC(#{connection.quote(period.to_s)}, #{column}), " \
-            "#{connection.quote(formats.fetch(:postgres))}" \
-            ")"
-        end
-
-        def validated_period(period)
-          normalized_period = period.try(:to_sym)
-          return normalized_period if PERIOD_FORMATS.key?(normalized_period)
-
-          raise ArgumentError, "invalid period: #{period.inspect}"
-        end
-
-        def period_column_expression(column)
-          column = column.to_s
-          return "#{quoted_table_name}.#{connection.quote_column_name(column)}" if column_names.include?(column)
-
-          raise ArgumentError, "invalid period column: #{column.inspect}"
-        end
-      end
-    end
-  end
-end

data/app/models/llm_cost_tracker/ledger/period/total.rb

@@ -1,13 +0,0 @@
-# frozen_string_literal: true
-
-require "active_record"
-
-module LlmCostTracker
-  module Ledger
-    module Period
-      class Total < ActiveRecord::Base
-        self.table_name = "llm_cost_tracker_period_totals"
-      end
-    end
-  end
-end

data/app/models/llm_cost_tracker/ledger/tags/accessors.rb

@@ -1,19 +0,0 @@
-# frozen_string_literal: true
-
-require "json"
-
-module LlmCostTracker
-  module Ledger
-    module Tags
-      module Accessors
-        def parsed_tags
-          return tags.transform_keys(&:to_s) if tags.is_a?(Hash)
-
-          JSON.parse(tags || "{}")
-        rescue JSON::ParserError
-          {}
-        end
-      end
-    end
-  end
-end

data/lib/llm_cost_tracker/configuration/instrumentation.rb

@@ -1,33 +0,0 @@
-# frozen_string_literal: true
-
-module LlmCostTracker
-  module ConfigurationInstrumentation
-    def instrument(*names)
-      ensure_shared_configuration_mutable!
-      @instrumented_integrations = (@instrumented_integrations + normalize_instrumentation_names(names)).uniq
-    end
-
-    def instrumented?(name)
-      @instrumented_integrations.include?(name.to_sym)
-    end
-
-    private
-
-    def normalize_instrumentation_names(names)
-      names.flatten.flat_map do |name|
-        key = name.to_sym
-        next Integrations.names if key == :all
-
-        validate_instrumentation_name!(key)
-        key
-      end
-    end
-
-    def validate_instrumentation_name!(name)
-      return if Integrations.names.include?(name)
-
-      raise Error, "Unknown integration: #{name.inspect}. " \
-                   "Use one of: #{Integrations.names.join(', ')}"
-    end
-  end
-end

data/lib/llm_cost_tracker/generators/llm_cost_tracker/add_ingestion_generator.rb

@@ -1,29 +0,0 @@
-# frozen_string_literal: true
-
-require "rails/generators"
-require "rails/generators/active_record"
-
-module LlmCostTracker
-  module Generators
-    class AddIngestionGenerator < Rails::Generators::Base
-      include ActiveRecord::Generators::Migration
-
-      source_root File.expand_path("templates", __dir__)
-
-      desc "Creates a migration to add durable ActiveRecord ingestion"
-
-      def create_migration_file
-        migration_template(
-          "add_ingestion_to_llm_cost_tracker.rb.erb",
-          "db/migrate/add_ingestion_to_llm_cost_tracker.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/add_latency_ms_generator.rb

@@ -1,29 +0,0 @@
-# frozen_string_literal: true
-
-require "rails/generators"
-require "rails/generators/active_record"
-
-module LlmCostTracker
-  module Generators
-    class AddLatencyMsGenerator < Rails::Generators::Base
-      include ActiveRecord::Generators::Migration
-
-      source_root File.expand_path("templates", __dir__)
-
-      desc "Creates a migration to add llm_api_calls.latency_ms"
-
-      def create_migration_file
-        migration_template(
-          "add_latency_ms_to_llm_api_calls.rb.erb",
-          "db/migrate/add_latency_ms_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/add_period_totals_generator.rb

@@ -1,29 +0,0 @@
-# frozen_string_literal: true
-
-require "rails/generators"
-require "rails/generators/active_record"
-
-module LlmCostTracker
-  module Generators
-    class AddPeriodTotalsGenerator < Rails::Generators::Base
-      include ActiveRecord::Generators::Migration
-
-      source_root File.expand_path("templates", __dir__)
-
-      desc "Creates a migration to add llm_cost_tracker_period_totals"
-
-      def create_migration_file
-        migration_template(
-          "add_period_totals_to_llm_cost_tracker.rb.erb",
-          "db/migrate/add_period_totals_to_llm_cost_tracker.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/add_provider_response_id_generator.rb

@@ -1,29 +0,0 @@
-# frozen_string_literal: true
-
-require "rails/generators"
-require "rails/generators/active_record"
-
-module LlmCostTracker
-  module Generators
-    class AddProviderResponseIdGenerator < Rails::Generators::Base
-      include ActiveRecord::Generators::Migration
-
-      source_root File.expand_path("templates", __dir__)
-
-      desc "Creates a migration to add llm_api_calls.provider_response_id"
-
-      def create_migration_file
-        migration_template(
-          "add_provider_response_id_to_llm_api_calls.rb.erb",
-          "db/migrate/add_provider_response_id_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/add_streaming_generator.rb

@@ -1,29 +0,0 @@
-# frozen_string_literal: true
-
-require "rails/generators"
-require "rails/generators/active_record"
-
-module LlmCostTracker
-  module Generators
-    class AddStreamingGenerator < Rails::Generators::Base
-      include ActiveRecord::Generators::Migration
-
-      source_root File.expand_path("templates", __dir__)
-
-      desc "Creates a migration to add llm_api_calls.stream and llm_api_calls.usage_source"
-
-      def create_migration_file
-        migration_template(
-          "add_streaming_to_llm_api_calls.rb.erb",
-          "db/migrate/add_streaming_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/add_token_usage_generator.rb

@@ -1,42 +0,0 @@
-# frozen_string_literal: true
-
-require "rails/generators"
-require "rails/generators/active_record"
-
-module LlmCostTracker
-  module Generators
-    class AddTokenUsageGenerator < Rails::Generators::Base
-      include ActiveRecord::Generators::Migration
-
-      TOKEN_COLUMNS = %w[
-        cache_read_input_tokens
-        cache_write_input_tokens
-        cache_write_1h_input_tokens
-        hidden_output_tokens
-      ].freeze
-      COST_COLUMNS = %w[
-        cache_read_input_cost
-        cache_write_input_cost
-        cache_write_1h_input_cost
-      ].freeze
-      COLUMN_NAMES = (TOKEN_COLUMNS + COST_COLUMNS + %w[pricing_mode]).freeze
-
-      source_root File.expand_path("templates", __dir__)
-
-      desc "Creates a migration to add token usage and token cost columns to llm_api_calls"
-
-      def create_migration_file
-        migration_template(
-          "add_token_usage_to_llm_api_calls.rb.erb",
-          "db/migrate/add_token_usage_to_llm_api_calls.rb"
-        )
-      end
-
-      private
-
-      def migration_version
-        "[#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}]"
-      end
-    end
-  end
-end