Zekun Wu commited on
Commit
65499e3
1 Parent(s): f97d08f
Files changed (1) hide show
  1. util/injection.py +1 -0
util/injection.py CHANGED
@@ -47,6 +47,7 @@ def invoke_retry(prompt, agent, parameters,string_input=False):
47
  while attempts < max_attempts:
48
  try:
49
  score_text = agent.invoke(prompt, **parameters)
 
50
  print(f"Score text: {score_text}")
51
  print("=============================================================")
52
  if string_input:
 
47
  while attempts < max_attempts:
48
  try:
49
  score_text = agent.invoke(prompt, **parameters)
50
+ print(f"Prompt: {prompt}")
51
  print(f"Score text: {score_text}")
52
  print("=============================================================")
53
  if string_input: