The previous change didn't fully suppress harmless mistral exceptions…

…, this one fixes it. PiperOrigin-RevId: 672837691 Change-Id: Ie232ceb2ab69961cc6e978fa32a4862961c7cd87
google-deepmind · Sep 10, 2024 · f4d0804 · f4d0804
1 parent f2d9447
commit f4d0804
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/concordia/language_model/mistral_model.py b/concordia/language_model/mistral_model.py
@@ -114,7 +114,8 @@ def _complete_text(
             random_seed=seed,
         )
       except mistralai.models.sdkerror.SDKError as err:
-        print(f'  Exception: {err}')
+        if attempts >= _NUM_SILENT_ATTEMPTS:
+          print(f'  Exception: {err}')
         continue
       else:
         result = response.choices[0].message.content
@@ -179,7 +180,8 @@ def _chat_text(
             random_seed=seed,
         )
       except mistralai.models.sdkerror.SDKError as err:
-        print(f'  Exception: {err}')
+        if attempts >= _NUM_SILENT_ATTEMPTS:
+          print(f'  Exception: {err}')
         continue
       else:
         return response.choices[0].message.content