2 files changed, 6 insertions, 0 deletions
diff --git a/internal/cli/run.go b/internal/cli/run.go
index 2da7b79..2d7c3d7 100644
--- a/internal/cli/run.go
+++ b/internal/cli/run.go
@@ -100,6 +100,9 @@ func runTasks(file string, parallel int, dryRun bool) error {
 		LLM:              localClient,
 		GeminiBinaryPath: cfg.GeminiBinaryPath,
 	}
+	if localClient != nil {
+		pool.LLM = localClient
+	}
 
 	// Handle graceful shutdown.
 	ctx, cancel := context.WithCancel(context.Background())
diff --git a/internal/cli/serve.go b/internal/cli/serve.go
index 2263d01..5101b81 100644
--- a/internal/cli/serve.go
+++ b/internal/cli/serve.go
@@ -88,6 +88,9 @@ func serve(addr string) error {
 		LLM:              localClient,
 		GeminiBinaryPath: cfg.GeminiBinaryPath,
 	}
+	if localClient != nil {
+		pool.LLM = localClient
+	}
 	pool.RecoverStaleRunning(context.Background())
 	pool.RecoverStaleQueued(context.Background())
 	pool.RecoverStaleBlocked()