Spaces:
MiklX
/
Runtime error

MiklX commited on
Commit
ecc5130
·
1 Parent(s): c3005f0

Create copyapi.py

Browse files
Files changed (1) hide show
  1. copyapi.py +30 -0
copyapi.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, request, jsonify
2
+ import requests
3
+ app = Flask(__name__)
4
+ @app.route('/claude', methods=['POST'])
5
+ def claude():
6
+ model = request.get_json().get("model", "claude-2")
7
+
8
+ API_KEY = request.get_json().get("api_key")
9
+
10
+ messages = request.get_json().get("messages")
11
+ headers = {'Authorization': API_KEY}
12
+ prompt = ""
13
+ for i in messages:
14
+ role = "Human" if i["role"] == "user" else (
15
+ f'{i["role"][0].upper()}{i["role"][1:]}'
16
+ )
17
+ prompt += f"\n\n{role}: {i['content']}"
18
+ prompt += '\n\nAssistant: '
19
+ data = {
20
+ 'model': model,
21
+ 'prompt': prompt
22
+ }
23
+ response = requests.post(
24
+ 'https://api.ddosxd.ru/v1/prompt',
25
+ headers=headers, json=data,
26
+ )
27
+ print(response)
28
+ return jsonify(response.json())
29
+ if __name__ == '__main__':
30
+ app.run(host="0.0.0.0", port=7860, debug=False)