From 27b91d4ad606876a06f25ea82c989374d54311fd Mon Sep 17 00:00:00 2001
From: Motov Yurii <109919500+YuriiMotov@users.noreply.github.com>
Date: Wed, 4 Feb 2026 15:05:19 +0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A8=20Update=20translation=20script=20?=
 =?UTF-8?q?to=20retry=20if=20LLM-response=20doesn't=20pass=20validation=20?=
 =?UTF-8?q?with=20Translation=20Fixer=20tool=20(#14749)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
---
 scripts/translate.py | 28 +++++++++++++++++++++++++---
 1 file changed, 25 insertions(+), 3 deletions(-)

diff --git a/scripts/translate.py b/scripts/translate.py
index eba4ad6a2f..9eda7b3903 100644
--- a/scripts/translate.py
+++ b/scripts/translate.py
@@ -10,6 +10,7 @@ from typing import Annotated
 import git
 import typer
 import yaml
+from doc_parsing_utils import check_translation
 from github import Github
 from pydantic_ai import Agent
 from rich import print
@@ -119,9 +120,30 @@ def translate_page(
         ]
     )
     prompt = "\n\n".join(prompt_segments)
-    print(f"Running agent for {out_path}")
-    result = agent.run_sync(prompt)
-    out_content = f"{result.output.strip()}\n"
+
+    MAX_ATTEMPTS = 3
+    for attempt_no in range(1, MAX_ATTEMPTS + 1):
+        print(f"Running agent for {out_path} (attempt {attempt_no}/{MAX_ATTEMPTS})")
+        result = agent.run_sync(prompt)
+        out_content = f"{result.output.strip()}\n"
+        try:
+            check_translation(
+                doc_lines=out_content.splitlines(),
+                en_doc_lines=original_content.splitlines(),
+                lang_code=language,
+                auto_fix=False,
+                path=str(out_path),
+            )
+            break  # Exit loop if no errors
+        except ValueError as e:
+            print(
+                f"Translation check failed on attempt {attempt_no}/{MAX_ATTEMPTS}: {e}"
+            )
+            continue  # Retry if not reached max attempts
+    else:  # Max retry attempts reached
+        print(f"Translation failed for {out_path} after {MAX_ATTEMPTS} attempts")
+        raise typer.Exit(code=1)
+
     print(f"Saving translation to {out_path}")
     out_path.write_text(out_content, encoding="utf-8", newline="\n")