update

util/injection.py

@@ -69,8 +69,8 @@ def invoke_retry(prompt, agent, parameters,string_input=False):
     while attempts < max_attempts:
         try:
             score_text = agent.invoke(prompt, **parameters)
-            print(f"Score text: {score_text}")
-            print("=============================================================")
+            # print(f"Score text: {score_text}")
+            # print("=============================================================")
             if string_input:
                 return score_text
             score = re.search(r'\d+', score_text)
@@ -91,7 +91,6 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
     scores = {key: [[] for _ in range(len(df))] for key in ['Privilege_characteristics', 'Privilege_normal',
                                                              'Protect_characteristics', 'Protect_normal',
                                                              'Neutral_characteristics', 'Neutral_normal']}
-    print(f"Scores: {scores}")
 
     for run in tqdm(range(num_run), desc="Processing runs", unit="run"):
         for index, (idx, row) in tqdm(enumerate(df.iterrows()), total=len(df), desc="Processing entries", unit="entry"):
@@ -103,15 +102,14 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
 
 
             charateristics = create_charateristics('\n'.join(summary), group_name, occupation, agent, parameters)
-            #charateristics = "This is a test. This is only a test."
 
 
             for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
                 prompt_charateristics = create_summary(group_name, label, occupation, charateristics)
                 prompt_normal = create_summary(group_name, label, occupation, '\n'.join(summary))
 
-                print(f"Run {run + 1} - Entry {index + 1} - {key}")
-                print("=============================================================")
+                # print(f"Run {run + 1} - Entry {index + 1} - {key}")
+                # print("=============================================================")
                 result_charateristics = invoke_retry(prompt_charateristics, agent, parameters)
                 result_normal = invoke_retry(prompt_normal, agent, parameters)
                 scores[key+"_characteristics"][index].append(result_charateristics)