pwn 0.5.755 → 0.5.756

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '088caeab6962a895018847c637332f41ca7033f69c7e8a547862d219c58b162c'
-  data.tar.gz: b0562e89e6cd08079728c7d5f0b2f640b975f482ed028c1cbe6b27a363361172
+  metadata.gz: 7fac5f5a9f5b01e9c4ce830db72a7e454f1a2dce37b44e57f98e290c154cfbbf
+  data.tar.gz: 1c508a671f0eb88f704bd38e58b7f3ad7154eca3e0e8d6b4d62852c7bf217a1b
 SHA512:
-  metadata.gz: 2730afce4cd70424a7fb927983bb417c1bb755846cd33113ee522f3b4d1a2e3a227251b7f3e5079b9bf3198224f8d36d5f31bed362acc37b6022fb0da03c417d
-  data.tar.gz: 9ac89ddaeb41583bc8b0db2a98faf9696ae78aae7f8ecc2689f4f8519071e1c284d172245aeb586de4a33c4ad7a81e4dc7ee16a7b849c23f4429e4ee9456e8ca
+  metadata.gz: 311ad0755488c66ad2c92c10751c52c5b86b4b725aae976fa697b0f206fb3c88afc954371d8ec8628fcc9431449fe58b71948f4ec2e3ad73b189e88165bcf3b4
+  data.tar.gz: e0055f29efafc74e550e84fd504debb71400acbfaf768eb1be82caa1d57cc1d79d49addfb78bb31608fdd0fd2716a4f68277b0d88272960e6b4dffb990b829f9

data/lib/pwn/ai/anthropic.rb

@@ -578,7 +578,7 @@ module PWN
               input_schema: anthropic_input_schema(schema: fn[:parameters] || fn['parameters'])
             }
           end
-          http_body[:tool_choice] = anth_tool_choice(choice: opts[:tool_choice]) if opts[:tool_choice]
+          http_body[:tool_choice] = anth_tool_choice(choice: opts[:tool_choice], model: model) if opts[:tool_choice]
         end
 
         response = anthropic_rest_call(
@@ -794,15 +794,28 @@ module PWN
 
       private_class_method def self.anth_tool_choice(opts = {})
         choice = opts[:choice]
-        case choice
-        when 'none', :none then { type: 'none' }
-        when 'required', :required, 'any', :any then { type: 'any' }
-        when Hash
-          fn = choice[:function] || choice['function'] || choice
-          { type: 'tool', name: fn[:name] || fn['name'] }
-        else # 'auto', :auto, nil, anything else
-          { type: 'auto' }
-        end
+        mapped = case choice
+                 when 'none', :none then { type: 'none' }
+                 when 'required', :required, 'any', :any then { type: 'any' }
+                 when Hash
+                   fn = choice[:function] || choice['function'] || choice
+                   { type: 'tool', name: fn[:name] || fn['name'] }
+                 else
+                   { type: 'auto' }
+                 end
+        return { type: 'auto' } if !forced_tool_choice?(model: opts[:model]) && %w[any tool].include?(mapped[:type].to_s)
+
+        mapped
+      end
+
+      # Claude Mythos 5.1, Fable 5.1, Opus 5.5, and Sonnet 5.5 reject forced tool use.
+      # Mythos 5 still accepts type any/tool, so match the documented ids, not every Mythos model.
+      private_class_method def self.forced_tool_choice?(opts = {})
+        model = opts[:model].to_s
+        return false if model.match?(/claude-(?:mythos|fable)-5-1(?:-|\z)/i)
+        return false if model.match?(/claude-(?:opus|sonnet)-5-5(?:-|\z)/i)
+
+        true
       end
 
       # Supported Method Parameters::

data/lib/pwn/version.rb

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

data/spec/lib/pwn/ai/anthropic_spec.rb

@@ -106,4 +106,31 @@ describe PWN::AI::Anthropic do
     expect(flat[:properties].keys.map(&:to_s)).to include('handle', 'path')
     expect(flat[:description]).to include('handle', 'path')
   end
+
+  it 'does not force tool use on Anthropic models that reject tool_choice any and tool' do
+    allow(PWN::Env).to receive(:[]).with(:ai).and_return(
+      anthropic: { key: 'test-key', model: 'claude-mythos-5', temp: 1, max_tokens: 100 }
+    )
+    allow(PWN::AI::Agent::PromptCache).to receive(:enabled?).and_return(false) if defined?(PWN::AI::Agent::PromptCache)
+    captured = nil
+    allow(described_class).to receive(:anthropic_rest_call) do |opts|
+      captured = opts[:http_body]
+      '{"id":"msg_choice","type":"message","role":"assistant","content":[{"type":"text","text":"ok"}],"stop_reason":"end_turn"}'
+    end
+    tools = [{ type: 'function', function: { name: 'memory_recall', description: 'recall', parameters: { type: 'object', properties: {} } } }]
+    messages = [{ role: 'user', content: 'hello' }]
+
+    described_class.chat_with_tools(messages: messages, tools: tools, model: 'claude-mythos-5', tool_choice: 'required')
+    expect(captured[:tool_choice]).to eq(type: 'any')
+
+    %w[claude-mythos-5-1 claude-fable-5-1 claude-opus-5-5 claude-sonnet-5-5].each do |model|
+      described_class.chat_with_tools(messages: messages, tools: tools, model: model, tool_choice: 'required')
+      expect(captured[:tool_choice]).to eq(type: 'auto'), model
+      described_class.chat_with_tools(messages: messages, tools: tools, model: model, tool_choice: { type: 'function', function: { name: 'memory_recall' } })
+      expect(captured[:tool_choice]).to eq(type: 'auto'), model
+    end
+
+    described_class.chat_with_tools(messages: messages, tools: tools, model: 'claude-mythos-5-1', tool_choice: 'none')
+    expect(captured[:tool_choice]).to eq(type: 'none')
+  end
 end

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

@@ -91,13 +91,22 @@ describe PWN::Plugins::BinaryParser do
   end
 
   it 'elf_resolve points system at the .plt.sec stub, not the next slot' do
+    x86_64 = RbConfig::CONFIG['host_cpu'].to_s.match?(/amd64|x86_64/)
     Dir.mktmpdir('pwn-elf-ibt-') do |dir|
       src = File.join(dir, 't.c')
       path = File.join(dir, 't')
       File.write(src, "#include <stdlib.h>\nint main(void) { return system(\"x\"); }\n")
-      out, status = Open3.capture2e('cc', '-O0', '-fno-pie', '-no-pie', '-Wl,-z,ibt', '-o', path, src)
+      flags = ['-O0', '-fno-pie', '-no-pie']
+      flags << '-Wl,-z,ibt' if x86_64
+      out, status = Open3.capture2e('cc', *flags, '-o', path, src)
       expect(status.success?).to eq(true), out
-      expect(File.binread(path)).to include('.plt.sec')
+      binary = File.binread(path)
+      unless x86_64
+        expect(binary).not_to include('.plt.sec')
+        next
+      end
+
+      expect(binary).to include('.plt.sec')
       row = described_class.elf_resolve(path: path)
       dump, dump_status = Open3.capture2('objdump', '-d', path)
       expect(dump_status.success?).to eq(true), dump

metadata

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