llm_cost_tracker 0.3.2 → 0.4.0

data/lib/llm_cost_tracker/unknown_pricing.rb

@@ -1,9 +1,13 @@
 # frozen_string_literal: true
 
+require "monitor"
+
 require_relative "logging"
 
 module LlmCostTracker
   class UnknownPricing
+    MUTEX = Monitor.new
+
     class << self
       def handle!(model)
         model = normalized_model_name(model)
@@ -18,6 +22,10 @@ module LlmCostTracker
         end
       end
 
+      def reset!
+        MUTEX.synchronize { @warned_models = Set.new }
+      end
+
       private
 
       def normalized_model_name(model)
@@ -25,6 +33,12 @@ module LlmCostTracker
       end
 
       def warn_missing(model)
+        should_warn = MUTEX.synchronize do
+          @warned_models ||= Set.new
+          @warned_models.add?(model)
+        end
+        return unless should_warn
+
         Logging.warn(
           "No pricing configured for model #{model.inspect}. " \
           "Cost and budget guardrails will be skipped for this event. " \

data/lib/llm_cost_tracker/usage_breakdown.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module LlmCostTracker
+  UsageBreakdown = Data.define(
+    :input_tokens,
+    :cache_read_input_tokens,
+    :cache_write_input_tokens,
+    :output_tokens,
+    :hidden_output_tokens
+  ) do
+    def self.build(input_tokens:, output_tokens:, cache_read_input_tokens: 0,
+                   cache_write_input_tokens: 0, hidden_output_tokens: 0)
+      new(
+        input_tokens: input_tokens.to_i,
+        cache_read_input_tokens: cache_read_input_tokens.to_i,
+        cache_write_input_tokens: cache_write_input_tokens.to_i,
+        output_tokens: output_tokens.to_i,
+        hidden_output_tokens: hidden_output_tokens.to_i
+      )
+    end
+
+    def total_tokens
+      input_tokens + cache_read_input_tokens + cache_write_input_tokens + output_tokens
+    end
+
+    def to_h
+      super.merge(total_tokens: total_tokens).compact
+    end
+  end
+end

data/lib/llm_cost_tracker/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LlmCostTracker
-  VERSION = "0.3.2"
+  VERSION = "0.4.0"
 end

data/lib/llm_cost_tracker.rb

@@ -10,6 +10,7 @@ require_relative "llm_cost_tracker/errors"
 require_relative "llm_cost_tracker/logging"
 require_relative "llm_cost_tracker/parameter_hash"
 require_relative "llm_cost_tracker/cost"
+require_relative "llm_cost_tracker/usage_breakdown"
 require_relative "llm_cost_tracker/event"
 require_relative "llm_cost_tracker/parsed_usage"
 require_relative "llm_cost_tracker/price_registry"
@@ -60,6 +61,8 @@ module LlmCostTracker
 
     def reset_configuration!
       CONFIGURATION_MUTEX.synchronize { @configuration = Configuration.new }
+      UnknownPricing.reset! if defined?(UnknownPricing)
+      Storage::ActiveRecordStore.reset! if defined?(Storage::ActiveRecordStore)
     end
 
     def enforce_budget!
@@ -67,7 +70,7 @@ module LlmCostTracker
     end
 
     def track(provider:, model:, input_tokens:, output_tokens:, latency_ms: nil, stream: false, usage_source: :manual,
-              enforce_budget: false, provider_response_id: nil, **metadata)
+              enforce_budget: false, provider_response_id: nil, pricing_mode: nil, **metadata)
       enforce_budget! if enforce_budget
       Tracker.record(
         provider: provider.to_s,
@@ -78,11 +81,13 @@ module LlmCostTracker
         stream: stream,
         usage_source: usage_source,
         provider_response_id: provider_response_id,
+        pricing_mode: pricing_mode,
         metadata: metadata
       )
     end
 
-    def track_stream(provider:, model:, latency_ms: nil, enforce_budget: false, provider_response_id: nil, **metadata)
+    def track_stream(provider:, model:, latency_ms: nil, enforce_budget: false, provider_response_id: nil,
+                     pricing_mode: nil, **metadata)
       require_relative "llm_cost_tracker/stream_collector"
       enforce_budget! if enforce_budget
       collector = StreamCollector.new(
@@ -90,6 +95,7 @@ module LlmCostTracker
         model: model,
         latency_ms: latency_ms,
         provider_response_id: provider_response_id,
+        pricing_mode: pricing_mode,
         metadata: metadata
       )
       yield collector
