active_harness 0.2.20 → 0.2.22

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d1188ca468f31f46c40d118b7e719a849925aa11ddab3a4a87ea4df0fd9bdb84
-  data.tar.gz: 1ed36314efd943e6f4e62bbc43b2ec00d0b0486bdbf015e6b590341d240ffcdc
+  metadata.gz: 33e2cd8c861873028568322fd36104414d698fb4888810388d0ac9b1c1b5b9d4
+  data.tar.gz: cd792deb836b9cbb5237f9c8c02a6f436c3c48162b21edb412f9797a39a94e16
 SHA512:
-  metadata.gz: 37e82033bddd4069f26619d2bfc5faec831c32618216bedb6cb9ce19ca7fbbaac1057e829809ee15f1d88d1a6fc3b3832cd2c02d1f112c850a974a26205d3727
-  data.tar.gz: 3a4fafd8184d55f116689e28c4f411d102f7279d290093c9cc32dafabfc9eafe9aa55abde59a04cec37985e74da94a8ddb14153febce35aa41c2eb0404eed2c6
+  metadata.gz: 93365a99d1965c177e379a48027f5fc2dee3fa06dba5061d41be5b7948ba134ed8307cf65c4e57951f21798150c05745311ee793c49fb08be11669ebb13f8e8d
+  data.tar.gz: 3fb4814a98bde51eba1982f55cd22457bd96613be7f1b44b08434fe0c3c3dc467d7d6ccb0fe36cbbb590885139c1b047c9dc79b5293e7d5fd4f5be32df4e5ca7

data/lib/active_harness/agent/hooks.rb

@@ -32,7 +32,7 @@ module ActiveHarness
         end
 
         agent_config[:hooks] ||= {}
-        agent_config[:hooks][event] = block
+        (agent_config[:hooks][event] ||= []) << block
       end
 
       # Rails-style aliases for +on+:
@@ -59,27 +59,20 @@ module ActiveHarness
       end
     end
 
+    include Core::HookRunner
+
     private
 
     def run_hook(event, *args)
-      hooks = @config[:hooks] || {}
-      return unless hooks[event]
-
-      if args.any?
-        instance_exec(*args, &hooks[event])
-      else
-        instance_eval(&hooks[event])
-      end
+      run_hooks(@config[:hooks] || {}, event, *args)
     end
 
     # Unified internal method: fires the DSL hook AND the external event_stream lambda.
     # Consistent with Tribunal#fire and Pipeline#fire.
     def fire(event, *args)
-      result = run_hook(event, *args)
+      run_hook(event, *args)
       @event_stream&.call(event, *args)
-      result
     rescue IOError, ActionController::Live::ClientDisconnected
-      result
     end
   end
 end

data/lib/active_harness/agent.rb

@@ -11,8 +11,22 @@ module ActiveHarness
       #   SupportAgent.call(input: "Hi")
       #   SupportAgent.call(input: "Hi", context: { user_id: 42 })
       #   SupportAgent.call(input: "Hi", memory: memory)
-      def call(input: nil, context: {}, models: nil, memory: nil, streams: {})
-        new(input: input, context: context, models: models, memory: memory, streams: streams).call
+      def call(
+        input:   nil,
+        context: {},
+        params:  {},
+        models:  nil,
+        memory:  nil,
+        streams: {}
+      )
+        new(
+          input:   input,
+          context: context,
+          params:  params,
+          models:  models,
+          memory:  memory,
+          streams: streams
+        ).call
       end
 
       # Each subclass gets its own isolated config hash.
@@ -36,8 +50,12 @@ module ActiveHarness
     # -------------------------------------------------------------------------
     # Instance API
     # -------------------------------------------------------------------------
-    attr_accessor :input, :context
-    attr_reader   :result, :token_stream, :event_stream
+    attr_accessor :input,
+                  :context,
+                  :params
+    attr_reader   :result,
+                  :token_stream,
+                  :event_stream
 
     def models=(list)
       @models_override = Array(list)
