active_harness 0.2.18 → 0.2.20

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2e8e3cbcdf3101e80ac7872c4249e271e847ce2d925cf63d869e34bf5f08fc3d
-  data.tar.gz: 73f663a4e25f23c431b35c53a464c440dcccc793fb71844e9df5428c11283c68
+  metadata.gz: d1188ca468f31f46c40d118b7e719a849925aa11ddab3a4a87ea4df0fd9bdb84
+  data.tar.gz: 1ed36314efd943e6f4e62bbc43b2ec00d0b0486bdbf015e6b590341d240ffcdc
 SHA512:
-  metadata.gz: 46a7fd94cc3d98427e466a30daff7baae44a8683ad6fe6235988cad5dbc33bdf8f4170a89d3108f51eb84aeafdfe596e04490a556c01ea8d7bc7603d18894383
-  data.tar.gz: 669ee179cc1f1e2eae8a5b9a0f0e026fa2752a601d7a9388d5956f065e07ba93b3edc03b770017dd3ba9445c02f2623ae700e18921c67de6534b75883bfb6717
+  metadata.gz: 37e82033bddd4069f26619d2bfc5faec831c32618216bedb6cb9ce19ca7fbbaac1057e829809ee15f1d88d1a6fc3b3832cd2c02d1f112c850a974a26205d3727
+  data.tar.gz: 3a4fafd8184d55f116689e28c4f411d102f7279d290093c9cc32dafabfc9eafe9aa55abde59a04cec37985e74da94a8ddb14153febce35aa41c2eb0404eed2c6

data/lib/active_harness/agent/hooks.rb

@@ -71,5 +71,15 @@ module ActiveHarness
         instance_eval(&hooks[event])
       end
     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)
+      @event_stream&.call(event, *args)
+      result
+    rescue IOError, ActionController::Live::ClientDisconnected
+      result
+    end
   end
 end

data/lib/active_harness/agent/output_parser.rb

@@ -49,7 +49,7 @@ module ActiveHarness
         #     puts "Parse failed: #{error.message}"
         #     { "result" => nil, "reason" => "parse error" }   # fallback
         #   end
-        fallback = run_hook(:parse_error, raw, e)
+        fallback = fire(:parse_error, raw, e)
         fallback.nil? ? raise : fallback
       end
     end

data/lib/active_harness/agent/prompt.rb

@@ -22,7 +22,7 @@ module ActiveHarness
     public :system_prompt
 
     def resolve_system_prompt
-      run_hook(:before_system_prompt)
+      fire(:before_system_prompt)
 
       sp = @config[:system_prompt]
 
@@ -45,7 +45,7 @@ module ActiveHarness
         end
       end
 
-      run_hook(:after_system_prompt, prompt)
+      fire(:after_system_prompt, prompt)
       prompt
     end
 

data/lib/active_harness/agent.rb

@@ -11,8 +11,8 @@ 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, stream: nil, token_stream: nil, event_stream: nil)
-        new(input: input, context: context, models: models, memory: memory, stream: stream, token_stream: token_stream, event_stream: event_stream).call
+      def call(input: nil, context: {}, models: nil, memory: nil, streams: {})
+        new(input: input, context: context, models: models, memory: memory, streams: streams).call
       end
 
       # Each subclass gets its own isolated config hash.
@@ -36,8 +36,8 @@ module ActiveHarness
     # -------------------------------------------------------------------------
     # Instance API
     # -------------------------------------------------------------------------
-    attr_accessor :input, :context, :stream, :token_stream, :event_stream
-    attr_reader :result
+    attr_accessor :input, :context
+    attr_reader   :result, :token_stream, :event_stream
 
     def models=(list)
       @models_override = Array(list)
@@ -48,18 +48,17 @@ module ActiveHarness
       @memory = obj
     end
 
-    def initialize(input: nil, context: {}, models: nil, memory: nil, stream: nil, token_stream: nil, event_stream: nil)
+    def initialize(input: nil, context: {}, models: nil, memory: nil, streams: {})
       @input           = input
       @config          = self.class.agent_config
       normalize_input!
       @context         = context
       @models_override = Array(models) if models
