NeonBohdan commited on
Commit
dbfda4a
·
1 Parent(s): 1fecca1

Added historical figure line

Browse files
Files changed (1) hide show
  1. utils.py +5 -1
utils.py CHANGED
@@ -79,11 +79,15 @@ class PersonaConverter:
79
  "You're attempting to provide a concise response within a 60-word limit.\n"
80
  )
81
 
82
- def __call__(self, name: str, description: str, imaginary: bool, short: bool):
83
  system_prompt_list = []
84
  if (name != "None"):
85
  system_prompt_list.append(self.personas[name])
86
 
 
 
 
 
87
  if (len(description) != 0):
88
  system_prompt_list.append(f"{description}\n")
89
 
 
79
  "You're attempting to provide a concise response within a 60-word limit.\n"
80
  )
81
 
82
+ def __call__(self, name: str, figure: str, description: str, imaginary: bool, short: bool):
83
  system_prompt_list = []
84
  if (name != "None"):
85
  system_prompt_list.append(self.personas[name])
86
 
87
+ if (figure != "None"):
88
+ historical_figure = self.historical_figure_format.format(biography=self.historical_figures[figure])
89
+ system_prompt_list.append(historical_figure)
90
+
91
  if (len(description) != 0):
92
  system_prompt_list.append(f"{description}\n")
93