From b9be689e399b36984a42aba7f022e14cdd6a7cd8 Mon Sep 17 00:00:00 2001
From: Nithwin <vmnithwin@gmail.com>
Date: Sun, 26 Jul 2026 13:08:21 +0530
Subject: [PATCH] fix(agent): implement exponential backoff retry in loop

---
 internal/agent/loop.go | 31 ++++++++++++++++++++++++++++---
 1 file changed, 28 insertions(+), 3 deletions(-)

diff --git a/internal/agent/loop.go b/internal/agent/loop.go
index f17754c..131444f 100644
--- a/internal/agent/loop.go
+++ b/internal/agent/loop.go
@@ -3,6 +3,7 @@ package agent
 import (
 	"context"
 	"os"
+	"time"
 
 	"github.com/Nithwin/WindMist/internal/agent/prompt"
 	"github.com/Nithwin/WindMist/internal/ai"
@@ -35,9 +36,33 @@ func (a *Agent) runLoop(ctx context.Context, messages []ai.Message, userPrompt s
 		var resp *ai.GenerateResponse
 		var err error
 
-		// Use stream only for the first turn (to show the user something is happening)
-		// Or always stream. Since we patched providers to return GenerateResponse, we can always Stream!
-		resp, err = a.provider.Stream(ctx, req, onChunk)
+		maxRetries := 3
+		backoff := 1 * time.Second
+
+		for attempt := 0; attempt <= maxRetries; attempt++ {
+			resp, err = a.provider.Stream(ctx, req, onChunk)
+			if err == nil {
+				break
+			}
+
+			// Don't retry if context is cancelled by user
+			if ctx.Err() != nil {
+				break
+			}
+
+			if attempt == maxRetries {
+				break
+			}
+
+			// Wait before retrying
+			select {
+			case <-ctx.Done():
+				break
+			case <-time.After(backoff):
+			}
+			backoff *= 2
+		}
+
 		if err != nil {
 			return nil, err
 		}