Add 'system' messages to format_prompt
Browse files
app.py
CHANGED
|
@@ -252,6 +252,8 @@ def format_prompt(messages):
|
|
| 252 |
prompt += f"User: {msg['content'].strip()}\n"
|
| 253 |
elif role == "assistant":
|
| 254 |
prompt += f"Assistant: {msg['content'].strip()}\n"
|
|
|
|
|
|
|
| 255 |
prompt += "Assistant:"
|
| 256 |
return prompt
|
| 257 |
|
|
|
|
| 252 |
prompt += f"User: {msg['content'].strip()}\n"
|
| 253 |
elif role == "assistant":
|
| 254 |
prompt += f"Assistant: {msg['content'].strip()}\n"
|
| 255 |
+
elif role == "system":
|
| 256 |
+
prompt += f"System: {msg['content'].strip()}\n"
|
| 257 |
prompt += "Assistant:"
|
| 258 |
return prompt
|
| 259 |
|