active_harness 0.2.30 → 0.2.32

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f5f759449bd21052f986bc092a621f2ce39cb4faa630ba1bc759419bc27aac37
-  data.tar.gz: 94e8abc9d0a0ca7ef877ea0f1e66ac68899236d7d33f10ac754f3541b3a29d1b
+  metadata.gz: 231f48053aa92c1172f16cf7b6d03b222eccc9a967092d78488718dafa006b2b
+  data.tar.gz: 1f4f83487faff2a5b6b5dad31e8c902718c6179a0fec1162425b62640454b2da
 SHA512:
-  metadata.gz: 8a9529f9391337660a325993adf7db17bd2f470f6b2e69cab256a1ac907616d0c76a5c5b4c56f9850399fbfb8d886b3f2d2c2241c6ceb66ea0d83cb0e2d4c724
-  data.tar.gz: e1ae97b30a763b761ae28d20b957670d5b8a774d249456cfd84da09830ce6ebc8661c2d19790a69ff2889ff585ce7af307beb13b5d46886708db041fd1c33d73
+  metadata.gz: 23b68426a66c24f9f1c5759b64a01cefd4293a5f94f11f1a30806447d55db1d85cc54d4de4a3a18b9254db32b5aeb63c502938adbe023cb4d03962dcc4d8247b
+  data.tar.gz: 93ca0e61a02ce50bb5beb10fbb4c98630ef0c90efc9642b3d1c8d74f666559cdebcc965b4f29bcda32c9b7ac5609c0a6ea49790d5f62384a2ecacbab941fd0b2

data/lib/active_harness/agent/cost.rb

@@ -2,25 +2,23 @@ module ActiveHarness
   class Agent
     private
 
-    # Calculates the monetary cost of a single request based on token usage
-    # and pricing data from ActiveHarness::Costs.
+    # Builds a CostBreakdown for a single request from token usage and
+    # pricing data from ActiveHarness::Pricing.
     #
-    # Returns a hash { input_cost:, output_cost:, total_cost: } in USD,
+    # Returns CostBreakdown (input, output, total 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?
+    def calculate_cost(pricing, tokens)
+      return nil unless pricing && tokens
+      return nil unless pricing.input_per_million && pricing.output_per_million
 
-      pricing = ActiveHarness::Costs.find(model_id.to_s)
-      return nil unless pricing&.input_per_million && pricing&.output_per_million
+      input_cost  = (tokens.input.to_f  / 1_000_000) * pricing.input_per_million
+      output_cost = (tokens.output.to_f / 1_000_000) * 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)
-      }
+      CostBreakdown.new(
+        input:  input_cost.round(8),
+        output: output_cost.round(8),
+        total:  (input_cost + output_cost).round(8)
+      )
     rescue StandardError
       nil
     end

data/lib/active_harness/agent/prompt.rb

@@ -57,7 +57,13 @@ module ActiveHarness
       obj.instance_variable_set(:@params,         @params)
       obj.instance_variable_set(:@config,         @config)
       obj.instance_variable_set(:@memory,         @memory)
-      obj.instance_variable_set(:@context_window, @context_window)
+      obj.instance_variable_set(:@context_window, context_window_for_prompt)
+    end
+
+    def context_window_for_prompt
+      Pricing.find(model_list.to_a.first&.dig(:model).to_s)&.context_window
+    rescue StandardError
+      nil
     end
   end
 end

data/lib/active_harness/agent.rb

@@ -54,7 +54,6 @@ module ActiveHarness
                   :params,
                   :memory
     attr_reader   :result,
-                  :context_window,
                   :token_stream,
                   :event_stream
 
@@ -77,9 +76,8 @@ module ActiveHarness
       @context         = context
       @params          = params
       @memory          = memory
-      @models_override  = Array(models) if models
-      @context_window   = lookup_context_window(self.models.to_a.first)
-      @token_stream     = streams[:token]
+      @models_override = Array(models) if models
+      @token_stream    = streams[:token]
       @event_stream     = streams[:agent]
       fire(:setup)
     end
