Update app.py

app.py

@@ -1,39 +1,30 @@
 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
     }
-
-    # Используем функцию send_request с повторными попытками в случае ошибки
-    response = send_request(model, headers, data)
-
+    response = requests.post(
+        'https://api.ddosxd.ru/v1/prompt',
+        headers=headers, json=data,
+    )
     print(response)
     return jsonify(response.json())
-
 if __name__ == '__main__':
     app.run(host="0.0.0.0", port=7860, debug=False)