llm_cost_tracker 0.5.3 → 0.6.0

data/lib/llm_cost_tracker.rb

@@ -31,6 +31,7 @@ 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/tag_sanitizer"
+require_relative "llm_cost_tracker/active_record_adapter"
 require_relative "llm_cost_tracker/tags_column"
 require_relative "llm_cost_tracker/tag_key"
 require_relative "llm_cost_tracker/tag_sql"
@@ -53,6 +54,10 @@ module LlmCostTracker
       CONFIGURATION_MUTEX.synchronize { @configuration ||= Configuration.new }
     end
 
+    def configuration_generation
+      CONFIGURATION_MUTEX.synchronize { @configuration_generation ||= 0 }
+    end
+
     def configure
       config = CONFIGURATION_MUTEX.synchronize do
         current = @configuration || Configuration.new
@@ -61,6 +66,7 @@ module LlmCostTracker
         yield(current)
         current.openai_compatible_providers = current.openai_compatible_providers.dup
         current.finalize!
+        @configuration_generation = @configuration_generation.to_i + 1
         current
       end
       Integrations.install!
@@ -68,12 +74,39 @@ module LlmCostTracker
     end
 
     def reset_configuration!
-      CONFIGURATION_MUTEX.synchronize { @configuration = Configuration.new }
+      Storage::ActiveRecordInbox.reset! if defined?(Storage::ActiveRecordInbox)
+      Storage::ActiveRecordIngestor.shutdown!(drain: false) if defined?(Storage::ActiveRecordIngestor)
+      CONFIGURATION_MUTEX.synchronize do
+        @configuration = Configuration.new
+        @configuration_generation = @configuration_generation.to_i + 1
+      end
       UnknownPricing.reset! if defined?(UnknownPricing)
       Storage::ActiveRecordStore.reset! if defined?(Storage::ActiveRecordStore)
+      Storage::ActiveRecordInbox.reset! if defined?(Storage::ActiveRecordInbox)
+      Storage::ActiveRecordIngestor.reset! if defined?(Storage::ActiveRecordIngestor)
       TagContext.clear! if defined?(TagContext)
     end
 
+    def flush!(timeout: nil)
+      return true unless defined?(Storage::ActiveRecordIngestor)
+
+      if timeout
+        Storage::ActiveRecordIngestor.flush!(timeout: timeout)
+      else
+        Storage::ActiveRecordIngestor.flush!
+      end
+    end
+
+    def shutdown!(timeout: nil, drain: true)
+      return true unless defined?(Storage::ActiveRecordIngestor)
+
+      if timeout
+        Storage::ActiveRecordIngestor.shutdown!(timeout: timeout, drain: drain)
+      else
+        Storage::ActiveRecordIngestor.shutdown!(drain: drain)
+      end
+    end
+
     def enforce_budget!
       Tracker.enforce_budget!
     end
@@ -140,3 +173,5 @@ if defined?(Faraday)
     llm_cost_tracker: LlmCostTracker::Middleware::Faraday
   )
 end
+
+at_exit { LlmCostTracker.shutdown!(drain: false) if defined?(LlmCostTracker) }

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: llm_cost_tracker
 version: !ruby/object:Gem::Version
-  version: 0.5.3
+  version: 0.6.0
 platform: ruby
 authors:
 - Sergii Khomenko
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-04-28 00:00:00.000000000 Z
+date: 2026-04-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -301,6 +301,7 @@ files:
 - docs/technical/operational-notes.md
 - docs/upgrading.md
 - lib/llm_cost_tracker.rb
+- lib/llm_cost_tracker/active_record_adapter.rb
 - lib/llm_cost_tracker/assets.rb
 - lib/llm_cost_tracker/budget.rb
 - lib/llm_cost_tracker/capture_verifier.rb
@@ -310,11 +311,13 @@ files:
 - lib/llm_cost_tracker/cost.rb
 - lib/llm_cost_tracker/doctor.rb
 - lib/llm_cost_tracker/doctor/capture_check.rb
+- lib/llm_cost_tracker/doctor/ingestion_check.rb
 - lib/llm_cost_tracker/engine.rb
 - lib/llm_cost_tracker/engine_compatibility.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_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
@@ -322,6 +325,7 @@ files:
 - 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_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
@@ -333,6 +337,8 @@ files:
 - 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/inbox_event.rb
+- lib/llm_cost_tracker/ingestor_lease.rb
 - lib/llm_cost_tracker/integrations/anthropic.rb
 - lib/llm_cost_tracker/integrations/base.rb
 - lib/llm_cost_tracker/integrations/object_reader.rb
@@ -375,6 +381,15 @@ files:
 - lib/llm_cost_tracker/request_url.rb
 - lib/llm_cost_tracker/retention.rb
 - lib/llm_cost_tracker/storage/active_record_backend.rb
+- lib/llm_cost_tracker/storage/active_record_connection_cleanup.rb
+- lib/llm_cost_tracker/storage/active_record_inbox.rb
+- lib/llm_cost_tracker/storage/active_record_inbox_batch.rb
+- lib/llm_cost_tracker/storage/active_record_ingestor.rb
+- lib/llm_cost_tracker/storage/active_record_ingestor_lease.rb
+- lib/llm_cost_tracker/storage/active_record_period_totals.rb
+- lib/llm_cost_tracker/storage/active_record_periods.rb
+- lib/llm_cost_tracker/storage/active_record_rollup_batch.rb
+- lib/llm_cost_tracker/storage/active_record_rollup_upsert_sql.rb
 - lib/llm_cost_tracker/storage/active_record_rollups.rb
 - lib/llm_cost_tracker/storage/active_record_store.rb
 - lib/llm_cost_tracker/storage/custom_backend.rb