@@ -48,11 +66,19 @@ module ActiveHarness
       @memory = obj
     end
 
-    def initialize(input: nil, context: {}, models: nil, memory: nil, streams: {})
+    def initialize(
+      input:   nil,
+      context: {},
+      params:  {},
+      models:  nil,
+      memory:  nil,
+      streams: {}
+    )
       @input           = input
       @config          = self.class.agent_config
       normalize_input!
       @context         = context
+      @params          = params
       @models_override = Array(models) if models
       @token_stream    = streams[:token]
       @event_stream    = streams[:agent]

data/lib/active_harness/core/hooks.rb

@@ -0,0 +1,26 @@
+module ActiveHarness
+  module Core
+    # Shared hook execution logic included by Agent, Tribunal, and Pipeline.
+    #
+    # Hooks are stored in arrays so multiple +on+/+before+/+after+/+callback+
+    # calls with the same event name accumulate — later registrations append
+    # rather than overwrite. This lets modules register default hooks without
+    # blocking user-defined hooks on the same event.
+    #
+    #   class MyAgent < ActiveHarness::Agent
+    #     include SomeTracingConcern   # registers before(:call) internally
+    #     before(:call) { ... }        # appends — both hooks run in order
+    #   end
+    module HookRunner
+      private
+
+      # Execute every block registered for +event+, passing +args+ to each.
+      # Blocks run in the receiver's instance context (instance_exec / instance_eval).
+      def run_hooks(hooks_hash, event, *args)
+        Array(hooks_hash[event]).each do |blk|
+          args.any? ? instance_exec(*args, &blk) : instance_eval(&blk)
+        end
+      end
+    end
+  end
+end

data/lib/active_harness/pipeline/hooks.rb

@@ -0,0 +1,77 @@
+module ActiveHarness
+  class Pipeline
+    VALID_HOOKS      = %i[before_step after_step stopped complete].freeze
+    VALID_STEP_HOOKS = %i[before_step after_step].freeze
+
+    class << self
+      # Register a global or per-step hook.
+      #
+      # Global hooks fire on every step:
+      #   on :before_step do |step_name, payload| ... end
+      #   on :after_step  do |step_name, result|  ... end
+      #   on :stopped     do |step_name, result|  ... end
+      #   on :complete    do |last_result|         ... end
+      #
+      # Per-step hooks fire only for the named step (no step_name passed):
+      #   on :before_step, :translate do |payload| ... end
+      #   on :after_step,  :translate do |result|  ... end
+      def on(event, step_name = nil, &block)
+        if step_name
+          unless VALID_STEP_HOOKS.include?(event)
+            raise ArgumentError,
+              "Per-step hooks support: #{VALID_STEP_HOOKS.join(", ")}. Got :#{event}"
+          end
+          pipeline_config[:step_hooks][step_name] ||= {}
+          (pipeline_config[:step_hooks][step_name][event] ||= []) << block
+        else
+          unless VALID_HOOKS.include?(event)
+            raise ArgumentError,
+              "Unknown Pipeline hook :#{event}. Valid: #{VALID_HOOKS.join(", ")}"
+          end
+          (pipeline_config[:hooks][event] ||= []) << block
+        end
+      end
+
+      # Rails-style aliases for +on+:
+      #
+      # Global:
+      #   before :step                          do |name, payload| end  # → on :before_step
+      #   after  :step                          do |name, result|  end  # → on :after_step
+      #   callback :stopped                     do |name, result|  end  # → on :stopped
+      #   callback :complete                    do |result|        end  # → on :complete
+      #
+      # Per-step:
+      #   after  :step, :translate              do |result| end
+      #   before :step, :translate              do |payload| end
+      def before(event, step_name = nil, &block)
+        on(:"before_#{event}", step_name, &block)
+      end
+
+      def after(event, step_name = nil, &block)
+        on(:"after_#{event}", step_name, &block)
+      end
+
+      def callback(event, &block)
+        on(event, &block)
+      end
+    end
+
+    include Core::HookRunner
+
+    private
+
+    # Fires global hook AND pipeline_event_stream. Consistent with Agent#fire and Tribunal#fire.
+    def fire(event, step_name, data, config)
+      run_hooks(config[:hooks], event, step_name, data)
+      @pipeline_event_stream&.call(event, step_name, data)
+    rescue IOError, ActionController::Live::ClientDisconnected
+      nil
+    end
+
+    # Per-step hook: receives (data) only — not forwarded to pipeline_event_stream
+    # (global fire already covers the step event with step_name context).
+    def fire_step(event, step_name, data, config)
+      run_hooks(config[:step_hooks][step_name] || {}, event, data)
+    end
+  end
+end

