From 062642dfd3916172ed1b69c96ac29cca831bb52f Mon Sep 17 00:00:00 2001
From: Ollie Agent <ollie@example.com>
Date: Tue, 18 Aug 2026 06:54:48 +0200
Subject: [PATCH] Bound parallel tool dispatch

---
 cmd/olliesrv/internal/agent/loop.go | 57 +++++++++++++++++------------
 1 file changed, 34 insertions(+), 23 deletions(-)

diff --git a/cmd/olliesrv/internal/agent/loop.go b/cmd/olliesrv/internal/agent/loop.go
index 09c57b2..b1ed250 100644
--- a/cmd/olliesrv/internal/agent/loop.go
+++ b/cmd/olliesrv/internal/agent/loop.go
@@ -15,7 +15,10 @@ import (
 	"ollie/util"
 )
 
-const maxTransientRetries = 3
+const (
+	maxTransientRetries  = 3
+	maxParallelToolCalls = 16
+)
 
 // retryBaseDelay is the base delay for rate-limit retries (5<<attempt seconds).
 // Tests can override this to speed up retry tests.
@@ -162,31 +165,31 @@ func (ag *Agent) run(ctx context.Context) error {
 			}
 		}
 
-			msg := backend.Message{ID: responseID, Role: "assistant", Content: content, Reasoning: reasoning, ToolCalls: toolCalls}
-			results, interrupted := ag.execToolCalls(ctx, toolCalls)
+		msg := backend.Message{ID: responseID, Role: "assistant", Content: content, Reasoning: reasoning, ToolCalls: toolCalls}
+		results, interrupted := ag.execToolCalls(ctx, toolCalls)
 
-			h.update(msg, results)
-			ag.save()
+		h.update(msg, results)
+		ag.save()
 
-			if interrupted {
-				return ctx.Err()
-			}
-			if len(toolCalls) == 0 {
-				break
-			}
-
-			if abort := ag.trackErrors(ctx, results, &errs, step); abort != nil {
-				return abort
-			}
-
-			_, compactErr := ag.autoCompact(ctx)
-			if compactErr != nil {
-				ag.emit(Event{Role: "error", Content: compactErr.Error()})
-				break
-			}
-
-			step++
+		if interrupted {
+			return ctx.Err()
 		}
+		if len(toolCalls) == 0 {
+			break
+		}
+
+		if abort := ag.trackErrors(ctx, results, &errs, step); abort != nil {
+			return abort
+		}
+
+		_, compactErr := ag.autoCompact(ctx)
+		if compactErr != nil {
+			ag.emit(Event{Role: "error", Content: compactErr.Error()})
+			break
+		}
+
+		step++
+	}
 
 	return nil
 }
@@ -467,6 +470,7 @@ func (ag *Agent) execBatch(ctx context.Context, batch []backend.ToolCall) ([]too
 	uniqueResults := make([]inflightResult, len(batch))
 
 	var wg sync.WaitGroup
+	sem := make(chan struct{}, maxParallelToolCalls)
 	for k, call := range batch {
 		key := call.Name + "\x00" + string(call.Arguments)
 		if _, dup := inflight[key]; dup {
@@ -476,6 +480,13 @@ func (ag *Agent) execBatch(ctx context.Context, batch []backend.ToolCall) ([]too
 		wg.Add(1)
 		go func(k int, call backend.ToolCall) {
 			defer wg.Done()
+			select {
+			case sem <- struct{}{}:
+			case <-ctx.Done():
+				uniqueResults[k].wasInt = true
+				return
+			}
+			defer func() { <-sem }()
 			uniqueResults[k].tr, uniqueResults[k].wasInt = ag.execOne(ctx, call)
 		}(k, call)
 	}