From 4a425a8e758c51fd542ba4f80765ffce09b47b2a Mon Sep 17 00:00:00 2001
From: Akhilesh <ai.team@projecttech4dev.org>
Date: Thu, 23 Jul 2026 07:05:01 +0000
Subject: [PATCH] fix(google-genai): Handle transient 500s from Gemini TTS

---
 .../services/llm/providers/google_aistudio.py | 39 +++++++++++++++++--
 1 file changed, 35 insertions(+), 4 deletions(-)

diff --git a/backend/app/services/llm/providers/google_aistudio.py b/backend/app/services/llm/providers/google_aistudio.py
index 18c270d8a..35ea205aa 100644
--- a/backend/app/services/llm/providers/google_aistudio.py
+++ b/backend/app/services/llm/providers/google_aistudio.py
@@ -1,5 +1,6 @@
 import logging
 import base64
+import time
 from typing import Any
 from google import genai
 from google.genai import errors as genai_errors
@@ -331,10 +332,40 @@ def _execute_tts(
 
         config = GenerateContentConfig(**config_kwargs)
 
-        # Execute TTS
-        response: GenerateContentResponse = self.client.models.generate_content(
-            model=model, contents=resolved_input, config=config
-        )
+        # Execute TTS — retry on transient Google 5xx errors (preview model instability)
+        max_attempts = 3
+        retry_delays = [2, 4]  # seconds between attempts (exponential-ish)
+        response: GenerateContentResponse | None = None
+        for attempt in range(1, max_attempts + 1):
+            try:
+                response = self.client.models.generate_content(
+                    model=model, contents=resolved_input, config=config
+                )
+                break  # success — exit retry loop
+            except genai_errors.ServerError as e:
+                if attempt < max_attempts:
+                    delay = retry_delays[attempt - 1]
+                    logger.warning(
+                        f"[GoogleAIProvider._execute_tts] Transient server error from Gemini "
+                        f"(attempt {attempt}/{max_attempts}), retrying in {delay}s | "
+                        f"provider={provider}, model={model}, code={e.code}, error={e.message or str(e)}"
+                    )
+                    time.sleep(delay)
+                else:
+                    error_message = (
+                        f"[GEMINI] Server error after {max_attempts} attempts "
+                        f"(code: {e.code} {e.status or ''}): "
+                        f"{e.message or str(e)}. This is typically transient "
+                        f"(Gemini overloaded or internal error) — retry in a "
+                        f"few seconds. If the issue persists, contact Kaapi."
+                    )
+                    logger.error(
+                        f"[GoogleAIProvider._execute_tts] {error_message} | "
+                        f"provider={provider}, model={model}",
+                        exc_info=True,
+                    )
+                    return None, error_message
+
         if not response.response_id:
             error_message = (
                 "[GEMINI] TTS response is missing a response_id. This indicates "