llm_cost_tracker 0.4.1 → 0.5.0

data/lib/llm_cost_tracker.rb

@@ -25,9 +25,11 @@ require_relative "llm_cost_tracker/parsers/gemini"
 require_relative "llm_cost_tracker/parsers/sse"
 require_relative "llm_cost_tracker/parsers/registry"
 require_relative "llm_cost_tracker/middleware/faraday"
+require_relative "llm_cost_tracker/integrations/registry"
 require_relative "llm_cost_tracker/budget"
 require_relative "llm_cost_tracker/unknown_pricing"
 require_relative "llm_cost_tracker/event_metadata"
+require_relative "llm_cost_tracker/tag_context"
 require_relative "llm_cost_tracker/tags_column"
 require_relative "llm_cost_tracker/tag_key"
 require_relative "llm_cost_tracker/tag_query"
@@ -37,6 +39,7 @@ require_relative "llm_cost_tracker/retention"
 require_relative "llm_cost_tracker/report_data"
 require_relative "llm_cost_tracker/report_formatter"
 require_relative "llm_cost_tracker/report"
+require_relative "llm_cost_tracker/doctor"
 
 module LlmCostTracker
   CONFIGURATION_MUTEX = Monitor.new
@@ -52,10 +55,11 @@ module LlmCostTracker
         current = current.dup_for_configuration if current.finalized?
         @configuration = current
         yield(current)
-        current.normalize_openai_compatible_providers!
+        current.openai_compatible_providers = current.openai_compatible_providers.dup
         current.finalize!
         current
       end
+      Integrations.install!
       warn_for_configuration!(config)
     end
 
@@ -63,14 +67,20 @@ module LlmCostTracker
       CONFIGURATION_MUTEX.synchronize { @configuration = Configuration.new }
       UnknownPricing.reset! if defined?(UnknownPricing)
       Storage::ActiveRecordStore.reset! if defined?(Storage::ActiveRecordStore)
+      TagContext.clear! if defined?(TagContext)
     end
 
     def enforce_budget!
       Tracker.enforce_budget!
     end
 
-    def track(provider:, model:, input_tokens:, output_tokens:, latency_ms: nil, stream: false, usage_source: :manual,
-              enforce_budget: false, provider_response_id: nil, pricing_mode: nil, **metadata)
+    def with_tags(tags = nil, **kwargs, &)
+      merged = (tags || {}).to_h.merge(kwargs)
+      TagContext.with(merged, &)
+    end
+
+    def track(provider:, input_tokens:, output_tokens:, model: nil, latency_ms: nil, stream: false,
+              usage_source: :manual, enforce_budget: false, provider_response_id: nil, pricing_mode: nil, **metadata)
       enforce_budget! if enforce_budget
       Tracker.record(
         provider: provider.to_s,
@@ -86,7 +96,7 @@ module LlmCostTracker
       )
     end
 
