pwn 0.5.659 → 0.5.660

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1cab86775d3b1611d7b0aadf6bd7ceb11ba1e673d239ff4b141bbf29a0212217
-  data.tar.gz: 1833f2e8bff11bbeedfc059734c99d3b1743291795829a65d24066e8319d8272
+  metadata.gz: '06678f228c6acc69d9fb173f01f3825af6a7a8dcffcb8ed24c865bf281e0c65c'
+  data.tar.gz: 9967223f974d122e30cbb4be7aea369108e28b951ace091a1ed90d54825a4720
 SHA512:
-  metadata.gz: 44a37c7080857da5ca07df39e9bb4856d8c30c621b77e8348be7f4257afdcb5c05bd4d0c4816b67b20415c5e22c80d44b4bf276823b02c7028067c4bdbae95bd
-  data.tar.gz: 2ddbac637da6b33a5f96e45d58f9210ec843fd6077b5815ccd1698d13143d7ec645aa2a7c2dc2a23a84b93e0d1cc4498283119565d61152af49805704b4ab58e
+  metadata.gz: 6feb604302f7e6fd428e8dcd2d1c6a888ed2e6d4c96f3e7b75b6537a5694c3053da6a69da751275c8c168f627f6b71b031ed502cf02b1b0eb8bb2dfd72e2e354
+  data.tar.gz: 26f65f4e16312694c1163b0de7fe0545ca8ad7b8e1cdea647d931f131de6aaef89f0fe8b39be72509932295c1ad1521066c88722d5654cb6a856891c1c19e860

data/lib/pwn/ai/agent/task_summarizer.rb

@@ -0,0 +1,119 @@
+# frozen_string_literal: true
+
+module PWN
+  module AI
+    module Agent
+      # Coalesce a burst of tool calls into periodic
+      # "summary_of_current_task" UI lines without stopping the loop.
+      module TaskSummarizer
+        DEFAULT_EVERY = 5
+        DEFAULT_INTERVAL_S = 8.0
+        MAX_BUFFER = 64
+
+        module_function
+
+        def enabled?
+          v = PWN::Env.dig(:ai, :agent, :task_summary)
+          v.nil? || !!v
+        rescue StandardError
+          true
+        end
+
+        def verbose?
+          !!PWN::Env.dig(:ai, :agent, :task_summary_verbose)
+        rescue StandardError
+          false
+        end
+
+        def every_n
+          n = PWN::Env.dig(:ai, :agent, :task_summary_every)
+          n = DEFAULT_EVERY if n.nil?
+          [n.to_i, 1].max
+        rescue StandardError
+          DEFAULT_EVERY
+        end
+
+        def interval_s
+          t = PWN::Env.dig(:ai, :agent, :task_summary_interval_s)
+          t = DEFAULT_INTERVAL_S if t.nil?
+          [t.to_f, 1.0].max
+        rescue StandardError
+          DEFAULT_INTERVAL_S
+        end
+
+        # Per-run state (also safe for nested/swarm if callers keep their own hash)
+        def fresh(opts = {})
+          {
+            request: opts[:request].to_s,
+            events: [],
+            since_emit: 0,
+            last_emit_at: Time.now,
+            total: 0,
+            counts: Hash.new(0)
+          }
+        end
+
+        def record!(state, name, args, result)
+          return nil unless state
+
+          preview = args.is_a?(String) ? args.to_s[0, 60] : args.inspect[0, 60]
+          rs = result.to_s
+          ok = !rs.match?(/\A\s*\{?\s*"?(success|ok)"?\s*=>\s*false/i) &&
+               !rs.match?(/ERROR:|Traceback|NoMethodError|StandardError/i)
+          state[:events] << { name: name.to_s, preview: preview, ok: ok, t: Time.now }
+          state[:events].shift while state[:events].size > MAX_BUFFER
+          state[:counts][name.to_s] += 1
+          state[:total] += 1
+          state[:since_emit] += 1
+
+          due = state[:since_emit] >= every_n ||
+                (Time.now - state[:last_emit_at]) >= interval_s
+          due ? emit!(state) : nil
+        end
+
+        def emit!(state, final: false)
+          return nil if state.nil? || state[:events].empty?
+
+          counts = state[:counts].sort_by { |_, c| -c }.map { |n, c| "#{n}×#{c}" }
+          recent = state[:events].last(every_n)
+          focus = recent.map { |e| e[:name] }.uniq.first(4).join(', ')
+          fails = state[:events].count { |e| !e[:ok] }
+          tail = recent.map { |e| e[:preview] }.compact.reject(&:empty?).last
+          bit = tail && !tail.empty? ? "; e.g. #{tail}" : ''
+          fail_bit = fails.positive? ? "; failures=#{fails}" : ''
+          phase = final ? 'done' : 'in progress'
+          state[:since_emit] = 0
+          state[:last_emit_at] = Time.now
+          "#{phase}: #{focus} — #{state[:total]} tools (#{counts.first(6).join(', ')})#{fail_bit}#{bit}"
+        end
+
+        def flush!(state)
+          emit!(state, final: true)
+        end
+
+        # Author(s):: 0day Inc. <support@0dayinc.com>
+
+        public_class_method def self.authors
+          "AUTHOR(S):\n  0day Inc. <support@0dayinc.com>\n"
+        end
+
+        # Display Usage for this Module
+
+        public_class_method def self.help
+          puts <<~USAGE
+            USAGE:
+              state = PWN::AI::Agent::TaskSummarizer.fresh(request: 'do the thing')
+              line  = PWN::AI::Agent::TaskSummarizer.record!(state, 'shell', 'ls', '{success:true}')
+              line  = PWN::AI::Agent::TaskSummarizer.flush!(state)
+              PWN::AI::Agent::TaskSummarizer.enabled?
+              PWN::AI::Agent::TaskSummarizer.verbose?
+              PWN::AI::Agent::TaskSummarizer.every_n
+              PWN::AI::Agent::TaskSummarizer.interval_s
+
+              #{self}.authors
+          USAGE
+        end
+      end
+    end
+  end
+end

data/lib/pwn/ai/agent.rb

@@ -31,6 +31,7 @@ module PWN
       autoload :Swarm,         'pwn/ai/agent/swarm'
       autoload :Reward,        'pwn/ai/agent/reward'
       autoload :Curriculum,    'pwn/ai/agent/curriculum'
+      autoload :TaskSummarizer, 'pwn/ai/agent/task_summarizer'
 
       # Display a List of Every PWN::AI Module
 

data/lib/pwn/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module PWN
-  VERSION = '0.5.659'
+  VERSION = '0.5.660'
 end

data/spec/lib/pwn/ai/agent/task_summarizer_spec.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe PWN::AI::Agent::TaskSummarizer do
+  it 'fresh method should exist' do
+    expect(described_class).to respond_to :fresh
+  end
+
+  it 'record! method should exist' do
+    expect(described_class).to respond_to :record!
+  end
+
+  it 'emit! method should exist' do
+    expect(described_class).to respond_to :emit!
+  end
+
+  it 'flush! method should exist' do
+    expect(described_class).to respond_to :flush!
+  end
+
+  it 'enabled? method should exist' do
+    expect(described_class).to respond_to :enabled?
+  end
+
+  it 'should display information for authors' do
+    expect(described_class).to respond_to :authors
+  end
+
+  it 'should display information for existing help method' do
+    expect(described_class).to respond_to :help
+  end
+
+  it 'coalesces tool bursts into a summary line' do
+    allow(PWN::Env).to receive(:dig).and_call_original
+    allow(PWN::Env).to receive(:dig).with(:ai, :agent, :task_summary_every).and_return(2)
+    allow(PWN::Env).to receive(:dig).with(:ai, :agent, :task_summary_interval_s).and_return(9_999)
+    allow(PWN::Env).to receive(:dig).with(:ai, :agent, :task_summary).and_return(true)
+    allow(PWN::Env).to receive(:dig).with(:ai, :agent, :task_summary_verbose).and_return(false)
+
+    state = described_class.fresh(request: 'fix rake')
+    expect(described_class.record!(state, 'shell', 'rake', '{success:true}')).to be_nil
+    line = described_class.record!(state, 'shell', 'rake 2>&1', '{success:true}')
+    expect(line).to be_a(String)
+    expect(line).to include('in progress')
+    expect(line).to include('shell')
+    expect(state[:total]).to eq 2
+
+    fin = described_class.flush!(state)
+    expect(fin).to include('done')
+    expect(fin).to include('2 tools')
+  end
+end

data/third_party/pwn_rdoc.jsonl

@@ -460,6 +460,16 @@
 {"messages":[{"role":"user","content":"PWN::AI::Agent::Swarm.retire Usage"},{"role":"assistant","content":"`PWN::AI::Agent::Swarm.retire`: Supported Method Parameters\n\nPWN::AI::Agent::Swarm.retire(name: ‘required - persona name’)\n"}]}
 {"messages":[{"role":"user","content":"PWN::AI::Agent::Swarm.spawn Usage"},{"role":"assistant","content":"`PWN::AI::Agent::Swarm.spawn`: Supported Method Parameters\n\nPWN::AI::Agent::Swarm.spawn(\n\nname: 'required - persona name (snake_case)',\nrole: 'required - system_role_content overlay for this persona',\ntoolsets: 'optional - Array of Registry toolset names',\nengine: 'optional - :openai / :anthropic / :grok / :gemini / :ollama',\nmax_iters: 'optional - per-turn iteration cap for this persona'\n\n)\n"}]}
 {"messages":[{"role":"user","content":"PWN::AI::Agent::Swarm.with_persona_env Usage"},{"role":"assistant","content":"`PWN::AI::Agent::Swarm.with_persona_env`: "}]}
+{"messages":[{"role":"user","content":"PWN::AI::Agent::TaskSummarizer.authors Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TaskSummarizer.authors`: Author(s)\n\n0day Inc. <support@0dayinc.com>\n"}]}
+{"messages":[{"role":"user","content":"PWN::AI::Agent::TaskSummarizer.help Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TaskSummarizer.help`: "}]}
+{"messages":[{"role":"user","content":"PWN::AI::Agent::TaskSummarizer.emit! Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TaskSummarizer.emit!`: "}]}
+{"messages":[{"role":"user","content":"PWN::AI::Agent::TaskSummarizer.enabled? Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TaskSummarizer.enabled?`: "}]}
+{"messages":[{"role":"user","content":"PWN::AI::Agent::TaskSummarizer.every_n Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TaskSummarizer.every_n`: "}]}
+{"messages":[{"role":"user","content":"PWN::AI::Agent::TaskSummarizer.flush! Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TaskSummarizer.flush!`: "}]}
+{"messages":[{"role":"user","content":"PWN::AI::Agent::TaskSummarizer.fresh Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TaskSummarizer.fresh`: "}]}
+{"messages":[{"role":"user","content":"PWN::AI::Agent::TaskSummarizer.interval_s Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TaskSummarizer.interval_s`: "}]}
+{"messages":[{"role":"user","content":"PWN::AI::Agent::TaskSummarizer.record! Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TaskSummarizer.record!`: "}]}
+{"messages":[{"role":"user","content":"PWN::AI::Agent::TaskSummarizer.verbose? Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TaskSummarizer.verbose?`: "}]}
 {"messages":[{"role":"user","content":"PWN::AI::Agent::TransparentBrowser.analyze Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TransparentBrowser.analyze`: Supported Method Parameters\n\nai_analysis = PWN::AI::Agent::TransparentBrowser.analyze(\n\nrequest: 'required - current step in the JavaScript debugging session to analyze',\nsource_to_review: 'required - the block of JavaScript code in which the current step resides'\n\n)\n"}]}
 {"messages":[{"role":"user","content":"PWN::AI::Agent::TransparentBrowser.authors Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TransparentBrowser.authors`: Author(s)\n\n0day Inc. <support@0dayinc.com>\n"}]}
 {"messages":[{"role":"user","content":"PWN::AI::Agent::TransparentBrowser.help Usage"},{"role":"assistant","content":"`PWN::AI::Agent::TransparentBrowser.help`: "}]}

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.5.659
+  version: 0.5.660
 platform: ruby
 authors:
 - 0day Inc.
@@ -1856,6 +1856,7 @@ files:
 - lib/pwn/ai/agent/reward.rb
 - lib/pwn/ai/agent/sast.rb
 - lib/pwn/ai/agent/swarm.rb
+- lib/pwn/ai/agent/task_summarizer.rb
 - lib/pwn/ai/agent/tools/cron.rb
 - lib/pwn/ai/agent/tools/curriculum.rb
 - lib/pwn/ai/agent/tools/extrospection.rb
@@ -2317,6 +2318,7 @@ files:
 - spec/lib/pwn/ai/agent/reward_spec.rb
 - spec/lib/pwn/ai/agent/sast_spec.rb
 - spec/lib/pwn/ai/agent/swarm_spec.rb
+- spec/lib/pwn/ai/agent/task_summarizer_spec.rb
 - spec/lib/pwn/ai/agent/tools/cron_spec.rb
 - spec/lib/pwn/ai/agent/tools/curriculum_spec.rb
 - spec/lib/pwn/ai/agent/tools/extrospection_spec.rb