fix: upd

Author: maxprilutskiy

src/lingodotdev/engine.py

@@ -350,17 +350,11 @@ def wrapped_progress_callback(progress: int, source_chunk: Dict[str, str], proce
             wrapped_progress_callback
         )
 
-        result = []
-        for key, value in localized.items():
-            if key.startswith('chat_'):
-                index = int(key.split('_')[1])
-                if index < len(chat):
-                    result.append({
-                        'name': chat[index]['name'],
-                        'text': value
-                    })
+        # The API returns the localized chat in the same structure
+        if 'chat' in localized and isinstance(localized['chat'], list):
+            return localized['chat']
 
-        return result
+        return []
 
     def recognize_locale(self, text: str) -> str:
         """

tests/test_engine.py

@@ -229,8 +229,10 @@ def test_batch_localize_text_missing_target_locales(self):
     def test_localize_chat(self, mock_localize_raw):
         """Test chat localization"""
         mock_localize_raw.return_value = {
-            "chat_0": "hola",
-            "chat_1": "adiós"
+            "chat": [
+                {"name": "Alice", "text": "hola"},
+                {"name": "Bob", "text": "adiós"}
+            ]
         }
 
         chat = [