llm_cost_tracker 0.1.0 → 0.1.2

data/lib/llm_cost_tracker.rb

@@ -5,31 +5,39 @@ 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/price_registry"
 require_relative "llm_cost_tracker/pricing"
 require_relative "llm_cost_tracker/parsers/base"
 require_relative "llm_cost_tracker/parsers/openai"
+require_relative "llm_cost_tracker/parsers/openai_compatible"
 require_relative "llm_cost_tracker/parsers/anthropic"
 require_relative "llm_cost_tracker/parsers/gemini"
 require_relative "llm_cost_tracker/parsers/registry"
 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/tracker"
 
 module LlmCostTracker
-  class Error < StandardError; end
-
   class << self
+    CONFIGURATION_MUTEX = Mutex.new
+
     attr_writer :configuration
 
     def configuration
-      @configuration ||= Configuration.new
+      @configuration || CONFIGURATION_MUTEX.synchronize { @configuration ||= Configuration.new }
     end
 
     def configure
       yield(configuration)
+      configuration.normalize_openai_compatible_providers!
+      warn_for_configuration!
     end
 
     def reset_configuration!
-      @configuration = Configuration.new
+      CONFIGURATION_MUTEX.synchronize { @configuration = Configuration.new }
     end
 
     # Manual tracking for non-Faraday clients
@@ -42,15 +50,35 @@ module LlmCostTracker
     #     feature: "chat",
     #     user_id: current_user.id
     #   )
-    def track(provider:, model:, input_tokens:, output_tokens:, **metadata)
+    def track(provider:, model:, input_tokens:, output_tokens:, latency_ms: nil, **metadata)
       Tracker.record(
         provider: provider.to_s,
         model: model,
         input_tokens: input_tokens,
         output_tokens: output_tokens,
+        latency_ms: latency_ms,
         metadata: metadata
       )
     end
+
+    private
+
+    def warn_for_configuration!
+      return unless (configuration.budget_exceeded_behavior || :notify).to_sym == :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
+    end
   end
 end
 

data/llm_cost_tracker.gemspec

@@ -8,10 +8,11 @@ Gem::Specification.new do |spec|
   spec.authors       = ["Sergii Khomenko"]
   spec.email         = ["sergey@mm.st"]
 
-  spec.summary       = "Provider-agnostic LLM API cost tracking for Ruby"
-  spec.description   = "Automatically tracks token usage and costs for LLM API calls (OpenAI, Anthropic, Google Gemini, and more). " \
-                        "Works as Faraday middleware — plugs into any Ruby HTTP client. " \
-                        "Provides ActiveRecord storage, per-user/per-feature attribution, and budget alerts."
+  spec.summary       = "Self-hosted LLM API cost guardrails for Ruby and Rails"
+  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."
   spec.homepage      = "https://github.com/sergey-homenko/llm_cost_tracker"
   spec.license       = "MIT"
 
@@ -19,6 +20,7 @@ Gem::Specification.new do |spec|
 
   spec.metadata["homepage_uri"]  = spec.homepage
   spec.metadata["changelog_uri"] = "#{spec.homepage}/blob/main/CHANGELOG.md"
+  spec.metadata["rubygems_mfa_required"] = "true"
 
   spec.files = Dir.chdir(__dir__) do
     `git ls-files -z`.split("\x0").reject do |f|
@@ -29,13 +31,13 @@ Gem::Specification.new do |spec|
 
   spec.require_paths = ["lib"]
 
-  spec.add_dependency "faraday", ">= 1.0", "< 3.0"
   spec.add_dependency "activesupport", ">= 7.0", "< 9.0"
+  spec.add_dependency "faraday", ">= 1.0", "< 3.0"
 
   spec.add_development_dependency "activerecord", ">= 7.0", "< 9.0"
   spec.add_development_dependency "rake", "~> 13.0"
   spec.add_development_dependency "rspec", "~> 3.0"
-  spec.add_development_dependency "webmock", "~> 3.0"
-  spec.add_development_dependency "sqlite3", "~> 2.0"
   spec.add_development_dependency "rubocop", "~> 1.0"
+  spec.add_development_dependency "sqlite3", "~> 2.0"
+  spec.add_development_dependency "webmock", "~> 3.0"
 end

metadata

@@ -1,55 +1,55 @@
 --- !ruby/object:Gem::Specification
 name: llm_cost_tracker
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Sergii Khomenko
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-04-16 00:00:00.000000000 Z
+date: 2026-04-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: faraday
+  name: activesupport
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: '7.0'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '9.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: '7.0'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '9.0'
 - !ruby/object:Gem::Dependency
-  name: activesupport
+  name: faraday
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.0'
+        version: '1.0'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '9.0'
+        version: '3.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.0'
+        version: '1.0'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '9.0'
+        version: '3.0'
 - !ruby/object:Gem::Dependency
   name: activerecord
   requirement: !ruby/object:Gem::Requirement
@@ -99,19 +99,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '3.0'
 - !ruby/object:Gem::Dependency
-  name: webmock
+  name: rubocop
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '1.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '1.0'
 - !ruby/object:Gem::Dependency
   name: sqlite3
   requirement: !ruby/object:Gem::Requirement
@@ -127,23 +127,23 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '2.0'
 - !ruby/object:Gem::Dependency
-  name: rubocop
+  name: webmock
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: '3.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.0'
-description: Automatically tracks token usage and costs for LLM API calls (OpenAI,
-  Anthropic, Google Gemini, and more). Works as Faraday middleware — plugs into any
-  Ruby HTTP client. Provides ActiveRecord storage, per-user/per-feature attribution,
-  and budget alerts.
+        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.
 email:
 - sergey@mm.st
 executables: []
@@ -151,26 +151,40 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".rspec"
+- ".rubocop.yml"
 - CHANGELOG.md
 - LICENSE.txt
 - README.md
 - Rakefile
 - lib/llm_cost_tracker.rb
+- lib/llm_cost_tracker/budget.rb
 - lib/llm_cost_tracker/configuration.rb
+- lib/llm_cost_tracker/errors.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/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/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/middleware/faraday.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/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/storage/active_record_store.rb
 - lib/llm_cost_tracker/tracker.rb
+- lib/llm_cost_tracker/unknown_pricing.rb
 - lib/llm_cost_tracker/version.rb
 - llm_cost_tracker.gemspec
 homepage: https://github.com/sergey-homenko/llm_cost_tracker
@@ -179,6 +193,7 @@ licenses:
 metadata:
   homepage_uri: https://github.com/sergey-homenko/llm_cost_tracker
   changelog_uri: https://github.com/sergey-homenko/llm_cost_tracker/blob/main/CHANGELOG.md
+  rubygems_mfa_required: 'true'
 post_install_message:
 rdoc_options: []
 require_paths:
@@ -197,5 +212,5 @@ requirements: []
 rubygems_version: 3.5.9
 signing_key:
 specification_version: 4
-summary: Provider-agnostic LLM API cost tracking for Ruby
+summary: Self-hosted LLM API cost guardrails for Ruby and Rails
 test_files: []