-      @stream          = stream
-      @token_stream    = token_stream
-      @event_stream    = event_stream
+      @token_stream    = streams[:token]
+      @event_stream    = streams[:agent]
       # memory: can be passed directly or via context[:memory]
       @memory = memory || @context[:memory]
-      run_hook(:setup)
+      fire(:setup)
     end
 
     # Attempts each model in order, returns the first successful Result.
@@ -68,15 +67,18 @@ module ActiveHarness
     # Optionally accepts input and stream callback inline:
     #   agent.call("What is the capital of Japan?")
     #   agent.call("...", stream: ->(token) { print token })
-    def call(input = nil, token_stream: nil)
+    def call(input = nil, streams: nil)
       if input
         @input = input
         normalize_input!
       end
-      @token_stream = token_stream if token_stream
+      if streams
+        @token_stream = streams[:token] if streams.key?(:token)
+        @event_stream = streams[:agent] if streams.key?(:agent)
+      end
       @memory&.load
       @system_prompt = resolve_system_prompt
-      run_hook(:before_call)
+      fire(:before_call)
       attempts = []
 
       cfg = ActiveHarness.config
@@ -91,22 +93,22 @@ module ActiveHarness
         elapsed  = (Process.clock_gettime(Process::CLOCK_MONOTONIC) - t0).round(3)
         result   = build_result(response, entry, attempts, elapsed)
         save_to_memory(result)
-        run_hook(:after_call, result)
+        fire(:after_call, result)
         @result = result
         return self
       rescue *RETRYABLE_ERRORS => e
         elapsed = (Process.clock_gettime(Process::CLOCK_MONOTONIC) - t0).round(3)
         attempts << attempt_entry(entry, e, elapsed)
-        run_hook(:retry, entry, e)
+        fire(:retry, entry, e)
         next
       rescue *STOP_ERRORS => e
         elapsed = (Process.clock_gettime(Process::CLOCK_MONOTONIC) - t0).round(3)
         attempts << attempt_entry(entry, e, elapsed)
-        run_hook(:retry, entry, e)
+        fire(:retry, entry, e)
         raise
       end
 
-      run_hook(:failure, attempts)
+      fire(:failure, attempts)
       raise Errors::AllModelsFailed, "All models failed. Attempts: #{attempts.inspect}"
     end
 

data/lib/active_harness/pipeline.rb

@@ -129,23 +129,21 @@ module ActiveHarness
       @payload        = value
     end
 
-    def initialize(input:, context: {}, memory: nil,
-                   stream: nil, agent_event_stream: nil,
-                   tribunal_event_stream: nil, pipeline_event_stream: nil)
-      @original_input           = input
-      @payload                  = input
-      @context                  = context.dup
-      @memory                   = memory
-      @stream                   = stream
-      @agent_event_stream       = agent_event_stream
-      @tribunal_event_stream    = tribunal_event_stream
-      @pipeline_event_stream    = pipeline_event_stream
-      @step_results             = {}
-      @stopped                  = false
-      @stopped_at               = nil
-      @stop_reason              = nil
-      @execution_time           = nil
-      @output                   = nil
+    def initialize(input:, context: {}, memory: nil, streams: {})
+      @original_input        = input
+      @payload               = input
+      @context               = context.dup
+      @memory                = memory
+      @token_stream          = streams[:token]
+      @agent_event_stream    = streams[:agent]
+      @tribunal_event_stream = streams[:tribunal]
+      @pipeline_event_stream = streams[:pipeline]
+      @step_results          = {}
+      @stopped               = false
+      @stopped_at            = nil
+      @stop_reason           = nil
+      @execution_time        = nil
+      @output                = nil
     end
 
     def stopped?
@@ -160,7 +158,7 @@ module ActiveHarness
       @memory&.load
 
       config[:steps].each do |step|
-        fire_global(:before_step, step.name, @payload, config)
+        fire(:before_step, step.name, @payload, config)
         fire_step(:before_step, step.name, @payload, config)
 
         result = execute_step(step)
