Add missed error state in download agent

video-db · Nov 25, 2024 · 03dc1dc · 03dc1dc
1 parent 0b65341
commit 03dc1dc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/backend/director/agents/download.py b/backend/director/agents/download.py
@@ -44,6 +44,8 @@ def run(self, stream_link: str, name: str = None, *args, **kwargs) -> AgentRespo
                 text_content.status_message = "Here is the download link"
                 self.output_message.publish()
             else:
+                text_content.status = MsgStatus.error
+                text_content.status_message = "Download failed"
                 return AgentResponse(
                     status=AgentStatus.ERROR,
                     message=f"Downloda failed with {download_response}",