update

util/injection.py

@@ -62,7 +62,7 @@ def create_summary(group_name, label, occupation, additional_charateristics):
     return "\n".join(summary)
 
 
-def invoke_retry(prompt, agent, parameters):
+def invoke_retry(prompt, agent, parameters,string_input=False):
     attempts = 0
     delay = 2  # Initial delay in seconds
     max_attempts = 20  # Maximum number of retry attempts
@@ -72,6 +72,8 @@ def invoke_retry(prompt, agent, parameters):
             score_text = agent.invoke(prompt, **parameters)
             print(f"Score text: {score_text}")
             print("=============================================================")
+            if string_input:
+                return score_text
             score = re.search(r'\d+', score_text)
             return int(score.group()) if score else -1
         except Exception as e: