Update app.py
Browse files
app.py
CHANGED
@@ -5,6 +5,7 @@ app = Flask(__name__)
|
|
5 |
def claude():
|
6 |
messages = request.json['messages']
|
7 |
API_KEY = request.json['API_KEY']
|
|
|
8 |
headers = {'Authorization': API_KEY}
|
9 |
prompt = ""
|
10 |
for i in messages:
|
@@ -14,7 +15,7 @@ def claude():
|
|
14 |
prompt += f"nn{role}: {i['content']}"
|
15 |
prompt += 'nnAssistant: '
|
16 |
data = {
|
17 |
-
'model':
|
18 |
'prompt': prompt
|
19 |
}
|
20 |
response = requests.post(
|
|
|
5 |
def claude():
|
6 |
messages = request.json['messages']
|
7 |
API_KEY = request.json['API_KEY']
|
8 |
+
model = request.json['model']
|
9 |
headers = {'Authorization': API_KEY}
|
10 |
prompt = ""
|
11 |
for i in messages:
|
|
|
15 |
prompt += f"nn{role}: {i['content']}"
|
16 |
prompt += 'nnAssistant: '
|
17 |
data = {
|
18 |
+
'model': model,
|
19 |
'prompt': prompt
|
20 |
}
|
21 |
response = requests.post(
|