update

app.py

@@ -33,7 +33,7 @@ def create_summary(row,label):
         f"Please act as a Human Resource Professional. This score will determine whether they will be hired or not."
     )
 
-def invoke_retry(prompt, **parameters):
+def invoke_retry(prompt,agent,parameters):
     attempts = 0
     delay = 2  # Initial delay in seconds
     max_attempts = 20  # Maximum number of retry attempts
@@ -51,7 +51,7 @@ def invoke_retry(prompt, **parameters):
 
     raise Exception("Failed to complete the API call after maximum retry attempts.")
 
-def process_scores(df, num_run,parameters,privilege_label,protect_label):
+def process_scores(df, num_run,parameters,privilege_label,protect_label,agent):
     """ Process entries and compute scores concurrently, with progress updates. """
     scores = {key: [[] for _ in range(len(df))] for key in ['Privilege', 'Protect', 'Neutral']}
 
@@ -59,7 +59,7 @@ def process_scores(df, num_run,parameters,privilege_label,protect_label):
         for index, row in tqdm(df.iterrows(), total=len(df), desc="Processing entries", unit="entry"):
             for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, None]):
                 prompt_temp = create_summary(row,label)
-                result = invoke_retry(prompt_temp, **parameters)
+                result = invoke_retry(prompt_temp,agent,parameters)
                 scores[key][index].append(result)
 
     # Assign score lists and calculate average scores