Print less warnings from mistral wrapper when they do not indicate a …

…real problem

PiperOrigin-RevId: 672574924
Change-Id: I6c5f33d8c80a35ba2d54bad1d6b0d92b3f1197c3

concordia/language_model/mistral_model.py

@@ -34,6 +34,8 @@
     'codestral-2405',
 )
 
+_NUM_SILENT_ATTEMPTS = 3
+
 
 class MistralLanguageModel(language_model.LanguageModel):
   """Language Model wrapper that uses Mistral models."""
@@ -97,8 +99,9 @@ def _complete_text(
     result = ''
     for attempts in range(_MAX_CHAT_ATTEMPTS):
       if attempts > 0:
-        print('Sleeping for 10 seconds... ' +
-              f'attempt: {attempts} / {_MAX_CHAT_ATTEMPTS}')
+        if attempts >= _NUM_SILENT_ATTEMPTS:
+          print('Sleeping for 10 seconds... ' +
+                f'attempt: {attempts} / {_MAX_CHAT_ATTEMPTS}')
         time.sleep(10)
       try:
         response = self._client.fim.complete(
@@ -163,8 +166,9 @@ def _chat_text(
     ]
     for attempts in range(_MAX_CHAT_ATTEMPTS):
       if attempts > 0:
-        print('Sleeping for 10 seconds... ' +
-              f'attempt: {attempts} / {_MAX_CHAT_ATTEMPTS}')
+        if attempts >= _NUM_SILENT_ATTEMPTS:
+          print('Sleeping for 10 seconds... ' +
+                f'attempt: {attempts} / {_MAX_CHAT_ATTEMPTS}')
         time.sleep(10)
       try:
         response = self._client.chat.complete(