@@ -169,7 +167,7 @@ module ActiveHarness
         @context[step.name]      = result
         @payload                 = result.output if step.transform?
 
-        fire_global(:after_step, step.name, result, config)
+        fire(:after_step, step.name, result, config)
         fire_step(:after_step, step.name, result, config)
 
         if step.stop_if && step.stop_if.call(result)
@@ -178,6 +176,7 @@ module ActiveHarness
           @stop_reason = result
           blk = config[:hooks][:stopped]
           instance_exec(step.name, result, &blk) if blk
+          @pipeline_event_stream&.call(:stopped, step.name, result)
           break
         end
       end
@@ -195,6 +194,7 @@ module ActiveHarness
         last_result = @step_results[@step_results.keys.last]
         blk = config[:hooks][:complete]
         instance_exec(last_result, &blk) if blk
+        @pipeline_event_stream&.call(:complete, last_result)
       end
 
       self
@@ -204,32 +204,33 @@ module ActiveHarness
 
     def execute_step(step)
       if step.tribunal?
-        agent = step.agent_class.new(
-          input:                 @payload,
-          context:               @context.dup,
-          stream:                @stream,
-          agent_event_stream:    @agent_event_stream,
-          tribunal_event_stream: @tribunal_event_stream
-        )
-        agent.call
+        agent_streams = { token: @token_stream, agent: @agent_event_stream, tribunal: @tribunal_event_stream }.compact
+        step.agent_class.new(
+          input:    @payload,
+          context:  @context.dup,
+          streams:  agent_streams
+        ).call
       else
-        agent = step.agent_class.new(
-          input:        @payload,
-          context:      @context.dup,
-          stream:       @stream,
-          event_stream: @agent_event_stream
-        )
-        agent.call.result
+        agent_streams = { token: @token_stream, agent: @agent_event_stream }.compact
+        step.agent_class.new(
+          input:    @payload,
+          context:  @context.dup,
+          streams:  agent_streams
+        ).call.result
       end
     end
 
-    # Global hook: receives (step_name, data)
-    def fire_global(event, step_name, data, config)
+    # 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
+    # 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

data/lib/active_harness/tribunal.rb

@@ -46,11 +46,12 @@ module ActiveHarness
     # -------------------------------------------------------------------------
     # Instance API
     # -------------------------------------------------------------------------
-    attr_accessor :input, :context, :stream, :agent_event_stream, :tribunal_event_stream
-    attr_reader :results, :errors, :verdict, :execution_time, :agent_execution_times
+    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,
-                   stream: nil, agent_event_stream: nil, tribunal_event_stream: nil,
+                   streams: {},
                    may_fail: :_unset)
       config = self.class.tribunal_config
 
@@ -63,9 +64,9 @@ module ActiveHarness
       @evaluate_block        = config[:evaluate_block]
       @may_fail              = may_fail == :_unset ? config[:may_fail] : may_fail
       @hooks                 = config[:hooks].dup
-      @stream                = stream
-      @agent_event_stream    = agent_event_stream
-      @tribunal_event_stream = tribunal_event_stream
+      @token_stream          = streams[:token]
+      @agent_event_stream    = streams[:agent]
+      @tribunal_event_stream = streams[:tribunal]
       @results               = []
       @errors                = []
       @verdict               = nil
@@ -151,18 +152,14 @@ module ActiveHarness
     end
 
     def resolve_agents
+      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,
-            stream:       @stream,
-            event_stream: @agent_event_stream
-          )
+          agent.new(input: @input, context: @context.dup, streams: agent_streams)
         else
-          agent.input        = @input if @input
-          agent.stream       = @stream if @stream
-          agent.event_stream = @agent_event_stream if @agent_event_stream
+          agent.input = @input if @input
+          agent.instance_variable_set(:@token_stream, @token_stream) if @token_stream
+          agent.instance_variable_set(:@event_stream, @agent_event_stream) if @agent_event_stream
           agent
         end
       end

data/lib/active_harness.rb

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

metadata

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