data/lib/active_harness/pipeline.rb

@@ -32,9 +32,6 @@ module ActiveHarness
   #   pipeline.step_results # => { translate: <Result>, ... }
   #
   class Pipeline
-    VALID_HOOKS      = %i[before_step after_step stopped complete].freeze
-    VALID_STEP_HOOKS = %i[before_step after_step].freeze
-
     # -------------------------------------------------------------------------
     # Class-level DSL
     # -------------------------------------------------------------------------
@@ -53,57 +50,6 @@ module ActiveHarness
         pipeline_config[:steps] << Pipeline::Step.new(name, agent_class, &block)
       end
 
-      # Register a global or per-step hook.
-      #
-      # Global hooks fire on every step:
-      #   on :before_step do |step_name, payload| ... end
-      #   on :after_step  do |step_name, result|  ... end
-      #   on :stopped     do |step_name, result|  ... end
-      #   on :complete    do |last_result|         ... end
-      #
-      # Per-step hooks fire only for the named step (no step_name passed):
-      #   on :before_step, :translate do |payload| ... end
-      #   on :after_step,  :translate do |result|  ... end
-      def on(event, step_name = nil, &block)
-        if step_name
-          unless VALID_STEP_HOOKS.include?(event)
-            raise ArgumentError,
-              "Per-step hooks support: #{VALID_STEP_HOOKS.join(", ")}. Got :#{event}"
-          end
-          pipeline_config[:step_hooks][step_name] ||= {}
-          pipeline_config[:step_hooks][step_name][event] = block
-        else
-          unless VALID_HOOKS.include?(event)
-            raise ArgumentError,
-              "Unknown Pipeline hook :#{event}. Valid: #{VALID_HOOKS.join(", ")}"
-          end
-          pipeline_config[:hooks][event] = block
-        end
-      end
-
-      # Rails-style aliases for +on+:
-      #
-      # Global:
-      #   before :step                          do |name, payload| end  # → on :before_step
-      #   after  :step                          do |name, result|  end  # → on :after_step
-      #   callback :stopped                     do |name, result|  end  # → on :stopped
-      #   callback :complete                    do |result|        end  # → on :complete
-      #
-      # Per-step:
-      #   after  :step, :translate              do |result| end
-      #   before :step, :translate              do |payload| end
-      def before(event, step_name = nil, &block)
-        on(:"before_#{event}", step_name, &block)
-      end
-
-      def after(event, step_name = nil, &block)
-        on(:"after_#{event}", step_name, &block)
-      end
-
-      def callback(event, &block)
-        on(event, &block)
-      end
-
       def pipeline_config
         @pipeline_config ||= { steps: [], hooks: {}, step_hooks: {} }
       end
@@ -120,19 +66,32 @@ module ActiveHarness
     # -------------------------------------------------------------------------
     # Instance API
     # -------------------------------------------------------------------------
-    attr_reader :original_input, :output, :stopped_at, :stop_reason,
-                :execution_time, :step_results, :context
-    attr_writer :context
+    attr_reader   :original_input,
+                  :output,
+                  :stopped_at,
+                  :stop_reason,
+                  :execution_time,
+                  :step_results,
+                  :context
+    attr_writer   :context
+    attr_accessor :params
 
     def input=(value)
       @original_input = value
       @payload        = value
     end
 
