broadfield-dev commited on
Commit
2b9bc40
·
verified ·
1 Parent(s): 819b530

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -59,13 +59,13 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
59
  else:
60
  lod=[]
61
  if role == "MANAGER":
62
- system_prompt = prompts.MANAGER.replace("**TIMELINE**",data[4]).replace("**HISTORY**",str(history))
63
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
64
  elif role == "PATHMAKER":
65
- system_prompt = 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))
66
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
67
  elif role == "CREATE_FILE":
68
- system_prompt = 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]))
69
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
70
  elif role == "SEARCH":
71
  system_prompt = prompts.SEARCH.replace("**DATA**",data)
 
59
  else:
60
  lod=[]
61
  if role == "MANAGER":
62
+ system_prompt = prompts.PREFIX + prompts.MANAGER.replace("**TIMELINE**",data[4]).replace("**HISTORY**",str(history))
63
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
64
  elif role == "PATHMAKER":
65
+ 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))
66
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
67
  elif role == "CREATE_FILE":
68
+ 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]))
69
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
70
  elif role == "SEARCH":
71
  system_prompt = prompts.SEARCH.replace("**DATA**",data)