brute 0.3.0 → 0.4.1

data/lib/brute/providers/opencode_go.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 module LLM
   ##
   # OpenAI-compatible provider for the OpenCode Go API gateway.

data/lib/brute/providers/opencode_zen.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 # Ensure the OpenAI provider is loaded (llm.rb lazy-loads providers).
 unless defined?(LLM::OpenAI)
   require "llm/providers/openai"
@@ -41,10 +46,10 @@ module LLM
     end
 
     ##
-    # Returns the default model (Claude Sonnet 4, the most common Zen model).
+    # Returns the default model.
     # @return [String]
     def default_model
-      "claude-sonnet-4-20250514"
+      "zen-bickpickle"
     end
 
     ##

data/lib/brute/providers/shell_response.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 require "securerandom"
 
 module Brute

data/lib/brute/system_prompt.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 module Brute
   # Deferred system prompt builder.
   #
@@ -164,3 +169,212 @@ module Brute
     end
   end
 end
+
+if __FILE__ == $0
+  require_relative "../../spec/spec_helper"
+
+  RSpec.describe Brute::SystemPrompt do
+    describe ".build" do
+      it "stores a block and executes it on prepare" do
+        sp = described_class.build do |prompt, ctx|
+          prompt << "hello #{ctx[:name]}"
+        end
+
+        result = sp.prepare(name: "world")
+        expect(result.to_s).to eq("hello world")
+      end
+
+      it "returns a Result with sections" do
+        sp = described_class.build do |prompt, _ctx|
+          prompt << "section one"
+          prompt << "section two"
+        end
+
+        result = sp.prepare({})
+        expect(result.sections).to eq(["section one", "section two"])
+      end
+
+      it "strips nil and empty sections" do
+        sp = described_class.build do |prompt, _ctx|
+          prompt << "kept"
+          prompt << nil
+          prompt << ""
+          prompt << "also kept"
+        end
+
+        result = sp.prepare({})
+        expect(result.sections).to eq(["kept", "also kept"])
+      end
+    end
+
+    describe Brute::SystemPrompt::Result do
+      subject(:result) { described_class.new(["a", "b", "c"]) }
+
+      it "joins sections with double newlines via to_s" do
+        expect(result.to_s).to eq("a\n\nb\n\nc")
+      end
+
+      it "iterates over sections via each" do
+        collected = []
+        result.each { |s| collected << s }
+        expect(collected).to eq(["a", "b", "c"])
+      end
+
+      it "reports empty? correctly" do
+        expect(described_class.new([]).empty?).to be true
+        expect(result.empty?).to be false
+      end
+    end
+
+    describe ".default" do
+      # Minimal context for all tests — enough to avoid nil errors
+      let(:base_ctx) do
+        {
+          provider_name: "anthropic",
+          model_name: "test-model",
+          cwd: Dir.pwd,
+          custom_rules: nil,
+          agent: nil,
+          agent_switched: nil,
+          max_steps_reached: nil,
+        }
+      end
+
+      let(:builder) { described_class.default }
+
+      # ── Provider stack selection ──
+
+      described_class::STACKS.each_key do |provider|
+        context "with provider '#{provider}'" do
+          it "produces non-empty sections" do
+            ctx = base_ctx.merge(provider_name: provider)
+            result = builder.prepare(ctx)
+            expect(result.sections).not_to be_empty
+            result.sections.each do |section|
+              expect(section).to be_a(String)
+            end
+          end
+        end
+      end
+
+      it "falls back to 'default' stack for unknown providers" do
+        default_result = builder.prepare(base_ctx.merge(provider_name: "default"))
+        unknown_result = builder.prepare(base_ctx.merge(provider_name: "unknown_provider"))
+
+        # Both should produce the same number of sections (same stack)
+        expect(unknown_result.sections.size).to eq(default_result.sections.size)
+      end
+
+      # ── Plan mode conditional ──
+
+      context "when agent is 'plan'" do
+        it "includes PlanReminder" do
+          result = builder.prepare(base_ctx.merge(agent: "plan"))
+          expect(result.to_s).to include("Plan Mode")
+          expect(result.to_s).to include("<system-reminder>")
+          expect(result.to_s).to include("READ-ONLY")
+        end
+      end
+
+      context "when agent is 'build'" do
+        it "does NOT include PlanReminder" do
+          result = builder.prepare(base_ctx.merge(agent: "build"))
+          expect(result.to_s).not_to include("Plan Mode - System Reminder")
+          expect(result.to_s).not_to include("READ-ONLY")
+        end
+      end
+
+      context "when agent is nil" do
+        it "does NOT include PlanReminder" do
+          result = builder.prepare(base_ctx.merge(agent: nil))
+          expect(result.to_s).not_to include("Plan Mode - System Reminder")
+        end
+      end
+
+      # ── Build switch conditional ──
+
+      context "when agent_switched is 'build'" do
+        it "includes BuildSwitch" do
+          result = builder.prepare(base_ctx.merge(agent_switched: "build"))
+          expect(result.to_s).to include("operational mode has changed from plan to build")
+        end
+      end
+
+      context "when agent_switched is nil" do
+        it "does NOT include BuildSwitch" do
+          result = builder.prepare(base_ctx.merge(agent_switched: nil))
+          expect(result.to_s).not_to include("operational mode has changed")
+        end
+      end
+
+      # ── Max steps conditional ──
+
+      context "when max_steps_reached is truthy" do
+        it "includes MaxSteps" do
+          result = builder.prepare(base_ctx.merge(max_steps_reached: true))
+          expect(result.to_s).to include("MAXIMUM STEPS REACHED")
+          expect(result.to_s).to include("Tools are disabled")
+        end
+      end
+
+      context "when max_steps_reached is falsy" do
+        it "does NOT include MaxSteps" do
+          result = builder.prepare(base_ctx.merge(max_steps_reached: nil))
+          expect(result.to_s).not_to include("MAXIMUM STEPS REACHED")
+        end
+      end
+
+      # ── Combined states (mid-session switch scenarios) ──
+
+      context "plan mode includes PlanReminder but excludes BuildSwitch" do
+        it "contains only plan-specific content" do
+          result = builder.prepare(base_ctx.merge(agent: "plan"))
+          expect(result.to_s).to include("READ-ONLY")
+          expect(result.to_s).not_to include("operational mode has changed")
+        end
+      end
+
+      context "build mode with agent_switched includes BuildSwitch but excludes PlanReminder" do
+        it "contains only build-switch content" do
+          result = builder.prepare(base_ctx.merge(agent: "build", agent_switched: "build"))
+          expect(result.to_s).to include("operational mode has changed from plan to build")
+          expect(result.to_s).not_to include("Plan Mode - System Reminder")
+        end
+      end
+
+      context "plan mode with max_steps_reached includes both" do
+        it "contains both PlanReminder and MaxSteps" do
+          result = builder.prepare(base_ctx.merge(agent: "plan", max_steps_reached: true))
+          expect(result.to_s).to include("READ-ONLY")
+          expect(result.to_s).to include("MAXIMUM STEPS REACHED")
+        end
+      end
+
+      # ── Provider stack composition ──
+
+      it "anthropic stack includes expected modules" do
+        result = builder.prepare(base_ctx.merge(provider_name: "anthropic"))
+        text = result.to_s
+        # Anthropic has Identity, Objectivity, Conventions, GitSafety etc.
+        expect(text).to include("Professional objectivity")
+        expect(text).to include("Following conventions")
+        expect(text).to include("Git safety")
+      end
+
+      it "openai stack includes editing-focused modules" do
+        result = builder.prepare(base_ctx.merge(provider_name: "openai"))
+        text = result.to_s
+        # OpenAI has EditingApproach, Autonomy, EditingConstraints
+        expect(text).to include("Following conventions")
+        expect(text).to include("Git safety")
+      end
+
+      it "google stack includes security module" do
+        result = builder.prepare(base_ctx.merge(provider_name: "google"))
+        text = result.to_s
+        expect(text).to include("Following conventions")
+        expect(text).to include("Git safety")
+      end
+    end
+  end
+end