-    def initialize(input:, context: {}, memory: nil, streams: {})
+    def initialize(
+      input:,
+      context: {},
+      params:  {},
+      memory:  nil,
+      streams: {}
+    )
       @original_input        = input
       @payload               = input
       @context               = context.dup
+      @params                = params
       @memory                = memory
       @token_stream          = streams[:token]
       @agent_event_stream    = streams[:agent]
@@ -174,8 +133,7 @@ module ActiveHarness
           @stopped     = true
           @stopped_at  = step.name
           @stop_reason = result
-          blk = config[:hooks][:stopped]
-          instance_exec(step.name, result, &blk) if blk
+          run_hooks(config[:hooks], :stopped, step.name, result)
           @pipeline_event_stream&.call(:stopped, step.name, result)
           break
         end
@@ -192,8 +150,7 @@ module ActiveHarness
         )
 
         last_result = @step_results[@step_results.keys.last]
-        blk = config[:hooks][:complete]
-        instance_exec(last_result, &blk) if blk
+        run_hooks(config[:hooks], :complete, last_result)
         @pipeline_event_stream&.call(:complete, last_result)
       end
 
@@ -208,6 +165,7 @@ module ActiveHarness
         step.agent_class.new(
           input:    @payload,
           context:  @context.dup,
+          params:   @params,
           streams:  agent_streams
         ).call
       else
@@ -215,27 +173,13 @@ module ActiveHarness
         step.agent_class.new(
           input:    @payload,
           context:  @context.dup,
+          params:   @params,
           streams:  agent_streams
         ).call.result
       end
     end
-
-    # Fires global hook AND pipeline_event_stream. Consistent with Agent#fire and Tribunal#fire.
-    def fire(event, step_name, data, config)
-      blk = config[:hooks][event]
-      instance_exec(step_name, data, &blk) if blk
-      @pipeline_event_stream&.call(event, step_name, data)
-    rescue IOError, ActionController::Live::ClientDisconnected
-      nil
-    end
-
-    # Per-step hook: receives (data) only — not forwarded to pipeline_event_stream
-    # (global fire already covers the step event with step_name context).
-    def fire_step(event, step_name, data, config)
-      blk = config[:step_hooks][step_name]&.dig(event)
-      instance_exec(data, &blk) if blk
-    end
   end
 end
 
+require_relative "pipeline/hooks"
 require_relative "pipeline/step"

data/lib/active_harness/tribunal/hooks.rb

@@ -26,7 +26,7 @@ module ActiveHarness
             "Unknown Tribunal hook :#{event}. Valid hooks: #{VALID_HOOKS.map { |h| ":#{h}" }.join(", ")}"
         end
 
-        tribunal_config[:hooks][event] = block
+        (tribunal_config[:hooks][event] ||= []) << block
       end
 
       # Rails-style aliases for +on+:
@@ -51,7 +51,7 @@ module ActiveHarness
       end
     end
 
-    # Instance-level hook registration — overrides class-level hooks for this instance.
+    # Instance-level hook registration — appends to class-level hooks for this event.
     # :before_verdict is a transform hook: its return value replaces the results array
     # passed to the process block.
     def on(event, &block)
@@ -60,20 +60,16 @@ module ActiveHarness
           "Unknown Tribunal hook :#{event}. Valid hooks: #{VALID_HOOKS.map { |h| ":#{h}" }.join(", ")}"
       end
 
-      @hooks[event] = block
+      (@hooks[event] ||= []) << block
       self
     end
 
+    include Core::HookRunner
+
     private
 
     def run_hook(event, *args)
-      return unless @hooks[event]
-
-      if args.any?
-        instance_exec(*args, &@hooks[event])
-      else
-        instance_eval(&@hooks[event])
-      end
+      run_hooks(@hooks, event, *args)
     end
 
     # Fire the DSL-registered hook AND the external tribunal_event_stream lambda (if set).
@@ -84,12 +80,10 @@ module ActiveHarness
       nil
     end
 
