llm_cost_tracker 0.1.2 → 0.1.4

data/lib/llm_cost_tracker.rb

@@ -6,9 +6,15 @@ require "active_support/notifications"
 require_relative "llm_cost_tracker/version"
 require_relative "llm_cost_tracker/configuration"
 require_relative "llm_cost_tracker/errors"
+require_relative "llm_cost_tracker/logging"
+require_relative "llm_cost_tracker/value_object"
+require_relative "llm_cost_tracker/cost"
+require_relative "llm_cost_tracker/event"
+require_relative "llm_cost_tracker/parsed_usage"
 require_relative "llm_cost_tracker/price_registry"
 require_relative "llm_cost_tracker/pricing"
 require_relative "llm_cost_tracker/parsers/base"
+require_relative "llm_cost_tracker/parsers/openai_usage"
 require_relative "llm_cost_tracker/parsers/openai"
 require_relative "llm_cost_tracker/parsers/openai_compatible"
 require_relative "llm_cost_tracker/parsers/anthropic"
@@ -18,7 +24,14 @@ require_relative "llm_cost_tracker/middleware/faraday"
 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/tags_column"
+require_relative "llm_cost_tracker/tag_query"
+require_relative "llm_cost_tracker/tag_accessors"
+require_relative "llm_cost_tracker/storage/backends"
 require_relative "llm_cost_tracker/tracker"
+require_relative "llm_cost_tracker/report_data"
+require_relative "llm_cost_tracker/report_formatter"
+require_relative "llm_cost_tracker/report"
 
 module LlmCostTracker
   class << self
@@ -30,6 +43,10 @@ module LlmCostTracker
       @configuration || CONFIGURATION_MUTEX.synchronize { @configuration ||= Configuration.new }
     end
 
+    # Configure the gem once during application boot.
+    #
+    # @yieldparam configuration [LlmCostTracker::Configuration]
+    # @return [void]
     def configure
       yield(configuration)
       configuration.normalize_openai_compatible_providers!
@@ -40,7 +57,7 @@ module LlmCostTracker
       CONFIGURATION_MUTEX.synchronize { @configuration = Configuration.new }
     end
 
-    # Manual tracking for non-Faraday clients
+    # Track an LLM request manually for non-Faraday clients.
     #
     #   LlmCostTracker.track(
     #     provider: :openai,
@@ -50,6 +67,14 @@ module LlmCostTracker
     #     feature: "chat",
     #     user_id: current_user.id
     #   )
+    #
+    # @param provider [String, Symbol] Provider name, such as :openai or :anthropic.
+    # @param model [String] Provider model identifier.
+    # @param input_tokens [Integer] Billed input token count.
+    # @param output_tokens [Integer] Billed output token count.
+    # @param latency_ms [Integer, nil] Optional request latency in milliseconds.
+    # @param metadata [Hash] Attribution tags and provider-specific usage metadata.
+    # @return [LlmCostTracker::Event] The tracked event.
     def track(provider:, model:, input_tokens:, output_tokens:, latency_ms: nil, **metadata)
       Tracker.record(
         provider: provider.to_s,
@@ -64,20 +89,10 @@ module LlmCostTracker
     private
 
     def warn_for_configuration!
-      return unless (configuration.budget_exceeded_behavior || :notify).to_sym == :block_requests
+      return unless configuration.budget_exceeded_behavior == :block_requests
       return if configuration.active_record?
 
-      log_warning(":block_requests requires storage_backend = :active_record; preflight blocking will be skipped.")
-    end
-
-    def log_warning(message)
-      message = "[LlmCostTracker] #{message}"
-
-      if defined?(Rails) && Rails.respond_to?(:logger) && Rails.logger
-        Rails.logger.warn(message)
-      else
-        warn message
-      end
+      Logging.warn(":block_requests requires storage_backend = :active_record; preflight blocking will be skipped.")
     end
   end
 end

data/lib/tasks/llm_cost_tracker.rake

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+namespace :llm_cost_tracker do
+  desc "Print an LLM cost report from ActiveRecord storage"
+  task report: :environment do
+    days = (ENV["DAYS"] || LlmCostTracker::Report::DEFAULT_DAYS).to_i
+    puts LlmCostTracker::Report.generate(days: days)
+  end
+end

data/llm_cost_tracker.gemspec

@@ -12,7 +12,7 @@ Gem::Specification.new do |spec|
   spec.description   = "Tracks token usage and estimated costs for OpenAI, Anthropic, Google Gemini, " \
                        "OpenRouter, DeepSeek, and OpenAI-compatible calls. " \
                        "Works as Faraday middleware for Ruby clients, with ActiveRecord storage, " \
-                       "per-user/per-feature attribution, budget alerts, and budget enforcement."
+                       "per-user/per-feature attribution, and budget guardrails."
   spec.homepage      = "https://github.com/sergey-homenko/llm_cost_tracker"
   spec.license       = "MIT"
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm_cost_tracker
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Sergii Khomenko
@@ -142,8 +142,8 @@ dependencies:
         version: '3.0'
 description: Tracks token usage and estimated costs for OpenAI, Anthropic, Google
   Gemini, OpenRouter, DeepSeek, and OpenAI-compatible calls. Works as Faraday middleware
-  for Ruby clients, with ActiveRecord storage, per-user/per-feature attribution, budget
-  alerts, and budget enforcement.
+  for Ruby clients, with ActiveRecord storage, per-user/per-feature attribution, and
+  budget guardrails.
 email:
 - sergey@mm.st
 executables: []
@@ -159,33 +159,52 @@ files:
 - lib/llm_cost_tracker.rb
 - lib/llm_cost_tracker/budget.rb
 - lib/llm_cost_tracker/configuration.rb
+- lib/llm_cost_tracker/cost.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_latency_ms_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/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/llm_api_call.rb
+- lib/llm_cost_tracker/logging.rb
 - lib/llm_cost_tracker/middleware/faraday.rb
+- lib/llm_cost_tracker/parsed_usage.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/price_registry.rb
 - lib/llm_cost_tracker/prices.json
 - lib/llm_cost_tracker/pricing.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/storage/active_record_backend.rb
 - lib/llm_cost_tracker/storage/active_record_store.rb
+- lib/llm_cost_tracker/storage/backends.rb
+- lib/llm_cost_tracker/storage/custom_backend.rb
+- lib/llm_cost_tracker/storage/log_backend.rb
+- lib/llm_cost_tracker/tag_accessors.rb
+- lib/llm_cost_tracker/tag_query.rb
+- lib/llm_cost_tracker/tags_column.rb
 - lib/llm_cost_tracker/tracker.rb
 - lib/llm_cost_tracker/unknown_pricing.rb
+- lib/llm_cost_tracker/value_object.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: