Skip to content

Commit e124fea

Browse files
committed
forgot return in retry multiple
1 parent da8208f commit e124fea

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/agentlab/llm/llm_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,10 @@ def retry_multiple(
136136
parsed_answers.append(parser(answer["content"]))
137137
except ParseError as parsing_error:
138138
errors.append(str(parsing_error))
139+
# if we have a valid answer, return it
140+
if parsed_answers:
141+
return parsed_answers, tries
142+
else:
139143
tries += 1
140144
if log:
141145
msg = f"Query failed. Retrying {tries}/{n_retry}.\n[LLM]:\n{answer['content']}\n[User]:\n{str(errors)}"

0 commit comments

Comments
 (0)