Update app.py

app.py

@@ -1,30 +1,39 @@
 from flask import Flask, request, jsonify
 import requests
+from retrying import retry
+
 app = Flask(__name__)
+
+# Функция для повторной попытки выполнения запроса в случае ошибки
+@retry(wait_exponential_multiplier=1000, wait_exponential_max=10000, stop_max_attempt_number=5)
+def send_request(model, headers, data):
+    return requests.post('https://api.ddosxd.ru/v1/prompt', headers=headers, json=data)
+
 @app.route('/claude', methods=['POST'])
 def claude():
     model = request.get_json().get("model", "claude-2")
-    
     API_KEY = request.get_json().get("api_key")
-    
     messages = request.get_json().get("messages")
     headers = {'Authorization': API_KEY}
     prompt = ""
+    
     for i in messages:
         role = "Human" if i["role"] == "user" else (
             f'{i["role"][0].upper()}{i["role"][1:]}'
         )
         prompt += f"\n\n{role}: {i['content']}"
+    
     prompt += '\n\nAssistant: '
     data = {
         'model': model,
         'prompt': prompt
     }
-    response = requests.post(
-        'https://api.ddosxd.ru/v1/prompt',
-        headers=headers, json=data,
-    )
+
+    # Используем функцию send_request с повторными попытками в случае ошибки
+    response = send_request(model, headers, data)
+
     print(response)
     return jsonify(response.json())
+
 if __name__ == '__main__':
     app.run(host="0.0.0.0", port=7860, debug=False)