@@ -105,7 +111,10 @@ module LlmCostTracker
       return unless config.budget_exceeded_behavior == :block_requests
       return if config.active_record?
 
-      Logging.warn(":block_requests requires storage_backend = :active_record; preflight blocking will be skipped.")
+      Logging.warn(
+        ":block_requests requires storage_backend = :active_record for monthly and daily preflight; " \
+        "preflight blocking will be skipped."
+      )
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: llm_cost_tracker
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.4.0
 platform: ruby
 authors:
 - Sergii Khomenko
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-04-22 00:00:00.000000000 Z
+date: 2026-04-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -278,13 +278,17 @@ files:
 - lib/llm_cost_tracker/event.rb
 - lib/llm_cost_tracker/event_metadata.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/install_generator.rb
 - lib/llm_cost_tracker/generators/llm_cost_tracker/prices_generator.rb
 - 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_usage_breakdown_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/llm_cost_tracker_prices.yml.erb
@@ -306,6 +310,7 @@ files:
 - 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_registry.rb
 - lib/llm_cost_tracker/price_sync.rb
 - lib/llm_cost_tracker/price_sync/fetcher.rb
@@ -327,6 +332,7 @@ files:
 - 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_rollups.rb
 - lib/llm_cost_tracker/storage/active_record_store.rb
 - lib/llm_cost_tracker/stream_collector.rb
 - lib/llm_cost_tracker/tag_accessors.rb
@@ -335,10 +341,10 @@ files:
 - lib/llm_cost_tracker/tags_column.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/version.rb
 - lib/tasks/llm_cost_tracker.rake
-- llm_cost_tracker.gemspec
 homepage: https://github.com/sergey-homenko/llm_cost_tracker
 licenses:
 - MIT
@@ -363,7 +369,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.9
+rubygems_version: 3.5.22
 signing_key:
 specification_version: 4
 summary: Self-hosted LLM usage and cost tracking for Ruby and Rails

data/llm_cost_tracker.gemspec

@@ -1,50 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "lib/llm_cost_tracker/version"
-
-Gem::Specification.new do |spec|
-  spec.name          = "llm_cost_tracker"
-  spec.version       = LlmCostTracker::VERSION
-  spec.authors       = ["Sergii Khomenko"]
-  spec.email         = ["sergey@mm.st"]
-
-  spec.summary       = "Self-hosted LLM usage and cost tracking for Ruby and Rails"
-  spec.description   = "Tracks token usage, latency, and estimated costs for 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."
-  spec.homepage      = "https://github.com/sergey-homenko/llm_cost_tracker"
-  spec.license       = "MIT"
-
-  spec.required_ruby_version = ">= 3.3.0"
-
-  spec.metadata["bug_tracker_uri"] = "#{spec.homepage}/issues"
-  spec.metadata["changelog_uri"]   = "#{spec.homepage}/blob/main/CHANGELOG.md"
-  spec.metadata["source_code_uri"] = spec.homepage
-  spec.metadata["documentation_uri"] = "#{spec.homepage}#readme"
-  spec.metadata["rubygems_mfa_required"] = "true"
-
-  spec.files = Dir.chdir(__dir__) do
-    `git ls-files -z`.split("\x0").reject do |f|
-      (File.expand_path(f) == __FILE__) ||
-        f.start_with?("bin/", "docs/", "test/", "spec/", ".git", ".github", "gemfiles/", ".rubocop", "Gemfile")
-    end
-  end
-
-  spec.require_paths = ["lib"]
-
-  spec.add_dependency "activesupport", ">= 7.1", "< 9.0"
-  spec.add_dependency "csv", "~> 3.0"
-  spec.add_dependency "faraday", ">= 2.0", "< 3.0"
-
-  spec.add_development_dependency "activerecord", ">= 7.1", "< 9.0"
-  spec.add_development_dependency "railties", ">= 7.1", "< 9.0"
-  spec.add_development_dependency "rake", "~> 13.0"
-  spec.add_development_dependency "rspec", "~> 3.0"
-  spec.add_development_dependency "rubocop", "~> 1.0"
-  spec.add_development_dependency "simplecov", "~> 0.22"
-  spec.add_development_dependency "simplecov-lcov", "~> 0.8"
-  spec.add_development_dependency "sqlite3", ">= 1.4", "< 3.0"
-  spec.add_development_dependency "webmock", "~> 3.0"
-end