Fix retries

glebm · Nov 7, 2024 · 0fdd55f · 0fdd55f
1 parent ea8a72f
commit 0fdd55f
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/lib/i18n/tasks/translators/openai_translator.rb b/lib/i18n/tasks/translators/openai_translator.rb
@@ -95,11 +95,12 @@ def translate_batch(batch, from, to, retry_count = 0)
 
       if result.size != batch.size
         if retry_count < RETRIES
-          translate_batch(batch, from, to, retry_count)
+          translate_batch(batch, from, to, retry_count + 1)
         elsif retry_count == RETRIES
-          # Try each string individually
-          batch.each do |string|
-            translate_batch([string], from, to, RETRIES + 1)
+          # Try each string individually once
+          result = []
+          batch.map do |string|
+            result << translate_batch([string], from, to, RETRIES + 1)
           end
         else
           error = I18n.t('i18n_tasks.openai_translate.errors.invalid_size', expected: batch.size, actual: result.size)