Add Chat#on_tool_call callback

This is useful when you want to show to the user tool calls as they
are made, particularly when handling streamed output.

Author: bryan-ash

lib/ruby_llm/chat.rb

@@ -29,7 +29,8 @@ def initialize(model: nil, provider: nil, assume_model_exists: false, context: n
       @schema = nil
       @on = {
         new_message: nil,
-        end_message: nil
+        end_message: nil,
+        tool_call: nil
       }
     end
 
@@ -112,6 +113,11 @@ def on_end_message(&block)
       self
     end
 
+    def on_tool_call(&block)
+      @on[:tool_call] = block
+      self
+    end
+
     def each(&)
       messages.each(&)
     end
@@ -181,6 +187,7 @@ def wrap_streaming_block(&block)
     def handle_tool_calls(response, &)
       response.tool_calls.each_value do |tool_call|
         @on[:new_message]&.call
+        @on[:tool_call]&.call(tool_call)
         result = execute_tool tool_call
         message = add_message role: :tool, content: result.to_s, tool_call_id: tool_call.id
         @on[:end_message]&.call(message)

spec/fixtures/vcr_cassettes/chat_tool_call_callbacks_calls_on_tool_call_callback_when_tools_are_used.yml

@@ -172,6 +172,25 @@ def execute
     end
   end
 
+  describe 'tool call callbacks' do
+    it 'calls on_tool_call callback when tools are used' do # rubocop:disable RSpec/ExampleLength,RSpec/MultipleExpectations
+      tool_calls_received = []
+
+      chat = RubyLLM.chat
+                    .with_tool(Weather)
+                    .on_tool_call { |tool_call| tool_calls_received << tool_call }
+
+      response = chat.ask("What's the weather in Berlin? (52.5200, 13.4050)")
+
+      expect(tool_calls_received).not_to be_empty
+      expect(tool_calls_received.first).to respond_to(:name)
+      expect(tool_calls_received.first).to respond_to(:arguments)
+      expect(tool_calls_received.first.name).to eq('weather')
+      expect(response.content).to include('15')
+      expect(response.content).to include('10')
+    end
+  end
+
   describe 'error handling' do
     it 'raises an error when tool execution fails' do # rubocop:disable RSpec/MultipleExpectations
       chat = RubyLLM.chat.with_tool(BrokenTool)