forgot return in retry multiple

ServiceNow · Dec 3, 2024 · e124fea · e124fea
1 parent da8208f
commit e124fea
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/agentlab/llm/llm_utils.py b/src/agentlab/llm/llm_utils.py
@@ -136,6 +136,10 @@ def retry_multiple(
                 parsed_answers.append(parser(answer["content"]))
             except ParseError as parsing_error:
                 errors.append(str(parsing_error))
+        # if we have a valid answer, return it
+        if parsed_answers:
+            return parsed_answers, tries
+        else:
             tries += 1
             if log:
                 msg = f"Query failed. Retrying {tries}/{n_retry}.\n[LLM]:\n{answer['content']}\n[User]:\n{str(errors)}"