active_harness 0.2.37 → 0.2.38

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5fb1f238ea442c93d556e18ef1b899ad8d8e8345c6ed18c8abd0d45370c847e0
-  data.tar.gz: 8c8b0dceb9c779f6432dc86f141fdda65144b365511ee3096957a96954447b4e
+  metadata.gz: '064709a78ef949aadd741237f4cdd5a2f36056d211fa2038048a2911fc45a5be'
+  data.tar.gz: 298318704ce083cf46b87dc5dbaebbae15df127a313685ea32d5c82b03191072
 SHA512:
-  metadata.gz: fd1da6cb093ea99b414576fe9e1fc2ec0fee4a894eca5bd5291582ea5daabdb1ee621f22a5b7e4460455c03fd47df5b152b5d9598c64df9a408cbaec93ca3371
-  data.tar.gz: 939c5d39b1f172ae0225087f095e8665c84b34f6cc384bf2e8f949e871a0be63c3ecf59253e253a1066c40c6c41528f57c8b09790720dca765c8760175ce36ad
+  metadata.gz: b7c4e5a06e8f8a66e89a0f0b2289c7f461e76cd942a374abb5e6a3845f35e348b8b5fc63182e384d825863ef4c8698c59b64d606d11b5eb27bcada47b1d57642
+  data.tar.gz: 3528828f40f1a11a0b62b45ff193b49e05cc202e09e754a578e94c372279cdb760622d5e33de3f61aeef62d7cf15166197a4e0d2a5637d0c4cb8e5d5cf332112

data/lib/active_harness/agent/cost.rb

@@ -1,12 +1,36 @@
 module ActiveHarness
   class Agent
+    # Providers that have a dedicated pricing source beyond ModelsDev.
+    # Consulted in tier-2 before the general ModelsDev fallback.
+    # Add entries here when a new provider-specific source is available.
+    PROVIDER_PRICING_SOURCES = {
+      openrouter: Pricing::OpenRouter
+    }.freeze
+
     private
 
-    # Builds a CostBreakdown for a single request from token usage and
-    # pricing data from ActiveHarness::Pricing.
-    #
-    # Returns CostBreakdown (input, output, total in USD),
-    # or nil if usage is absent or the model is not found in the pricing registry.
+    # Cost lookup — three-tier fallback:
+    #   1. provider_cost in the API response  (handled in build_usage)
+    #   2. provider-specific source           (e.g. Pricing::OpenRouter for :openrouter)
+    #   3. Pricing::ModelsDev general fallback
+    #   → nil when no data found at any tier
+    def lookup_model_cost(entry)
+      return nil unless entry
+
+      model    = entry[:model].to_s
+      provider = entry[:provider].to_sym
+
+      source = PROVIDER_PRICING_SOURCES[provider]
+      cost   = source&.find(model)
+      return cost if cost
+
+      Pricing::ModelsDev.find(model)
+    rescue StandardError
+      nil
+    end
+
+    # Builds a CostBreakdown from token counts and per-million rates.
+    # Returns nil when pricing or token data is absent.
     def calculate_cost(pricing, tokens)
       return nil unless pricing && tokens
       return nil unless pricing.input_per_million && pricing.output_per_million

data/lib/active_harness/agent.rb

@@ -198,18 +198,6 @@ module ActiveHarness
       UsageInfo.new(tokens: tokens, cost: cost)
     end
 
-    def lookup_model_cost(entry)
-      return nil unless entry
-
-      if entry[:provider].to_sym == :openrouter
-        Pricing::OpenRouter.find(entry[:model].to_s) || Pricing.find(entry[:model].to_s)
-      else
-        Pricing.find(entry[:model].to_s)
-      end
-    rescue StandardError
-      nil
-    end
-
     def normalize_input!
       return if @config.fetch(:normalize_input, true) == false
       @input = @input&.strip&.gsub(/\s+/, " ")

data/lib/active_harness.rb

@@ -32,7 +32,7 @@ require_relative "active_harness/pipeline"
 require_relative "active_harness/railtie" if defined?(Rails::Railtie)
 
 module ActiveHarness
-  VERSION = "0.2.37"
+  VERSION = "0.2.38"
 
   class << self
     # Configure ActiveHarness.

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: active_harness
 version: !ruby/object:Gem::Version
-  version: 0.2.37
+  version: 0.2.38
 platform: ruby
 authors:
 - the-teacher