Update app.py
Browse files
app.py
CHANGED
@@ -33,8 +33,10 @@ def respond(
|
|
33 |
):
|
34 |
result = 'none'
|
35 |
try:
|
36 |
-
|
37 |
-
|
|
|
|
|
38 |
for val in history:
|
39 |
if val[0]:
|
40 |
messages.append({"role": "user", "content": val[0]})
|
@@ -44,8 +46,8 @@ def respond(
|
|
44 |
messages.append({"role": "user", "content": message})
|
45 |
|
46 |
for message in messages:
|
47 |
-
|
48 |
-
|
49 |
# prompt += f"\n<|user|>\n{user_text}\n<|end|><|assistant|>\n"
|
50 |
|
51 |
# Generate the response
|
|
|
33 |
):
|
34 |
result = 'none'
|
35 |
try:
|
36 |
+
# Build the conversation prompt
|
37 |
+
prompt = ''
|
38 |
+
if (len(system_message)>0):
|
39 |
+
prompt = "<|assistant|>"+system_message+f"<|end|>\n"
|
40 |
for val in history:
|
41 |
if val[0]:
|
42 |
messages.append({"role": "user", "content": val[0]})
|
|
|
46 |
messages.append({"role": "user", "content": message})
|
47 |
|
48 |
for message in messages:
|
49 |
+
role = "<|assistant|>" if message['role'] == 'assistant' else "<|user|>"
|
50 |
+
prompt += f"\n{role}\n{message['content']}\n<|end|>\n"
|
51 |
# prompt += f"\n<|user|>\n{user_text}\n<|end|><|assistant|>\n"
|
52 |
|
53 |
# Generate the response
|