data/lib/brute/tools/delegate.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 module Brute
   module Tools
     class Delegate < LLM::Tool
@@ -57,3 +62,127 @@ module Brute
     end
   end
 end
+
+if __FILE__ == $0
+  require_relative "../../../spec/spec_helper"
+
+  RSpec.describe Brute::Tools::Delegate do
+    let(:provider) { MockProvider.new }
+
+    # Simple stand-in for LLM::Message in context history.
+    FakeMessage = Struct.new(:role, :content) do
+      def assistant?
+        role == :assistant
+      end
+    end
+
+    before do
+      allow(Brute).to receive(:provider).and_return(provider)
+    end
+
+    def fake_context(messages)
+      msgs_obj = Object.new
+      msgs_obj.define_singleton_method(:to_a) { messages }
+      ctx = Object.new
+      ctx.define_singleton_method(:messages) { msgs_obj }
+      ctx
+    end
+
+    describe "#call" do
+      it "returns the sub-agent's text content" do
+        result = described_class.new.call(task: "What files exist?")
+        expect(result).to be_a(Hash)
+        expect(result[:result]).to eq("mock response")
+      end
+    end
+
+    describe "#extract_content" do
+      let(:delegate) { described_class.new }
+
+      it "returns content when response has text" do
+        res = MockResponse.new(content: "analysis complete")
+        context = fake_context([])
+        result = delegate.send(:extract_content, res, context)
+        expect(result).to eq("analysis complete")
+      end
+
+      context "when res.content raises NoMethodError (tool-only response)" do
+        let(:bad_res) do
+          obj = Object.new
+          obj.define_singleton_method(:content) do
+            raise NoMethodError, "undefined method 'content' for nil"
+          end
+          obj
+        end
+
+        it "falls back to the last assistant text in context messages" do
+          context = fake_context([
+            FakeMessage.new(:user, "input"),
+            FakeMessage.new(:assistant, "found the answer"),
+          ])
+
+          result = delegate.send(:extract_content, bad_res, context)
+          expect(result).to eq("found the answer")
+        end
+
+        it "returns fallback text when no assistant messages exist" do
+          context = fake_context([])
+          result = delegate.send(:extract_content, bad_res, context)
+          expect(result).to eq("(sub-agent completed but produced no text response)")
+        end
+
+        it "skips assistant messages with empty content" do
+          context = fake_context([
+            FakeMessage.new(:assistant, "real answer"),
+            FakeMessage.new(:assistant, ""),
+          ])
+
+          result = delegate.send(:extract_content, bad_res, context)
+          expect(result).to eq("real answer")
+        end
+
+        it "skips assistant messages with non-string content" do
+          context = fake_context([
+            FakeMessage.new(:assistant, "text answer"),
+            FakeMessage.new(:assistant, [{"type" => "tool_use"}]),
+          ])
+
+          result = delegate.send(:extract_content, bad_res, context)
+          expect(result).to eq("text answer")
+        end
+      end
+
+      context "when res.content returns nil (empty response)" do
+        let(:nil_res) { Struct.new(:content).new(nil) }
+
+        it "falls back to the last assistant text in context messages" do
+          context = fake_context([
+            FakeMessage.new(:assistant, "previous answer"),
+          ])
+
+          result = delegate.send(:extract_content, nil_res, context)
+          expect(result).to eq("previous answer")
+        end
+
+        it "returns fallback text when no assistant messages exist" do
+          context = fake_context([])
+          result = delegate.send(:extract_content, nil_res, context)
+          expect(result).to eq("(sub-agent completed but produced no text response)")
+        end
+      end
+
+      context "when res.content returns empty string" do
+        let(:empty_res) { Struct.new(:content).new("") }
+
+        it "falls back to the last assistant text in context messages" do
+          context = fake_context([
+            FakeMessage.new(:assistant, "previous answer"),
+          ])
+
+          result = delegate.send(:extract_content, empty_res, context)
+          expect(result).to eq("previous answer")
+        end
+      end
+    end
+  end
+end