-    def track_stream(provider:, model:, latency_ms: nil, enforce_budget: false, provider_response_id: nil,
+    def track_stream(provider:, model: nil, 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

data/lib/tasks/llm_cost_tracker.rake

@@ -2,6 +2,14 @@
 
 # rubocop:disable Metrics/BlockLength
 namespace :llm_cost_tracker do
+  desc "Check LLM Cost Tracker setup"
+  task :doctor do
+    Rake::Task["environment"].invoke if Rake::Task.task_defined?("environment")
+    checks = LlmCostTracker::Doctor.call
+    puts LlmCostTracker::Doctor.report(checks)
+    abort("llm_cost_tracker: doctor found setup errors") unless LlmCostTracker::Doctor.healthy?(checks)
+  end
+
   desc "Print an LLM cost report from ActiveRecord storage"
   task report: :environment do
     days = (ENV["DAYS"] || LlmCostTracker::Report::DEFAULT_DAYS).to_i
@@ -18,7 +26,7 @@ namespace :llm_cost_tracker do
 
   namespace :prices do
     desc(
-      "Sync built-in pricing data from LiteLLM/OpenRouter JSON sources. " \
+      "Sync the configured pricing file from LiteLLM/OpenRouter JSON sources. " \
       "Use PREVIEW=1 to preview, STRICT=1 to fail on provider errors, " \
       "or OUTPUT=path/to/file.json."
     )
@@ -26,7 +34,7 @@ namespace :llm_cost_tracker do
       Rake::Task["environment"].invoke if Rake::Task.task_defined?("environment")
       require_relative "../llm_cost_tracker"
 
-      output_path = ENV["OUTPUT"] || LlmCostTracker.configuration.prices_file || LlmCostTracker::PriceSync::DEFAULT_OUTPUT_PATH
+      output_path = price_sync_output_path
       strict = ENV["STRICT"] == "1" || ARGV.include?("--strict")
       result = LlmCostTracker::PriceSync.sync(
         path: output_path,
@@ -57,7 +65,7 @@ namespace :llm_cost_tracker do
       Rake::Task["environment"].invoke if Rake::Task.task_defined?("environment")
       require_relative "../llm_cost_tracker"
 
-      output_path = ENV["OUTPUT"] || LlmCostTracker.configuration.prices_file || LlmCostTracker::PriceSync::DEFAULT_OUTPUT_PATH
+      output_path = price_sync_output_path
       result = LlmCostTracker::PriceSync.check(path: output_path)
 
       puts "llm_cost_tracker: checked pricing file #{result.path}"
@@ -131,3 +139,13 @@ def print_failures(failed_sources, heading:)
     puts "    - #{source}: #{message}"
   end
 end
+
+def price_sync_output_path
+  path = LlmCostTracker::PriceSync.configured_output_path
+  return path if path
+
+  abort(
+    "llm_cost_tracker: configure prices_file, run bin/rails generate llm_cost_tracker:prices, " \
+    "or set OUTPUT=config/llm_cost_tracker_prices.yml"
+  )
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: llm_cost_tracker
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.5.0
 platform: ruby
 authors:
 - Sergii Khomenko
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-04-24 00:00:00.000000000 Z
+date: 2026-04-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -272,7 +272,9 @@ files:
 - lib/llm_cost_tracker/assets.rb
 - lib/llm_cost_tracker/budget.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/engine.rb
 - lib/llm_cost_tracker/engine_compatibility.rb
 - lib/llm_cost_tracker/errors.rb
@@ -292,11 +294,15 @@ files:
 - 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
 - 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/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/llm_api_call.rb
 - lib/llm_cost_tracker/logging.rb
 - lib/llm_cost_tracker/middleware/faraday.rb
@@ -312,6 +318,7 @@ files:
 - 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
@@ -332,11 +339,13 @@ files:
 - 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/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
+- 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/tags_column.rb

data/lib/llm_cost_tracker/generators/llm_cost_tracker/templates/llm_cost_tracker_prices.yml.erb

@@ -1,51 +0,0 @@
-# Local LlmCostTracker price overrides.
-#
-# Add only the models you want to override or add. Built-in prices still come
-# from the gem's prices.json, and Ruby pricing_overrides still take precedence.
-#
-# Units: USD per 1M tokens.
-#
-# Supported price keys:
-# - input
-# - output
-# - cache_read_input
-# - cache_write_input
-# - mode_input / mode_output / mode_cache_read_input / mode_cache_write_input
-#
-# Optional metadata keys, ignored by cost calculation:
-# - _source
-# - _source_version
-# - _fetched_at
-# - _updated
-# - _notes
-# - _validator_override
-#
-# Example: custom fine-tune
-# models:
-#   "ft:gpt-4o-mini:my-org":
-#     input: 0.30
-#     cache_read_input: 0.15
-#     output: 1.20
-#     _notes: "Internal fine-tune rate"
-#
-# Example: alternate pricing mode
-# models:
-#   "batchable-model":
-#     input: 1.00
-#     output: 2.00
-#     batch_input: 0.50
-#     batch_output: 1.00
-#
-# Example: negotiated provider discount
-# models:
-#   "gpt-4o":
-#     input: 2.00
-#     output: 8.00
-#     _source: "manual"
-#     _updated: "2026-04-18"
-#
-# Use _source: "manual" for custom or orphaned entries you never want sync to touch.
-# Use _validator_override: ["skip_relative_change"] if a negotiated price would
-# otherwise trip the >3x sync warning.
-
-models: