Spaces:
Sleeping
Sleeping
Zekun Wu
commited on
Commit
•
0516ae0
1
Parent(s):
5b7915e
update
Browse files- 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 |
-
|
102 |
else:
|
103 |
-
|
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,
|
107 |
-
print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{
|
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)
|