Zekun Wu commited on
Commit
0516ae0
1 Parent(s): 5b7915e
Files changed (1) hide show
  1. util/injection.py +6 -4
util/injection.py CHANGED
@@ -20,6 +20,8 @@ def create_charateristics(original_resume, group_name, occupation, agent, parame
20
 
21
  additional_charateristics = invoke_retry(prompt, agent, parameters)
22
 
 
 
23
  combined_charateristics = f"{original_resume}\n{additional_charateristics}"
24
 
25
  return combined_charateristics
@@ -98,13 +100,13 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
98
  summary.append(f"{readable_name}: {value};")
99
 
100
  if additional_charateristics == True:
101
- additional_charateristics = create_charateristics('\n'.join(summary), group_name, occupation, agent, parameters)
102
  else:
103
- additional_charateristics = ""
104
 
105
  for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
106
- prompt_temp = create_summary(row, group_name, label, occupation, additional_charateristics)
107
- print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{additional_charateristics}")
108
  print("=============================================================")
109
  result = invoke_retry(prompt_temp, agent, parameters)
110
  scores[key][index].append(result)
 
20
 
21
  additional_charateristics = invoke_retry(prompt, agent, parameters)
22
 
23
+ print(f"Additional characteristics: {additional_charateristics}")
24
+
25
  combined_charateristics = f"{original_resume}\n{additional_charateristics}"
26
 
27
  return combined_charateristics
 
100
  summary.append(f"{readable_name}: {value};")
101
 
102
  if additional_charateristics == True:
103
+ charateristics = create_charateristics('\n'.join(summary), group_name, occupation, agent, parameters)
104
  else:
105
+ charateristics = ""
106
 
107
  for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
108
+ prompt_temp = create_summary(row, group_name, label, occupation, charateristics)
109
+ print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{charateristics}")
110
  print("=============================================================")
111
  result = invoke_retry(prompt_temp, agent, parameters)
112
  scores[key][index].append(result)