From 7e92dd8d1a92526c05b5c76123fbc89b2749a4f1 Mon Sep 17 00:00:00 2001 From: Zoe Xiao Date: Mon, 16 Jun 2025 15:31:42 -0400 Subject: [PATCH] only validate when final_answer is not None --- src/smolagents/agents.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/smolagents/agents.py b/src/smolagents/agents.py index e4f0f3dad..be3925767 100644 --- a/src/smolagents/agents.py +++ b/src/smolagents/agents.py @@ -509,7 +509,7 @@ def _execute_step(self, memory_step: ActionStep) -> Generator[ChatMessageStreamD yield el elif isinstance(el, FinalOutput): final_answer = el.output - if self.final_answer_checks: + if self.final_answer_checks and final_answer is not None: self._validate_final_answer(final_answer) yield final_answer