From 4dc3d84d3842a22e3f64d79b9be4b06462cc6f22 Mon Sep 17 00:00:00 2001
From: pavel <pavelflegr@gmail.com>
Date: Wed, 11 Feb 2026 00:26:47 +0100
Subject: [PATCH] fix: improve task status

---
 src/scheduler.rs | 8 ++++----
 src/server.rs    | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/scheduler.rs b/src/scheduler.rs
index d20f257..6f3632d 100644
--- a/src/scheduler.rs
+++ b/src/scheduler.rs
@@ -98,14 +98,14 @@ impl Scheduler {
         let mut agent = Agent::new(zen_key, tavily_key, task.goal.clone())?;
 
         tokio::spawn(async move {
-            let (logs, answer) = match agent.run().await {
-                Ok(res) => res,
-                Err(e) => (format!("Scheduled run failed: {}", e), None),
+            let (logs, answer, status) = match agent.run().await {
+                Ok((logs, answer)) => (logs, answer, "completed".to_string()),
+                Err(e) => (format!("Scheduled run failed: {}", e), None, "failed".to_string()),
             };
 
             let run_complete = task_run::ActiveModel {
                 id: Set(run_id),
-                status: Set("completed".to_string()),
+                status: Set(status),
                 logs: Set(logs),
                 answer: Set(answer),
                 ..Default::default()
diff --git a/src/server.rs b/src/server.rs
index 7d529e2..2033484 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -275,9 +275,9 @@ async fn execute_agent_run(
     )
     .map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, e.to_string()))?;
 
-    let (logs, answer) = match agent.run().await {
-        Ok((logs, answer)) => (logs, answer),
-        Err(e) => (format!("Execution failed: {}", e), None),
+    let (logs, answer, status) = match agent.run().await {
+        Ok((logs, answer)) => (logs, answer, "completed".to_string()),
+        Err(e) => (format!("Execution failed: {}", e), None, "failed".to_string()),
     };
 
     // Update with final logs and status
@@ -293,7 +293,7 @@ async fn execute_agent_run(
 
     run.logs = Set(logs.clone());
     run.answer = Set(answer.clone());
-    run.status = Set("completed".to_string());
+    run.status = Set(status);
 
     run.update(&state.db)
         .await