From 5a8f0194d90655429328a7ffb7add139f9ea710e Mon Sep 17 00:00:00 2001
From: Florian Zimmermeister <flozi00.fz@gmail.com>
Date: Mon, 5 May 2025 14:13:03 +0200
Subject: [PATCH] Check streaming delta is not None

---
 src/smolagents/agents.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/smolagents/agents.py b/src/smolagents/agents.py
index d052f144a..e79b2487d 100644
--- a/src/smolagents/agents.py
+++ b/src/smolagents/agents.py
@@ -453,8 +453,9 @@ def _generate_planning_step(
             if self.stream_outputs and hasattr(self.model, "generate_stream"):
                 plan_message_content = ""
                 for completion_delta in self.model.generate_stream(input_messages, stop_sequences=["<end_plan>"]):  # type: ignore
-                    plan_message_content += completion_delta.content
-                    yield completion_delta
+                    if completion_delta.content is not None:
+                        plan_message_content += completion_delta.content
+                        yield completion_delta
             else:
                 plan_message_content = self.model.generate(input_messages, stop_sequences=["<end_plan>"]).content
             plan = textwrap.dedent(
@@ -496,8 +497,9 @@ def _generate_planning_step(
             if self.stream_outputs and hasattr(self.model, "generate_stream"):
                 plan_message_content = ""
                 for completion_delta in self.model.generate_stream(input_messages, stop_sequences=["<end_plan>"]):  # type: ignore
-                    plan_message_content += completion_delta.content
-                    yield completion_delta
+                    if completion_delta.content is not None:
+                        plan_message_content += completion_delta.content
+                        yield completion_delta
             else:
                 plan_message_content = self.model.generate(input_messages, stop_sequences=["<end_plan>"]).content
             plan = textwrap.dedent(