broadfield-dev commited on
Commit
9677509
·
verified ·
1 Parent(s): 7475973

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -75,7 +75,7 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
75
  system_prompt = prompts.PREFIX + prompts.PATH_MAKER.replace("**FILE_LIST**",str(data[3])).replace("**CURRENT_OR_NONE**",str(data[4])).replace("**PROMPT**",json.dumps(data[0],indent=4)).replace("**HISTORY**",str(history))
76
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
77
  elif role == "CREATE_FILE":
78
- system_prompt = prompts.PREFIX + prompts.CREATE_FILE.replace("**FILE_LIST**",str(data[3])).replace("**TIMELINE**",data[4]).replace("**FILENAME**",str(data[1])).replace("**TEMPLATE_OR_NONE**",str(data[2]).format())
79
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
80
  elif role == "SEARCH":
81
  system_prompt = prompts.SEARCH.replace("**DATA**",data)
 
75
  system_prompt = prompts.PREFIX + prompts.PATH_MAKER.replace("**FILE_LIST**",str(data[3])).replace("**CURRENT_OR_NONE**",str(data[4])).replace("**PROMPT**",json.dumps(data[0],indent=4)).replace("**HISTORY**",str(history))
76
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
77
  elif role == "CREATE_FILE":
78
+ system_prompt = prompts.PREFIX + prompts.CREATE_FILE.replace("**FILE_LIST**",str(data[3])).replace("**TIMELINE**",data[4]).replace("**FILENAME**",str(data[1])).replace("**TEMPLATE_OR_NONE**",str(data[2]).strip("''").format())
79
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
80
  elif role == "SEARCH":
81
  system_prompt = prompts.SEARCH.replace("**DATA**",data)