-    # Like run_hook but uses the return value to replace the passed value.
+    # Like run_hook but chains all blocks, passing each return value to the next.
     # Used by :before_verdict to allow results transformation before verdict computation.
     def transform_hook(event, value)
-      return value unless @hooks[event]
-
-      instance_exec(value, &@hooks[event])
+      Array(@hooks[event]).reduce(value) { |val, blk| instance_exec(val, &blk) }
     end
   end
 end

data/lib/active_harness/tribunal.rb

@@ -46,24 +46,39 @@ module ActiveHarness
     # -------------------------------------------------------------------------
     # Instance API
     # -------------------------------------------------------------------------
-    attr_accessor :input, :context
-    attr_reader :results, :errors, :verdict, :execution_time, :agent_execution_times,
-                :token_stream, :agent_event_stream, :tribunal_event_stream
-
-    def initialize(input: nil, context: {}, agents: nil, timeout: 7,
-                   streams: {},
-                   may_fail: :_unset)
+    attr_accessor :input,
+                  :context,
+                  :params
+    attr_reader   :results,
+                  :errors,
+                  :verdict,
+                  :execution_time,
+                  :agent_execution_times,
+                  :token_stream,
+                  :agent_event_stream,
+                  :tribunal_event_stream
+
+    def initialize(
+      input:    nil,
+      context:  {},
+      params:   {},
+      agents:   nil,
+      timeout:  7,
+      streams:  {},
+      may_fail: :_unset
+    )
       config = self.class.tribunal_config
 
       @input                 = input
       @context               = context
+      @params                = params
       @agents                = agents || config[:agents]
       @timeout               = timeout
       @process_block         = config[:process]
       @strategy              = config[:strategy]
       @evaluate_block        = config[:evaluate_block]
       @may_fail              = may_fail == :_unset ? config[:may_fail] : may_fail
-      @hooks                 = config[:hooks].dup
+      @hooks                 = config[:hooks].transform_values { |v| Array(v).dup }
       @token_stream          = streams[:token]
       @agent_event_stream    = streams[:agent]
       @tribunal_event_stream = streams[:tribunal]
@@ -155,7 +170,7 @@ module ActiveHarness
       agent_streams = { token: @token_stream, agent: @agent_event_stream }.compact
       @agents.map do |agent|
         if agent.is_a?(Class)
-          agent.new(input: @input, context: @context.dup, streams: agent_streams)
+          agent.new(input: @input, context: @context.dup, params: @params, streams: agent_streams)
         else
           agent.input = @input if @input
           agent.instance_variable_set(:@token_stream, @token_stream) if @token_stream

data/lib/active_harness.rb

@@ -1,5 +1,6 @@
 require_relative "active_harness/configuration"
 require_relative "active_harness/core/errors"
+require_relative "active_harness/core/hooks"
 require_relative "active_harness/result"
 require_relative "active_harness/http/client"
 require_relative "active_harness/http/streaming_client"
@@ -29,7 +30,7 @@ require_relative "active_harness/pipeline"
 require_relative "active_harness/railtie" if defined?(Rails::Railtie)
 
 module ActiveHarness
-  VERSION = "0.2.20"
+  VERSION = "0.2.22"
 
   class << self
     # Configure ActiveHarness.

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: active_harness
 version: !ruby/object:Gem::Version
-  version: 0.2.20
+  version: 0.2.22
 platform: ruby
 authors:
 - the-teacher
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2026-05-27 00:00:00.000000000 Z
+date: 2026-05-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby
@@ -42,6 +42,7 @@ files:
 - lib/active_harness/agent/providers.rb
 - 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
@@ -51,6 +52,7 @@ files:
 - lib/active_harness/memory/adapter/base.rb
 - lib/active_harness/memory/adapter/file.rb
 - lib/active_harness/pipeline.rb
+- lib/active_harness/pipeline/hooks.rb
 - lib/active_harness/pipeline/step.rb
 - lib/active_harness/providers/PROVIDER_CONTRACT.md
 - lib/active_harness/providers/anthropic.rb