active_harness 0.2.27 → 0.2.29

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5b406bc5feeebdb4cc26e215a65d25dd931605684c6f05ee7cdd71de5910a1d6
-  data.tar.gz: a34d2e22840ae906d47b0e27fe271d97b42d97a0fce64a19a58e15c296f00bf7
+  metadata.gz: 7146a492cd9454703ab47eb1c6222962587b4fe5585aaef2a5390349da394a1a
+  data.tar.gz: 476ee4d22259b10fa7d58cd25208ea5c900e6523991b1f17e288f7c3104cfd92
 SHA512:
-  metadata.gz: f53ee25213ac7404840349cee732aa59b1dc25168b4e362028e7542fd51a0c51e9f2a2af4b8cbad9a552e74a9c4769d147b7d642e3b051df9c293af58bc87f34
-  data.tar.gz: d56efdd52b38afd9153903611fd860e316a59eb8f5d5b5c3ed0510b22647b8008a3b22f4e0c8430322548bb82a50c69cc4d6c086e8d965653c22d795fd43cae7
+  metadata.gz: d211c987a80244262c35fbc83202491b61807ee98f092808ca801cfcc57eb1ad58e4f48ccf8e364078b99a267e14816e8551b55c5b2a3de2fe8e6828321bc2b6
+  data.tar.gz: 71244082f0e0c81d9fe7d0271e05dd53ebf0e5a727ced5554d637e84aed8a58b6a014e8b0537ff2dcc10cc501385905b78c5aae8f328c1d774a874fc30d3c7b3

data/lib/active_harness/agent/prompt.rb

@@ -49,11 +49,15 @@ module ActiveHarness
       prompt
     end
 
+    # Injects agent state into a prompt class instance before #call.
+    # Available in prompt classes: @input, @context, @params, @memory, @context_window, @config
     def inject_agent_state(obj)
-      obj.instance_variable_set(:@input,   @input)
-      obj.instance_variable_set(:@context, @context)
-      obj.instance_variable_set(:@config,  @config)
-      obj.instance_variable_set(:@memory,  @memory)
+      obj.instance_variable_set(:@input,          @input)
+      obj.instance_variable_set(:@context,        @context)
+      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)
     end
   end
 end

data/lib/active_harness/agent.rb

@@ -54,6 +54,7 @@ module ActiveHarness
                   :params,
                   :memory
     attr_reader   :result,
+                  :context_window,
                   :token_stream,
                   :event_stream
 
@@ -76,9 +77,10 @@ module ActiveHarness
       @context         = context
       @params          = params
       @memory          = memory
-      @models_override = Array(models) if models
-      @token_stream    = streams[:token]
-      @event_stream    = streams[:agent]
+      @models_override  = Array(models) if models
+      @context_window   = lookup_context_window(self.models.to_a.first)
+      @token_stream     = streams[:token]
+      @event_stream     = streams[:agent]
       fire(:setup)
     end
 
@@ -144,9 +146,10 @@ module ActiveHarness
     end
 
     def build_result(response, entry, attempts, elapsed)
-      raw    = response[:content]
+      raw       = response[:content]
       processed = parse_output(raw)
-      usage  = response[:usage]
+      usage     = response[:usage]
+      cw        = lookup_context_window(entry)
 
       Result.new(
         input:          @input,
@@ -160,10 +163,18 @@ module ActiveHarness
         attempts:       attempts,
         execution_time: elapsed,
         usage:          usage,
-        cost:           calculate_cost(entry[:model], usage)
+        cost:           calculate_cost(entry[:model], usage),
+        context_window: cw
       )
     end
 
+    def lookup_context_window(entry)
+      return nil unless entry
+      Costs.find(entry[:model])&.context_window
+    rescue StandardError
+      nil
+    end
+
     def normalize_input!
       return if @config.fetch(:normalize_input, true) == false
       @input = @input&.strip&.gsub(/\s+/, " ")

data/lib/active_harness/costs.rb

@@ -62,12 +62,33 @@ module ActiveHarness
       :output_per_million,
       :cache_read_input_per_million,
       :cache_write_input_per_million,
+      :context_window,
+      :max_output_tokens,
+      :input_modalities,
+      :output_modalities,
       keyword_init: true
     ) do
+      # Returns capability tags derived from modality data and model id/name.
+      # Possible values: "vision", "pdf", "audio", "video", "imggen", "embed"
+      def categories
+        inp = input_modalities  || []
+        out = output_modalities || []
+        cats = []
+        cats << "vision" if inp.include?("image")
+        cats << "pdf"    if inp.include?("pdf")
+        cats << "audio"  if inp.include?("audio") || out.include?("audio")
+        cats << "video"  if inp.include?("video")  || out.include?("video")
+        cats << "imggen" if out.include?("image")
+        cats << "embed"  if id.to_s.match?(/embed/i) || name.to_s.match?(/embed/i)
+        cats
+      end
+
       def inspect
         parts = ["id=#{id.inspect}", "provider=#{provider.inspect}"]
         parts << "input=$#{input_per_million}/M"  if input_per_million
         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(' ')}>"
       end
     end
@@ -229,11 +250,16 @@ module ActiveHarness
               cache_write_input_per_million: cost[:cache_write]
             }.compact
 
