brute 2.0.0 → 2.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 19cba194a6650394c2b81805f70adcc5f774999962ca7d16123c7a335afc0532
-  data.tar.gz: c2a1d5bb0d1d2acceada34711f67adb586c3a366d68f22636dd754dce1587e19
+  metadata.gz: e70352bd81dcba123054a2e5fb1c5bf12642aec892cf41fa48297ecbb61fea65
+  data.tar.gz: 242e0ef1477ec8201c13ce5ec535bd45abbad9dca4252c88366477487a3c0913
 SHA512:
-  metadata.gz: b61457cd8041c561ddbd4764968cd0df6903b59e593f40af8bff9b63bb69624bf8f73f879130e336d64c25dce10f6f9d44d377837871db6ce9f19c0b20ff6cb3
-  data.tar.gz: 30989eae4bec648ba08036641cd5404a19c22ca477094896ccb6ca20d8a2ace194e82d1ffab4b5ef37436f38de350b90531c35ffb611974385888c3a0ea827ee
+  metadata.gz: 1999c090d40c1cb55d2e64152a8c633d364a461c84ac337f2b80c7d1151fd732c18075ddcc0709001949dbafa098d2707485fdf9e812e2f9eb485494607c8ce9
+  data.tar.gz: 6097b281be949bc6cf52934952ddac860da082045811a8a167883071b6d90946ddd96832594589a9ae399277d638e754a404ba8fb1813e013e55b7856522b5d9

data/lib/brute/agent.rb

@@ -41,7 +41,8 @@ module Brute
     end
 
     # Run one turn against the given session. The session is mutated
-    # in place (assistant + tool messages appended) and returned.
+    # in place (assistant + tool messages appended). Returns the env
+    # hash so callers can access metadata (timing, tokens, etc.).
     def call(session, events: NullSink.new)
       env = {
         messages:          session,
@@ -54,19 +55,20 @@ module Brute
         current_iteration: 1,
       }
       super(env)
-      session
+      env
     end
   end
 end
 
 test do
-  it "runs a turn and returns the session" do
+  it "runs a turn and returns the env with session in :messages" do
     agent = Brute::Agent.new(provider: :stub) do
       run ->(env) { env[:messages].assistant("hello") }
     end
     session = Brute::Session.new
     session.user("hi")
-    agent.call(session).should == session
+    env = agent.call(session)
+    env[:messages].should == session
   end
 
   it "passes provider/model/tools through env" do

data/lib/brute/middleware/005_tracing.rb

@@ -67,6 +67,14 @@ module Brute
           last_call_elapsed: elapsed
         }
 
+        if response.respond_to?(:usage) && (u = response.usage)
+          env[:metadata][:tokens] = {
+            total:        read_token(u, :total_tokens),
+            total_input:  read_token(u, :input_tokens),
+            total_output: read_token(u, :output_tokens),
+          }
+        end
+
         response
       end
 

data/lib/brute/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: brute
 version: !ruby/object:Gem::Version
-  version: 2.0.0
+  version: 2.0.2
 platform: ruby
 authors:
 - Brute Contributors
@@ -187,7 +187,6 @@ files:
 - lib/brute/system_prompt.rb
 - lib/brute/tool.rb
 - lib/brute/tools.rb
-- lib/brute/tools/delegate.rb
 - lib/brute/tools/fs_patch.rb
 - lib/brute/tools/fs_read.rb
 - lib/brute/tools/fs_remove.rb

data/lib/brute/tools/delegate.rb

@@ -1,109 +0,0 @@
-# frozen_string_literal: true
-
-require "bundler/setup"
-require "brute"
-
-module Brute
-  module Tools
-    class Delegate < RubyLLM::Tool
-      description "Delegate a research or analysis task to a specialist sub-agent. " \
-                  "The sub-agent can read files and search but cannot write or execute commands. " \
-                  "Use for code analysis, understanding patterns, or gathering information."
-
-      param :task, type: 'string', desc: "A clear, detailed description of the research task", required: true
-
-      def name; "delegate"; end
-
-      MAX_ROUNDS = 10
-
-      def execute(task:)
-        provider = Brute.provider
-        llm = provider.ruby_llm_provider
-        model_id = provider.default_model
-        model = Brute::Middleware::ModelRef.new(model_id, 16_384)
-
-        sub_tools = { read: FSRead.new, fs_search: FSSearch.new }
-
-        messages = [
-          RubyLLM::Message.new(
-            role: :system,
-            content: "You are a research agent. Analyze code, explain patterns, and answer questions. " \
-                     "You have read-only access to the filesystem. Be thorough and precise."
-          ),
-          RubyLLM::Message.new(role: :user, content: task),
-        ]
-
-        response = nil
-        MAX_ROUNDS.times do
-          response = llm.complete(messages, tools: sub_tools, temperature: nil, model: model)
-          messages << response
-
-          break unless response.tool_call?
-
-          response.tool_calls.each_value do |tc|
-            tool = sub_tools[tc.name.to_sym]
-            result = if tool
-                       tool.call(tc.arguments)
-                     else
-                       { error: "Unknown tool: #{tc.name}" }
-                     end
-            content = result.is_a?(String) ? result : result.to_s
-            messages << RubyLLM::Message.new(role: :tool, content: content, tool_call_id: tc.id)
-          end
-        end
-
-        { result: extract_content(response, messages) }
-      end
-
-      private
-
-      # Safely extract text content from the sub-agent response.
-      def extract_content(response, messages)
-        text = response&.content
-        return text if text.is_a?(::String) && !text.empty?
-
-        # Fall back to last assistant text in the conversation history
-        last_assistant = messages
-          .select { |m| m.role == :assistant }
-          .reverse
-          .find { |m| m.content.is_a?(::String) && !m.content.empty? }
-        last_assistant&.content || "(sub-agent completed but produced no text response)"
-      end
-    end
-  end
-end
-
-test do
-  require_relative "../../../spec/support/mock_provider"
-  require_relative "../../../spec/support/mock_response"
-
-  delegate = Brute::Tools::Delegate.new
-
-  it "returns content when response has text" do
-    res = RubyLLM::Message.new(role: :assistant, content: "analysis complete")
-    delegate.send(:extract_content, res, []).should == "analysis complete"
-  end
-
-  it "falls back to last assistant text on nil content" do
-    res = RubyLLM::Message.new(role: :assistant, content: "")
-    msgs = [
-      RubyLLM::Message.new(role: :user, content: "input"),
-      RubyLLM::Message.new(role: :assistant, content: "found the answer"),
-    ]
-    delegate.send(:extract_content, res, msgs).should == "found the answer"
-  end
-
-  it "returns fallback when no assistant messages exist" do
-    res = RubyLLM::Message.new(role: :assistant, content: "")
-    delegate.send(:extract_content, res, []).should == "(sub-agent completed but produced no text response)"
-  end
-
-  it "skips assistant messages with empty content" do
-    res = RubyLLM::Message.new(role: :assistant, content: "")
-    msgs = [
-      RubyLLM::Message.new(role: :assistant, content: "real answer"),
-      RubyLLM::Message.new(role: :assistant, content: ""),
-    ]
-    delegate.send(:extract_content, res, msgs).should == "real answer"
-  end
-end