fix videobots tests

GooeyAI · Feb 28, 2024 · 328889c · 328889c
1 parent 2589be3
commit 328889c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/recipes/VideoBots.py b/recipes/VideoBots.py
@@ -632,6 +632,10 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
                 """
             )
 
+        state.update(
+            dict(final_prompt=[], output_text=[], output_audio=[], output_video=[])
+        )
+
         user_input = request.input_prompt.strip()
         if not (user_input or request.input_images or request.input_documents):
             return
@@ -896,9 +900,6 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
             else:
                 yield f"Streaming{str(i + 1).translate(SUPERSCRIPT)} {model.value}..."
 
-        state["output_audio"] = []
-        state["output_video"] = []
-
         if not request.tts_provider:
             return
         tts_state = dict(state)