+            mods = m[:modalities] || {}
             {
-              id:       m[:id],
-              name:     m[:name] || m[:id],
-              provider: ah_provider,
-              pricing:  standard.any? ? { text_tokens: { standard: standard } } : {}
+              id:                m[:id],
+              name:              m[:name] || m[:id],
+              provider:          ah_provider,
+              context_window:    m[:context_window] || m.dig(:limit, :context),
+              max_output_tokens: m[:max_output_tokens] || m.dig(:limit, :output),
+              input_modalities:  Array(mods[:input]),
+              output_modalities: Array(mods[:output]),
+              pricing:           standard.any? ? { text_tokens: { standard: standard } } : {}
             }
           end
         end
@@ -248,7 +274,11 @@ module ActiveHarness
           input_per_million:             standard[:input_per_million],
           output_per_million:            standard[:output_per_million],
           cache_read_input_per_million:  standard[:cache_read_input_per_million],
-          cache_write_input_per_million: standard[:cache_write_input_per_million]
+          cache_write_input_per_million: standard[:cache_write_input_per_million],
+          context_window:                raw[:context_window],
+          max_output_tokens:             raw[:max_output_tokens],
+          input_modalities:              Array(raw[:input_modalities]),
+          output_modalities:             Array(raw[:output_modalities])
         )
       end
 

data/lib/active_harness/memory.rb

@@ -84,6 +84,12 @@ module ActiveHarness
       async:       false,
       **adapter_opts
     )
+      if self.class == Memory
+        raise NotImplementedError,
+          "Memory cannot be instantiated directly. " \
+          "Use Memory::JsonFile, Memory::Postgresql, or Memory::Sqlite."
+      end
+
       @session_id  = session_id
       @depth       = depth
       @enabled     = enabled
@@ -138,13 +144,15 @@ module ActiveHarness
 
     # Returns messages array for LLM consumption, respecting depth.
     # Optional filters:
-    #   filter: ->(turn) { turn[:agent] == "SupportAgent" }
-    #   since:  Time.now - 3600
-    def to_messages(filter: nil, since: nil)
+    #   filter:       ->(turn) { turn[:agent] == "SupportAgent" }
+    #   since:        Time.now - 3600
+    #   token_budget: 4000  # rough limit (chars / 4 estimate); trims oldest turns first
+    def to_messages(filter: nil, since: nil, token_budget: nil)
       turns = @turns.dup
       turns.select! { |t| filter.call(t) }   if filter
       turns.select! { |t| after?(t, since) } if since
       turns = turns.last(@depth)              if @depth
+      turns = trim_to_token_budget(turns, token_budget) if token_budget
 
       turns.flat_map do |t|
         [
@@ -210,5 +218,16 @@ module ActiveHarness
       turn_time = Time.parse(turn[:at].to_s) rescue nil
       turn_time ? turn_time >= time : true
     end
+
+    def trim_to_token_budget(turns, budget)
+      return turns if turns.empty?
+
+      total = 0
+      turns.reverse.take_while do |t|
+        tokens = ((t[:request].to_s.length + t[:response].to_s.length) / 4.0).ceil
+        total += tokens
+        total <= budget
+      end.reverse
+    end
   end
 end

data/lib/active_harness/pipeline/README.md

@@ -16,7 +16,7 @@ class SupportPipeline < ActiveHarness::Pipeline
 
   step :safety_tribunal do
     use SafetyTribunal
-    stop_if ->(result) { result.verdict == false }
+    stop_if ->(result) { result.processed["verdict"] == false }
   end
 end
 

data/lib/active_harness/pipeline.rb

@@ -14,7 +14,7 @@ module ActiveHarness
   #
   #     step :safety_tribunal do
   #       use SafetyTribunal
-  #       stop_if ->(result) { result.verdict == false }
+  #       stop_if ->(result) { result.processed["verdict"] == false }
   #     end
   #
   #     on :before_step do |step_name, payload| ... end

data/lib/active_harness/result.rb

@@ -11,7 +11,7 @@ module ActiveHarness
     :input,
     :output,
     :processed,
-    :system_prompt, 
+    :system_prompt,
     :provider, :model,
     :temperature,
     :model_list,
@@ -19,5 +19,6 @@ module ActiveHarness
     :execution_time,
     :usage,
     :cost,
+    :context_window,
     keyword_init: true)
 end

data/lib/active_harness.rb

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

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

@@ -43,11 +43,12 @@ class AiSupportController < ApplicationController
   # Returns verdict: true (safe) or false (rejected).
   # ---------------------------------------------------------------------------
   def tribunal
-    result = SupportGuardTribunal.call(input: params.require(:input))
+    tribunal = SupportGuardTribunal.new(input: params.require(:input))
+    tribunal.call
 
     render json: {
-      verdict: result.verdict,
-      time:    result.execution_time
+      verdict: tribunal.verdict,
+      time:    tribunal.execution_time
     }
   end
 

data/lib/generators/active_harness/install/templates/pipelines/support_pipeline.rb

@@ -7,7 +7,7 @@ class SupportPipeline < ActiveHarness::Pipeline
   # Step 1 — GUARD: reject spam before spending tokens on an answer
   step :spam_guard do
     use SupportGuardTribunal
-    stop_if ->(result) { result.verdict == false }
+    stop_if ->(result) { result.processed["verdict"] == false }
   end
 
   # Step 2 — RESPOND: generate the actual answer

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: active_harness
 version: !ruby/object:Gem::Version
-  version: 0.2.27
+  version: 0.2.29
 platform: ruby
 authors:
 - the-teacher
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2026-06-09 00:00:00.000000000 Z
+date: 2026-06-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby