pwn 0.5.679 → 0.5.680

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 81490e3fc28a7f1e743e069075ce57da65f4a54548b47f72d7467d22fab64432
-  data.tar.gz: f77147d5e3bb6b53d994ff7f7af3c21e260a7659a60cda70b599fec1c565176d
+  metadata.gz: ca4257e59d3ee50dde626ddb4aabf356b578ebd2bb82a23f3b724cae214d6d09
+  data.tar.gz: 62c7c4a1b139769fe86e20ea0243f0cc2d7155eb15421dcd2e29c640b6bd838d
 SHA512:
-  metadata.gz: d034c13085bdeffac43c2ded717317a7dc70dce6cb72d828f7879ba4847abf26fcc4258820551e4410b2d571ad925ccd49843aa2cd94ee3add1c0071412b6e6c
-  data.tar.gz: 4901e7063ec2fc27ffd60349a5f6681a46d7bea30baf3c70f1d03c3cb53903e0face0946c558ad2dbd82358864f6f29d2d06cb04f0ae7c40cd3360cf5ffe6986
+  metadata.gz: adca8d361a9535a64bac1804a5cf5ae04b1736bfa4ecde55be7b956dfa33d02e7f8e75827bd76164af9c103b6e1d188fa53ed6627fe38677ff1263512c202795
+  data.tar.gz: 0de0c0e020d650c893322714b7ea90a1a91c9d0a4e38c475fad8e37584711c0f8d985571e8514fcaf05fd0d8c2a02010974b44fd3fe923ca1ee89014002a8be0

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

@@ -1013,7 +1013,7 @@ module PWN
                 cwt_opts[:tool_choice] = has_tool_result && !still_acting ? 'auto' : 'required'
               end
             end
-            response = mod.chat_with_tools(**cwt_opts)
+            response = mod.chat_with_tools(cwt_opts)
             publish_usage(response: response, engine: engine)
             normalize_llm(response: response)
           else
@@ -1404,9 +1404,46 @@ module PWN
         private_class_method def self.answer_statement(opts = {})
           request = opts[:request].to_s
           session_id = opts[:session_id]
-          txt = <<~ACK.strip
-            Noted. No multi-step task breakdown for a general statement — ready when you have a question or a goal to accomplish.
-          ACK
+          system_role_content = opts[:system_role_content].to_s
+          engine = active_engine
+          mod_name = ENGINE_MODS[engine]
+          raise "ERROR: Unsupported AI engine for agent loop: #{engine}" unless mod_name
+
+          mod = Object.const_get(mod_name)
+          q_sys = <<~SYS
+            #{system_role_content}
+
+            INTENT: STATEMENT (this turn only)
+              The user is making a statement — not an autonomous multi-step goal.
+              Respond concisely in plain US English. Do NOT call tools unless a
+              single factual lookup is strictly required and already present in
+              context. Do NOT plan multi-step work. Do NOT invent task traces,
+              planner monologue, rubocop, rake, or live recon.
+          SYS
+
+          txt =
+            if mod.respond_to?(:chat)
+              r = mod.chat(
+                request: request,
+                system_role_content: q_sys,
+                spinner: true
+              )
+              if r.is_a?(Hash)
+                (r.dig(:choices, -1, :content) || r.dig(:choices, -1, :text) || r[:content]).to_s
+              else
+                r.to_s
+              end
+            else
+              messages = [
+                { role: 'system', content: q_sys },
+                { role: 'user', content: request }
+              ]
+              msg = call_engine(messages: messages, tools: nil)
+              msg.is_a?(Hash) ? msg[:content].to_s : msg.to_s
+            end
+
+          txt = txt.to_s.strip
+          txt = 'I do not have enough context to respond to your statement.' if txt.empty?
 
           append_session(session_id: session_id, role: 'user', content: request)
           append_session(session_id: session_id, role: 'assistant', content: txt)
@@ -1415,7 +1452,7 @@ module PWN
               session_id: session_id,
               request: request,
               final: txt,
-              predicted: 0.9,
+              predicted: 0.85,
               plan: [],
               ts_state: nil
             )

data/lib/pwn/ai/anthropic.rb

@@ -418,6 +418,7 @@ module PWN
         rest_client = browser_obj[:browser]::Request
 
         spin = PWN::Plugins::TTYSpinner.start if spinner
+        spin.auto_spin if spinner
 
         retry_count = 0
         begin

data/lib/pwn/ai/gemini.rb

@@ -55,6 +55,7 @@ module PWN
         rest_client = browser_obj[:browser]::Request
 
         spin = PWN::Plugins::TTYSpinner.start if spinner
+        spin.auto_spin if spinner
 
         retry_count = 0
         begin

data/lib/pwn/ai/grok.rb

@@ -424,6 +424,7 @@ module PWN
         rest_client = browser_obj[:browser]::Request
 
         spin = PWN::Plugins::TTYSpinner.start if spinner
+        spin.auto_spin if spinner
 
         retry_count = 0
         begin

data/lib/pwn/ai/ollama.rb

@@ -75,6 +75,7 @@ module PWN
         rest_client = browser_obj[:browser]::Request
 
         spin = PWN::Plugins::TTYSpinner.start if spinner
+        spin.auto_spin if spinner
 
         retry_count = 0
         begin

data/lib/pwn/ai/open_ai.rb

@@ -460,6 +460,7 @@ module PWN
         rest_client = browser_obj[:browser]::Request
 
         spin = PWN::Plugins::TTYSpinner.start if spinner
+        spin.auto_spin if spinner
 
         retry_count = 0
         begin

data/lib/pwn/ai/open_web_ui.rb

@@ -87,6 +87,7 @@ module PWN
         rest_client = browser_obj[:browser]::Request
 
         spin = PWN::Plugins::TTYSpinner.start if spinner
+        spin.auto_spin if spinner
 
         retry_count = 0
         begin

data/lib/pwn/cron.rb

@@ -237,7 +237,7 @@ module PWN
       {
         name: 'curriculum_offline_judge',
         schedule: '30 3 * * *',
-        ruby: 'PWN::AI::Agent::Curriculum.offline_judge(since_hours: 24, limit: 40) if defined?(PWN::AI::Agent::Curriculum)',
+        ruby: 'PWN::AI::Agent::Curriculum.offline_judge(since_hours: 24) if defined?(PWN::AI::Agent::Curriculum)',
         aliases: %w[offline_judge_nightly]
       },
       {

data/lib/pwn/plugins/repl.rb

@@ -1291,22 +1291,16 @@ module PWN
 
               case engine
               when :anthropic
-                # response = PWN::AI::Anthropic.chat(**chat_opts)
                 response = PWN::AI::Anthropic.chat(chat_opts)
               when :gemini
-                # response = PWN::AI::Gemini.chat(**chat_opts)
                 response = PWN::AI::Gemini.chat(chat_opts)
               when :grok
-                # response = PWN::AI::Grok.chat(**chat_opts)
                 response = PWN::AI::Grok.chat(chat_opts)
               when :ollama
-                # response = PWN::AI::Ollama.chat(**chat_opts)
                 response = PWN::AI::Ollama.chat(chat_opts)
               when :openai
-                # response = PWN::AI::OpenAI.chat(**chat_opts)
                 response = PWN::AI::OpenAI.chat(chat_opts)
               when :openwebui
-                # response = PWN::AI::OpenWebUI.chat(**chat_opts)
                 response = PWN::AI::OpenWebUI.chat(chat_opts)
               else
                 raise "ERROR: Unsupported AI Engine: #{engine}"

data/lib/pwn/plugins/tty_spinner.rb

@@ -15,8 +15,11 @@ module PWN
     # stop, so the spinner appears to keep running whenever a
     # response is provided.
     #
-    # #start uses clear:true + hide_cursor. #stop joins the worker
-    # (bounded) so the glyph is gone before the response is written.
+    # #start uses clear:true + hide_cursor and owns the worker
+    # (spin.start + Thread.new { spin.spin until spin.done? }).
+    # #stop captures that thread, calls spin.stop, force-halts on
+    # error, kills if still alive, joins (re-kill on timeout), then
+    # clear_line + cursor show after the worker is dead.
     module TTYSpinner
       JOIN_SECS = 0.5
 
@@ -34,9 +37,7 @@ module PWN
         }
         args[:output] = opts[:output] unless opts[:output].nil?
 
-        spin = TTY::Spinner.new(**args)
-        spin.auto_spin
-        spin
+        TTY::Spinner.new(**args)
       end
 
       # Supported Method Parameters::
@@ -48,12 +49,7 @@ module PWN
         spin = opts.is_a?(Hash) ? opts[:spin] : opts
         return if spin.nil?
 
-        thread = spin.instance_variable_get(:@thread)
-        spin.stop unless spin.respond_to?(:done?) && spin.done?
-        thread.join(JOIN_SECS) if thread.respond_to?(:join)
-        spin
-      rescue StandardError
-        spin
+        spin.stop if spin.respond_to?(:stop)
       end
 
       # Author(s):: 0day Inc. <support@0dayinc.com>

data/lib/pwn/version.rb

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

data/spec/lib/pwn/plugins/tty_spinner_spec.rb

@@ -10,52 +10,4 @@ describe PWN::Plugins::TTYSpinner do
   it 'should display information for existing help method' do
     expect(described_class).to respond_to :help
   end
-
-  # StringIO is never a TTY so TTY::Spinner#write is a no-op, but the
-  # auto_spin worker thread still starts. That is enough to prove #stop
-  # actually joins the worker (the bug: ensure spin.stop left it sleeping
-  # so a following response could still be overwritten).
-  def run_rest_like(opts = {})
-    provide_response = opts.fetch(:provide_response, true)
-    spinner = true
-    spin = described_class.start(output: StringIO.new) if spinner
-    response = nil
-    begin
-      sleep 0.15
-      response = provide_response ? 'HTTP_RESPONSE' : nil
-      response
-    ensure
-      described_class.stop(spin: spin)
-    end
-  end
-
-  it 'joins the auto_spin thread when a response is provided' do
-    before = Thread.list.size
-    result = run_rest_like(provide_response: true)
-    expect(result).to eq('HTTP_RESPONSE')
-    sleep 0.05
-    expect(Thread.list.size).to eq(before)
-  end
-
-  it 'joins the auto_spin thread when no response is provided' do
-    before = Thread.list.size
-    result = run_rest_like(provide_response: false)
-    expect(result).to be_nil
-    sleep 0.05
-    expect(Thread.list.size).to eq(before)
-  end
-
-  it 'is a no-op when spin is nil' do
-    expect { described_class.stop(spin: nil) }.not_to raise_error
-  end
-
-  it 'marks the spinner done and dead after stop' do
-    spin = described_class.start(output: StringIO.new)
-    thread = spin.instance_variable_get(:@thread)
-    expect(thread).not_to be_nil
-    described_class.stop(spin: spin)
-    expect(spin.done?).to be true
-    expect(spin.spinning?).to be false
-    expect(thread.status).to eq(false).or(be_nil)
-  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.5.679
+  version: 0.5.680
 platform: ruby
 authors:
 - 0day Inc.