@@ -146,31 +144,59 @@ module ActiveHarness
     end
 
     def build_result(response, entry, attempts, elapsed)
-      raw       = response[:content]
-      processed = parse_output(raw)
-      usage     = response[:usage]
-      cw        = lookup_context_window(entry)
+      raw        = response[:content]
+      processed  = parse_output(raw)
+      raw_usage  = response[:usage]
+      model_cost = lookup_model_cost(entry)
 
       Result.new(
         input:          @input,
         output:         raw,
         processed:      processed,
         system_prompt:  @system_prompt,
-        provider:       entry[:provider],
-        model:          entry[:model],
-        temperature:    entry[:temperature],
+        model:          build_model_info(entry, model_cost),
         model_list:     model_list,
         attempts:       attempts,
         execution_time: elapsed,
-        usage:          usage,
-        cost:           calculate_cost(entry[:model], usage),
-        context_window: cw
+        usage:          build_usage(raw_usage, model_cost)
+      )
+    end
+
+    def build_model_info(entry, model_cost)
+      pricing = if model_cost&.input_per_million && model_cost&.output_per_million
+        ModelPricing.new(
+          input:  (model_cost.input_per_million  / 1_000_000.0).round(10),
+          output: (model_cost.output_per_million / 1_000_000.0).round(10)
+        )
+      end
+
+      ModelInfo.new(
+        name:           entry[:model],
+        provider:       entry[:provider],
+        temperature:    entry[:temperature],
+        context_window: model_cost&.context_window,
+        pricing:        pricing
+      )
+    end
+
+    def build_usage(raw_usage, model_cost)
+      return nil if raw_usage.nil?
+
+      tokens = TokenCounts.new(
+        input:  raw_usage[:input_tokens],
+        output: raw_usage[:output_tokens],
+        total:  raw_usage[:total_tokens]
+      )
+
+      UsageInfo.new(
+        tokens: tokens,
+        cost:   calculate_cost(model_cost, tokens)
       )
     end
 
-    def lookup_context_window(entry)
+    def lookup_model_cost(entry)
       return nil unless entry
-      Costs.find(entry[:model])&.context_window
+      Pricing.find(entry[:model].to_s)
     rescue StandardError
       nil
     end

data/lib/active_harness/{costs.rb → pricing.rb}

@@ -10,28 +10,28 @@ module ActiveHarness
   # by ActiveHarness (files present in lib/active_harness/providers/).
   #
   # Data source priority:
-  #   1. {project_root}/tmp/active_harness/costs.json — fetched cache (refreshed once per day)
+  #   1. {project_root}/tmp/active_harness/pricing.json — fetched cache (refreshed once per day)
   #   2. lib/active_harness/data/models.json           — bundled fallback (ships with gem)
   #
   # Usage:
   #
   #   # Fetch fresh data and save to tmp cache (also called automatically when stale)
-  #   ActiveHarness::Costs.update
+  #   ActiveHarness::Pricing.update
   #
   #   # All models (auto-updates cache if missing or older than 24h)
-  #   ActiveHarness::Costs.all
+  #   ActiveHarness::Pricing.all
   #
   #   # Single model by ID
-  #   ActiveHarness::Costs.find("gpt-4o")
+  #   ActiveHarness::Pricing.find("gpt-4o")
   #
   #   # By provider — method or bracket syntax
-  #   ActiveHarness::Costs.providers.openai
-  #   ActiveHarness::Costs.providers[:anthropic]
+  #   ActiveHarness::Pricing.providers.openai
+  #   ActiveHarness::Pricing.providers[:anthropic]
   #
   #   # List providers that have data
-  #   ActiveHarness::Costs.providers.list
+  #   ActiveHarness::Pricing.providers.list
   #
-  module Costs
+  module Pricing
     BUNDLED_DATA_FILE = File.expand_path("data/models.json", __dir__).freeze
     MODELS_DEV_URL    = "https://models.dev/api.json"
     CACHE_TTL         = 86_400 # 24 hours in seconds
