MilanM commited on
Commit
7e07c33
·
verified ·
1 Parent(s): ec878df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -67,7 +67,7 @@ def setup_client():
67
  def prepare_prompt(prompt, chat_history):
68
  if genparam.TYPE == "chat" and chat_history:
69
  chats = "\n".join([f"{message['role']}: \"{message['content']}\"" for message in chat_history])
70
- return f"Conversation History:\n{chats}\n\n ###\nNew User Prompt: {prompt}"
71
  return prompt
72
 
73
  def apply_prompt_syntax(prompt, system_prompt, prompt_template, bake_in_prompt_syntax):
@@ -82,9 +82,9 @@ def apply_prompt_syntax(prompt, system_prompt, prompt_template, bake_in_prompt_s
82
  "granite_code - with system": """System:\n{system_prompt}\n\nQuestion:\n{prompt}\n\nAnswer:\n""",
83
  "granite_code - instruction only": """Question:\n{prompt}\n\nAnswer:\n""",
84
  ### Mistral Models
85
- "mistral_ai_models_sys": """<s>[INST] System Prompt: {system_prompt} \n\n User Prompt: {prompt} [/INST]\n\n""",
86
- "mistral_ai_models": """<s>[INST] {prompt} [/INST]\n\n""",
87
- "mistral_ai_small_sys": """<s>[SYSTEM_PROMPT]{system_prompt}[/SYSTEM_PROMPT][INST]{prompt}[/INST]\n\n""",
88
  "mistral_ai_small_raw": """[SYSTEM_PROMPT]{system_prompt}[/SYSTEM_PROMPT][INST]{prompt}[/INST]""",
89
  ### No Syntax
90
  "no syntax - system": """{system_prompt}\n\n{prompt}""",
 
67
  def prepare_prompt(prompt, chat_history):
68
  if genparam.TYPE == "chat" and chat_history:
69
  chats = "\n".join([f"{message['role']}: \"{message['content']}\"" for message in chat_history])
70
+ return f"Conversation History:\n{chats}\n\n -----\nNew User Prompt: {prompt}"
71
  return prompt
72
 
73
  def apply_prompt_syntax(prompt, system_prompt, prompt_template, bake_in_prompt_syntax):
 
82
  "granite_code - with system": """System:\n{system_prompt}\n\nQuestion:\n{prompt}\n\nAnswer:\n""",
83
  "granite_code - instruction only": """Question:\n{prompt}\n\nAnswer:\n""",
84
  ### Mistral Models
85
+ "mistral_ai_models_sys": """<s>[INST]System Prompt: {system_prompt} \n\n User Prompt: {prompt}[/INST]""",
86
+ "mistral_ai_models": """<s>[INST]{prompt}[/INST]""",
87
+ "mistral_ai_small_sys": """<s>[SYSTEM_PROMPT]{system_prompt}[/SYSTEM_PROMPT][INST]{prompt}[/INST]""",
88
  "mistral_ai_small_raw": """[SYSTEM_PROMPT]{system_prompt}[/SYSTEM_PROMPT][INST]{prompt}[/INST]""",
89
  ### No Syntax
90
  "no syntax - system": """{system_prompt}\n\n{prompt}""",