Ruurd commited on
Commit
fecd98b
·
1 Parent(s): 46c0897

Add 'system' messages to format_prompt

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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