active_harness 0.2.12 → 0.2.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f4f069d1d894324475b0a5395d5728fa77626698255bc3400428ef66a2228ac4
-  data.tar.gz: 1cbfee779b26e122d0e01e3fa1e4a80907398eec5f0895b28bc1dcb19484d8a4
+  metadata.gz: 0c5339e68dc9f7de87fc67a4e986cd1858e86f52e95ee277ee0f12563eedf428
+  data.tar.gz: d563d7088e6b29c2119160a9eb0e0a39eb550d12e1bf828937a7d20fddcf7ad4
 SHA512:
-  metadata.gz: c2293baad25b739468f7fb3306583f2e475409edc7b1533a03e83da9e28776ba17a27ae6a9aebf47fb18d4a1953957240d2615a277443e39781876a685d7da1e
-  data.tar.gz: 6e78901ee75204d853b53df08baef9215f7f225f15065da771262ee084af17243a796ab132526701b016e4d0066762338e01fd01fd36f5e316e1045599cc8c3b
+  metadata.gz: 9d710fedd64ae1bd1ee40616aacb884e698491ec223895651c1ad28a307121575f5078df54074e3218ade227cf376f20a881af15ecef640e63f74887cd5a9115
+  data.tar.gz: '075778aea6ae0b53624b18fd64835cdaaec6e0381ff7616faab86c2b9813d5ae80830bdcf59ce71b255a13930ad6d35f99b1953bde11e58c7a8bda03c36732df'

data/lib/active_harness/agent/cost.rb

@@ -0,0 +1,28 @@
+module ActiveHarness
+  class Agent
+    private
+
+    # Calculates the monetary cost of a single request based on token usage
+    # and pricing data from ActiveHarness::Costs.
+    #
+    # Returns a hash { input_cost:, output_cost:, total_cost: } in USD,
+    # or nil if usage is absent or the model is not found in the pricing registry.
+    def calculate_cost(model_id, usage)
+      return nil if model_id.nil? || usage.nil?
+
+      pricing = ActiveHarness::Costs.find(model_id.to_s)
+      return nil unless pricing&.input_per_million && pricing&.output_per_million
+
+      input_cost  = (usage[:input_tokens].to_f  / 1_000_000) * pricing.input_per_million
+      output_cost = (usage[:output_tokens].to_f / 1_000_000) * pricing.output_per_million
+
+      {
+        input_cost:  input_cost.round(8),
+        output_cost: output_cost.round(8),
+        total_cost:  (input_cost + output_cost).round(8)
+      }
+    rescue StandardError
+      nil
+    end
+  end
+end

data/lib/active_harness/agent.rb

@@ -113,6 +113,7 @@ module ActiveHarness
     def build_result(response, entry, attempts, elapsed)
       raw    = response[:content]
       parsed = parse_output(raw)
+      usage  = response[:usage]
 
       Result.new(
         input:          @input,
@@ -125,7 +126,8 @@ module ActiveHarness
         model_list:     model_list,
         attempts:       attempts,
         execution_time: elapsed,
-        usage:          response[:usage]
+        usage:          usage,
+        cost:           calculate_cost(entry[:model], usage)
       )
     end
 
@@ -152,4 +154,5 @@ require_relative "agent/models"
 require_relative "agent/providers"
 require_relative "agent/output_parser"
 require_relative "agent/ruby_llm_backend"
+require_relative "agent/cost"
 

data/lib/active_harness/costs.rb

@@ -183,10 +183,21 @@ module ActiveHarness
       end
 
       def registry
-        @registry ||= begin
-          source = File.exist?(cache_file) ? cache_file : BUNDLED_DATA_FILE
-          JSON.parse(File.read(source), symbolize_names: true)
+        @registry ||= load_registry
+      end
+
+      def load_registry
+        if File.exist?(cache_file)
+          begin
+            data = JSON.parse(File.read(cache_file), symbolize_names: true)
+            return data if data.is_a?(Array)
+          rescue JSON::ParserError
+            # Cache file corrupted — fall through to bundled data
+          end
         end
+        JSON.parse(File.read(BUNDLED_DATA_FILE), symbolize_names: true)
+      rescue JSON::ParserError, Errno::ENOENT
+        []
       end
 
       def fetch_models_dev

data/lib/active_harness/result.rb

@@ -6,5 +6,6 @@ module ActiveHarness
   # output — raw string from the provider
   # parsed — for format :json: a Ruby Hash/Array; for format :text: same as output
   # usage  — token counts: { input_tokens:, output_tokens:, total_tokens: } or nil for streaming
-  Result = Struct.new(:input, :output, :parsed, :system_prompt, :provider, :model, :temperature, :model_list, :attempts, :execution_time, :usage, keyword_init: true)
+  # cost   — { input_cost:, output_cost:, total_cost: } in USD, or nil if pricing unavailable
+  Result = Struct.new(:input, :output, :parsed, :system_prompt, :provider, :model, :temperature, :model_list, :attempts, :execution_time, :usage, :cost, keyword_init: true)
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: active_harness
 version: !ruby/object:Gem::Version
-  version: 0.2.12
+  version: 0.2.13
 platform: ruby
 authors:
 - the-teacher
@@ -33,6 +33,7 @@ extra_rdoc_files: []
 files:
 - lib/active_harness.rb
 - lib/active_harness/agent.rb
+- lib/active_harness/agent/cost.rb
 - lib/active_harness/agent/hooks.rb
 - lib/active_harness/agent/models.rb
 - lib/active_harness/agent/output_parser.rb