brute 6.1.4 → 6.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f639a3b9b3ec1317f80ee945eb7d610af350d4c862206745aefad42a70afbb4d
-  data.tar.gz: 994bf0aaf3104bfad636338639db4c175de7bbcd468a9abd8f90114c4a8e233e
+  metadata.gz: fe1e1cc4c5d064aba3ae09ac3cce887a9b116b7ebd52ccefedb6a1e8287050c2
+  data.tar.gz: af8a28e3643305f90f59eef3878e8ac1618ccad54b15cdafed63f33773d4d717
 SHA512:
-  metadata.gz: 29b29a00aad896433765e8bbff99ec42422af1c133b4b73231837f85ed7d0ffd4ea2101859fc37ca81c16678860f9a6de72d437d674faf3b9e12d3734f6c36e7
-  data.tar.gz: b43248273fc878a7c553e968b010552a7a670933b694a5a952bb3ecb5c197e08595e5134262475ee296c41b538e601347d34bfc7694cb6880b766e78d8231ea8
+  metadata.gz: 29b0f666931c9891647bf88a1d8ebf6e750c75d94782e53b73d404fd36401a70273dd0fcf9d156e34b6ac5fda952b49f851e1b3f8e29f291d544c1614680a9d6
+  data.tar.gz: 5babc3e4185d9d68ca6f86613cbe03116adb5af0483d16f04f6a5b517d4233b46f6a458df3b67b985d038852619865dc36ecf695e9b3bf379503cd7ddb6585a9

data/lib/brute/message_transport/open_router.rb

@@ -37,6 +37,10 @@ module Brute
           # wire's: whatever goes out under these keys is put there here.
           hash.delete(:reasoning)
 
+          if message.tool_call?
+            hash[:tool_calls] = message.tool_calls.map { |call| call_block(call) }
+          end
+
           reasoning = message.reasoning
 
           if reasoning&.detailed?
@@ -47,6 +51,16 @@ module Brute
         end
       end
 
+      # The inverse of wrap_tool_call: the wire keeps the name and the
+      # arguments under `function`, and the arguments as a JSON string.
+      def self.call_block(call)
+        {
+          id:       call.id,
+          type:     "function",
+          function: { name: call.name, arguments: JSON.generate(call.arguments) },
+        }
+      end
+
       # Each type names its own payload: reasoning text carries `text` and the
       # `signature` that verifies it, a summary carries `summary`, and an
       # encrypted item carries `data`. Written under the wrong key the payload
@@ -170,6 +184,30 @@ describe "brute/message_transport/open_router" do
     Brute::MessageTransport::OpenRouter.dump(m).should == { role: :user, content: "hi" }
   end
 
+  it "dumps a tool call as the wire wants it, and reads its own output back" do
+    call = Brute::ToolCall.new(id: "call_1", name: "search", arguments: { "query" => "cheese" })
+    dumped = Brute::MessageTransport::OpenRouter.dump(
+      Brute::Message.new(role: :assistant, content: "", tool_calls: [call]),
+    )
+
+    dumped[:tool_calls].should == [
+      { id: "call_1", type: "function", function: { name: "search", arguments: '{"query":"cheese"}' } },
+    ]
+
+    echoed = Struct.new(:choices).new([{ "message" => {
+      "role" => "assistant", "content" => "", "tool_calls" => dumped[:tool_calls],
+    } }])
+    back = Brute::MessageTransport::OpenRouter.new(echoed).wrap_each.to_a.first.tool_calls.first
+
+    back.id.should == "call_1"
+    back.name.should == "search"
+    back.arguments.should == { "query" => "cheese" }
+
+    Brute::MessageTransport::OpenRouter.dump(
+      Brute::Message.new(role: :user, content: "hi"),
+    ).key?(:tool_calls).should.be.false
+  end
+
   it "wraps a response's choice messages with symbolised roles, dropping provider extras" do
     fake_response = Struct.new(:choices).new([
       { "message" => { "role" => "assistant", "content" => "hi there",

data/lib/brute/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: brute
 version: !ruby/object:Gem::Version
-  version: 6.1.4
+  version: 6.1.5
 platform: ruby
 authors:
 - Brute Contributors