Update app.py

app.py

@@ -30,6 +30,7 @@ MISTRAL_API_KEY = os.getenv('MISTRAL_API_KEY')
 MISTRAL_API_KEY_2 = os.getenv('MISTRAL_API_KEY_2')
 MISTRAL_API_KEY_3 = os.getenv('MISTRAL_API_KEY_3')
 MISTRAL_API_KEY_4 = os.getenv('MISTRAL_API_KEY_4')
+MISTRAL_API_KEY_5 = os.getenv('MISTRAL_API_KEY_5')
 token = os.getenv('GITHUB_TOKEN')
 
 # Клиент для генерации сообщений
@@ -39,6 +40,7 @@ client_mistral_generate = Mistral(api_key=MISTRAL_API_KEY)
 client_mistral_check = Mistral(api_key=MISTRAL_API_KEY_2)
 client_mistral_check_2 = Mistral(api_key=MISTRAL_API_KEY_3)
 client_mistral_check_3 = Mistral(api_key=MISTRAL_API_KEY_4)
+client_mistral_check_4 = Mistral(api_key=MISTRAL_API_KEY_5)
 
 morph = pymorphy3.MorphAnalyzer()
 
@@ -346,7 +348,8 @@ async def generate_message_mistral_check_3(prompt, max_retries=5):
     print("Failed to get response from Mistral after maximum retries.")
     return None
 
-#def generate_check_gigachat_pro(prompt):
+async def generate_message_mistral_check_4(prompt, max_retries=5):
+#def generate_message_mistral_check(prompt):    
 #    try:
 #        messages = [SystemMessage(content=prompt)]
 #        res2 = chat_pro_check(messages)
@@ -354,6 +357,34 @@ async def generate_message_mistral_check_3(prompt, max_retries=5):
 #        return cleaned_message
 #    except Exception as e:
 #        return f"Ошибка при обращении к GigaChat-Pro: {e}"
+    retries = 0
+    while retries < max_retries:
+        try:
+            chat_response = await client_mistral_check_4.chat.complete_async(
+                model="mistral-large-latest",
+                temperature=0.2,
+                messages=[
+                    {
+                        "role": "user",
+                        "content": prompt
+                    },
+                ]
+            )
+            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
+            return cleaned_message
+        except Exception as e:
+            error_message = str(e)
+            if "Status 429" in error_message or "Server disconnected without sending a response" in error_message:
+                wait_time = 3
+                print(f"Rate limit exceeded or server did not respond. Waiting {wait_time} seconds before retrying...")
+                await asyncio.sleep(wait_time)
+                retries += 1
+            else:
+                print(f"Error calling Mistral: {e}")
+                return None
+    print("Failed to get response from Mistral after maximum retries.")
+    return None
+
 
 # Функция для замены сокращений с 'k' или 'К' на тысячи
 def replace_k_with_thousands(message):
@@ -1387,7 +1418,7 @@ async def check_synonymous_members(message):
 если таких слов или выражений нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
 **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check(prompt)
+    response = await generate_message_mistral_check_2(prompt)
 #    await asyncio.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
@@ -1425,7 +1456,7 @@ async def check_clickbait_phrases(message, description, benefits):
     
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check(prompt)
+    response = await generate_message_mistral_check_4(prompt)
 #    await asyncio.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
@@ -1514,7 +1545,7 @@ async def check_specialized_terms(message):
     
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check(prompt)
+    response = await generate_message_mistral_check_2(prompt)
 #    await asyncio.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
@@ -1543,7 +1574,7 @@ async def check_offensive_phrases(message):
 если таких фраз нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
 **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check(prompt)
+    response = await generate_message_mistral_check_4(prompt)
 #    await asyncio.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
@@ -1556,7 +1587,7 @@ async def check_offensive_phrases(message):
         return None
 
 # 28. Проверка на речевые клише, рекламные штампы и канцеляризмы
-async def check_cliches_and_bureaucratese_3(message, description, benefits):
+async def check_cliches_and_bureaucratese(message, description, benefits):
     print()
     print("Проверка 28: Проверка на речевые клише, рекламные штампы и канцеляризмы")
     print()
@@ -1588,7 +1619,7 @@ async def check_cliches_and_bureaucratese_3(message, description, benefits):
     если в тексте **есть** такие выражения, **верни только** JSON {{"decision": true, "explanation": "<пояснение>"}}.
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check(prompt)
+    response = await generate_message_mistral_check_3(prompt)
 #    await asyncio.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
@@ -1674,7 +1705,7 @@ async def check_product_name_consistency(message, product_name):
 Если название продукта **не** совпадает, **верни только** JSON {{"decision": true, "explanation": "<описание несоответствия>"}}.
 **Не добавляй никакого дополнительного текста. Ответ должен быть только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check(prompt)
+    response = await generate_message_mistral_check_3(prompt)
 #    await asyncio.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)