data/lib/brute/tools/fs_patch.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 module Brute
   module Tools
     class FSPatch < LLM::Tool
@@ -37,3 +42,51 @@ module Brute
     end
   end
 end
+
+if __FILE__ == $0
+  require_relative "../../../spec/spec_helper"
+
+  RSpec.describe Brute::Tools::FSPatch do
+    around(:each) { |ex| Dir.mktmpdir { |d| @dir = d; ex.run } }
+
+    let(:tool) { described_class.new }
+
+    it "replaces old_string with new_string" do
+      path = File.join(@dir, "test.rb")
+      File.write(path, "hello world\n")
+      result = tool.call(file_path: path, old_string: "world", new_string: "ruby")
+      expect(result[:success]).to be true
+      expect(File.read(path)).to eq("hello ruby\n")
+    end
+
+    it "returns a unified diff" do
+      path = File.join(@dir, "test.rb")
+      File.write(path, "line1\nold line\nline3\n")
+      result = tool.call(file_path: path, old_string: "old line", new_string: "new line")
+      expect(result[:diff]).to include("-old line")
+      expect(result[:diff]).to include("+new line")
+    end
+
+    it "raises when file not found" do
+      expect {
+        tool.call(file_path: File.join(@dir, "nope.rb"), old_string: "a", new_string: "b")
+      }.to raise_error(/File not found/)
+    end
+
+    it "raises when old_string not found" do
+      path = File.join(@dir, "test.rb")
+      File.write(path, "hello\n")
+      expect {
+        tool.call(file_path: path, old_string: "missing", new_string: "new")
+      }.to raise_error(/old_string not found/)
+    end
+
+    it "supports replace_all" do
+      path = File.join(@dir, "test.rb")
+      File.write(path, "aaa bbb aaa\n")
+      result = tool.call(file_path: path, old_string: "aaa", new_string: "ccc", replace_all: true)
+      expect(result[:replacements]).to eq(2)
+      expect(File.read(path)).to eq("ccc bbb ccc\n")
+    end
+  end
+end

data/lib/brute/tools/fs_read.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 module Brute
   module Tools
     class FSRead < LLM::Tool

data/lib/brute/tools/fs_remove.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 require "fileutils"
 
 module Brute

data/lib/brute/tools/fs_search.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 require "open3"
 
 module Brute

data/lib/brute/tools/fs_undo.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 module Brute
   module Tools
     class FSUndo < LLM::Tool

data/lib/brute/tools/fs_write.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 require 'fileutils'
 
 module Brute
@@ -26,3 +31,48 @@ module Brute
     end
   end
 end
+
+if __FILE__ == $0
+  require_relative "../../../spec/spec_helper"
+
+  RSpec.describe Brute::Tools::FSWrite do
+    around(:each) { |ex| Dir.mktmpdir { |d| @dir = d; ex.run } }
+
+    let(:tool) { described_class.new }
+
+    it "writes content to a new file" do
+      path = File.join(@dir, "new.rb")
+      result = tool.call(file_path: path, content: "hello\n")
+      expect(result[:success]).to be true
+      expect(File.read(path)).to eq("hello\n")
+    end
+
+    it "returns a diff for new files" do
+      path = File.join(@dir, "new.rb")
+      result = tool.call(file_path: path, content: "line1\nline2\n")
+      expect(result[:diff]).to include("+line1")
+      expect(result[:diff]).to include("+line2")
+    end
+
+    it "returns a diff for overwritten files" do
+      path = File.join(@dir, "existing.rb")
+      File.write(path, "old content\n")
+      result = tool.call(file_path: path, content: "new content\n")
+      expect(result[:diff]).to include("-old content")
+      expect(result[:diff]).to include("+new content")
+    end
+
+    it "creates parent directories" do
+      path = File.join(@dir, "deep", "nested", "file.rb")
+      result = tool.call(file_path: path, content: "nested\n")
+      expect(result[:success]).to be true
+      expect(File.exist?(path)).to be true
+    end
+
+    it "returns byte count" do
+      path = File.join(@dir, "test.rb")
+      result = tool.call(file_path: path, content: "hello")
+      expect(result[:bytes]).to eq(5)
+    end
+  end
+end

data/lib/brute/tools/net_fetch.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 require "net/http"
 require "uri"
 

data/lib/brute/tools/question.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 module Brute
   module Tools
     class Question < LLM::Tool

data/lib/brute/tools/shell.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 require "open3"
 
 module Brute

data/lib/brute/tools/todo_read.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 module Brute
   module Tools
     class TodoRead < LLM::Tool

data/lib/brute/tools/todo_write.rb

@@ -1,5 +1,10 @@
 # frozen_string_literal: true
 
+if __FILE__ == $0
+  require "bundler/setup"
+  require "brute"
+end
+
 module Brute
   module Tools
     class TodoWrite < LLM::Tool

data/lib/brute/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Brute
-  VERSION = "0.3.0"
+  VERSION = "0.4.1"
 end

data/lib/brute.rb

@@ -194,16 +194,19 @@ module Brute
   #
   # Checks in order:
   #   1. LLM_API_KEY + LLM_PROVIDER (explicit)
-  #   2. ANTHROPIC_API_KEY (implicit: provider = anthropic)
-  #   3. OPENAI_API_KEY   (implicit: provider = openai)
-  #   4. GOOGLE_API_KEY   (implicit: provider = google)
-  #   5. OPENCODE_API_KEY  (implicit: provider = opencode_zen)
+  #   2. OPENCODE_API_KEY (implicit: provider = opencode_zen)
+  #   3. ANTHROPIC_API_KEY (implicit: provider = anthropic)
+  #   4. OPENAI_API_KEY   (implicit: provider = openai)
+  #   5. GOOGLE_API_KEY   (implicit: provider = google)
   #
   # Returns nil if no key is found. Error is deferred to Orchestrator#run.
   def self.resolve_provider
     if ENV['LLM_API_KEY']
       key = ENV['LLM_API_KEY']
-      name = ENV.fetch('LLM_PROVIDER', 'anthropic').downcase
+      name = ENV.fetch('LLM_PROVIDER', 'opencode_zen').downcase
+    elsif ENV['OPENCODE_API_KEY']
+      key = ENV['OPENCODE_API_KEY']
+      name = 'opencode_zen'
     elsif ENV['ANTHROPIC_API_KEY']
       key = ENV['ANTHROPIC_API_KEY']
       name = 'anthropic'
@@ -213,9 +216,6 @@ module Brute
     elsif ENV['GOOGLE_API_KEY']
       key = ENV['GOOGLE_API_KEY']
       name = 'google'
-    elsif ENV['OPENCODE_API_KEY']
-      key = ENV['OPENCODE_API_KEY']
-      name = 'opencode_zen'
     else
       return nil
     end