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 |
|