|
from flask import Flask, request, jsonify |
|
import requests |
|
app = Flask(__name__) |
|
@app.route('/claude', methods=['POST']) |
|
def claude(): |
|
messages = request.json['messages'] |
|
API_KEY = request.json['API_KEY'] |
|
model = request.json['model'] |
|
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"nn{role}: {i['content']}" |
|
prompt += 'nnAssistant: ' |
|
data = { |
|
'model': model, |
|
'prompt': prompt |
|
} |
|
response = requests.post( |
|
'https://api.ddosxd.ru/v1/prompt', |
|
headers=headers, json=data, |
|
) |
|
return jsonify(response.json()) |
|
if __name__ == '__main__': |
|
app.run(debug=True) |
|
|