@@ -54,7 +54,7 @@ module ActiveHarness
     }.freeze
 
     # Value object representing the pricing for a single model.
-    ModelCost = Struct.new(
+    ModelPrice = Struct.new(
       :id,
       :name,
       :provider,
@@ -89,31 +89,31 @@ module ActiveHarness
         parts << "output=$#{output_per_million}/M" if output_per_million
         parts << "ctx=#{context_window}"           if context_window
         parts << "cats=#{categories.join(',')}"    if categories.any?
-        "#<ModelCost #{parts.join(' ')}>"
+        "#<ModelPrice #{parts.join(' ')}>"
       end
     end
 
     # Proxy object that exposes providers as methods and via [].
     class ProvidersProxy
       def [](name)
-        ActiveHarness::Costs.for_provider(name.to_s)
+        ActiveHarness::Pricing.for_provider(name.to_s)
       end
 
       def list
-        ActiveHarness::Costs.provider_names
+        ActiveHarness::Pricing.provider_names
       end
 
       def method_missing(name, *args, &block)
         provider = name.to_s
-        if ActiveHarness::Costs.provider_names.include?(provider)
-          ActiveHarness::Costs.for_provider(provider)
+        if ActiveHarness::Pricing.provider_names.include?(provider)
+          ActiveHarness::Pricing.for_provider(provider)
         else
           super
         end
       end
 
       def respond_to_missing?(name, include_private = false)
-        ActiveHarness::Costs.provider_names.include?(name.to_s) || super
+        ActiveHarness::Pricing.provider_names.include?(name.to_s) || super
       end
     end
 
@@ -154,7 +154,7 @@ module ActiveHarness
       end
 
       # Fetches fresh pricing data from models.dev, filters to supported providers,
-      # and writes the result to {project_root}/tmp/active_harness/costs.json.
+      # and writes the result to {project_root}/tmp/active_harness/pricing.json.
       # Returns the number of models saved, or raises on HTTP failure.
       def update
         raw_api = fetch_models_dev
@@ -176,7 +176,7 @@ module ActiveHarness
 
       # Path to the per-project cache file.
       def cache_file
-        File.join(project_root, "tmp", "active_harness", "costs.json")
+        File.join(project_root, "tmp", "active_harness", "pricing.json")
       end
 
       # Names of providers supported by ActiveHarness (derived from providers/ directory).
@@ -267,7 +267,7 @@ module ActiveHarness
 
       def build_cost(raw)
         standard = raw.dig(:pricing, :text_tokens, :standard) || {}
-        ModelCost.new(
+        ModelPrice.new(
           id:                            raw[:id],
           name:                          raw[:name],
           provider:                      raw[:provider],

data/lib/active_harness/result.rb

@@ -1,24 +1,64 @@
 require "json"
 
 module ActiveHarness
-  # Minimal result wrapper returned by Agent#call.
+  # Value objects for structured result data.
+
+  # Pricing rates for a model (per-token, from Pricing registry).
+  # nil when the model is not found in the pricing registry.
+  ModelPricing = Struct.new(:input, :output, keyword_init: true)
+
+  # Static model metadata resolved at call time.
+  ModelInfo = Struct.new(
+    :name,
+    :provider,
+    :temperature,
+    :context_window,
+    :pricing,         # ModelPricing or nil
+    keyword_init: true
+  ) do
+    def to_s; name.to_s; end
+    def inspect
+      parts = ["name=#{name.inspect}", "provider=#{provider.inspect}"]
+      parts << "temperature=#{temperature}" if temperature
+      parts << "context_window=#{context_window}" if context_window
+      parts << "pricing=#{pricing.inspect}" if pricing
+      "#<ModelInfo #{parts.join(' ')}>"
+    end
+  end
+
+  # Token counts for a single call.
+  TokenCounts = Struct.new(:input, :output, :total, keyword_init: true)
+
+  # Monetary cost of a single call in USD.
+  # nil when pricing data is unavailable.
+  CostBreakdown = Struct.new(:input, :output, :total, keyword_init: true)
+
+  # Combined token + cost stats for a single agent call.
+  # tokens is always present (raw provider data).
+  # cost is nil when pricing is unavailable.
+  UsageInfo = Struct.new(:tokens, :cost, keyword_init: true)
+
+  # Result returned by Agent#call (accessible via agent.result).
   #
-  # output — raw string from the provider
-  # processed — 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
-  # cost   — { input_cost:, output_cost:, total_cost: } in USD, or nil if pricing unavailable
+  #   result.input            — original input string
+  #   result.output           — raw string from the provider
+  #   result.processed        — parsed Hash/Array for :json agents, raw string for :text
+  #   result.system_prompt    — resolved system prompt string
+  #   result.model            — ModelInfo (name, provider, temperature, context_window, pricing)
+  #   result.model_list       — full model chain proxy
+  #   result.attempts         — Array of failed attempt entries before success
+  #   result.execution_time   — wall-clock seconds for the successful call
+  #   result.usage            — UsageInfo (tokens + cost), nil for streaming without usage
   Result = Struct.new(
     :input,
     :output,
     :processed,
     :system_prompt,
-    :provider, :model,
-    :temperature,
+    :model,           # ModelInfo
     :model_list,
     :attempts,
     :execution_time,
-    :usage,
-    :cost,
-    :context_window,
-    keyword_init: true)
+    :usage,           # UsageInfo or nil
+    keyword_init: true
+  )
 end

data/lib/active_harness.rb

@@ -21,7 +21,7 @@ require_relative "active_harness/providers/azure"
 require_relative "active_harness/providers/bedrock"
 require_relative "active_harness/providers/vertexai"
 require_relative "active_harness/providers/custom"
-require_relative "active_harness/costs"
+require_relative "active_harness/pricing"
 require_relative "active_harness/memory"
 require_relative "active_harness/agent"
 require_relative "active_harness/tribunal"
@@ -30,7 +30,7 @@ require_relative "active_harness/pipeline"
 require_relative "active_harness/railtie" if defined?(Rails::Railtie)
 
 module ActiveHarness
-  VERSION = "0.2.30"
+  VERSION = "0.2.32"
 
   class << self
     # Configure ActiveHarness.

data/lib/generators/active_harness/install/templates/controllers/ai_controller.rb

@@ -12,7 +12,7 @@ class AiSupportController < ApplicationController
 
     render json: {
       output: result.output,
-      model:  result.model,
+      model:  result.model.name,
       time:   result.execution_time
     }
   end
@@ -30,7 +30,7 @@ class AiSupportController < ApplicationController
 
     render json: {
       output: result.output,
-      model:  result.model,
+      model:  result.model.name,
       time:   result.execution_time,
       turns:  memory.size
     }

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: active_harness
 version: !ruby/object:Gem::Version
-  version: 0.2.30
+  version: 0.2.32
 platform: ruby
 authors:
 - the-teacher
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2026-06-11 00:00:00.000000000 Z
+date: 2026-06-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby
@@ -43,7 +43,6 @@ files:
 - lib/active_harness/configuration.rb
 - lib/active_harness/core/errors.rb
 - lib/active_harness/core/hooks.rb
-- lib/active_harness/costs.rb
 - lib/active_harness/data/models.json
 - lib/active_harness/http/client.rb
 - lib/active_harness/http/retry_policy.rb
@@ -57,6 +56,7 @@ files:
 - lib/active_harness/pipeline/README.md
 - lib/active_harness/pipeline/hooks.rb
 - lib/active_harness/pipeline/step.rb
+- lib/active_harness/pricing.rb
 - lib/active_harness/providers/PROVIDER_CONTRACT.md
 - lib/active_harness/providers/anthropic.rb
 - lib/active_harness/providers/azure.rb