fix(cli): verify task completion to detect partial responses

Fixes bounty issue #1506

The CLI run command now tracks whether a TaskComplete event was received.
If the event stream ends without completion or an explicit error, it is treated
as a partial response/error condition. This prevents misleading success
status when the server disconnects prematurely.

Author: factorydroid

cortex-cli/src/run_cmd.rs

@@ -241,17 +241,15 @@ impl RunCli {
     pub async fn run(self) -> Result<()> {
         // Validate temperature if provided
         if let Some(temp) = self.temperature
-            && !(0.0..=2.0).contains(&temp)
-        {
-            bail!("Temperature must be between 0.0 and 2.0, got {temp}");
-        }
+            && !(0.0..=2.0).contains(&temp) {
+                bail!("Temperature must be between 0.0 and 2.0, got {temp}");
+            }
 
         // Validate top_p if provided
         if let Some(top_p) = self.top_p
-            && !(0.0..=1.0).contains(&top_p)
-        {
-            bail!("top-p must be between 0.0 and 1.0, got {top_p}");
-        }
+            && !(0.0..=1.0).contains(&top_p) {
+                bail!("top-p must be between 0.0 and 1.0, got {top_p}");
+            }
 
         // Build the message from arguments
         let mut message = self
@@ -437,10 +435,9 @@ impl RunCli {
             SessionMode::New { title } => {
                 let id = uuid::Uuid::new_v4().to_string();
                 if let Some(ref t) = title
-                    && self.verbose
-                {
-                    eprintln!("New session: {id} (title: {t})");
-                }
+                    && self.verbose {
+                        eprintln!("New session: {id} (title: {t})");
+                    }
                 id
             }
         };
@@ -522,6 +519,7 @@ impl RunCli {
         let mut event_count = 0u64;
         let mut streaming_started = false;
         let mut error_occurred = false;
+        let mut completed = false;
 
         // Set up timeout if specified
         let timeout_duration = if self.timeout > 0 {
@@ -535,11 +533,11 @@ impl RunCli {
         while let Ok(event) = handle.event_rx.recv().await {
             // Check timeout
             if let Some(timeout) = timeout_duration
-                && start_time.elapsed() > timeout
-            {
-                eprintln!("Timeout reached after {} seconds", self.timeout);
-                break;
-            }
+                && start_time.elapsed() > timeout {
+                    eprintln!("Timeout reached after {} seconds", self.timeout);
+                    error_occurred = true;
+                    break;
+                }
 
             event_count += 1;
 
@@ -626,6 +624,7 @@ impl RunCli {
                     // Tool call completed
                 }
                 EventMsg::TaskComplete(_) => {
+                    completed = true;
                     break;
                 }
                 EventMsg::Error(e) => {
@@ -667,6 +666,28 @@ impl RunCli {
             }
         }
 
+        // Check for partial response (disconnected without TaskComplete or Error)
+        if !completed && !error_occurred {
+            error_occurred = true;
+            let err_msg = "Response incomplete (stream ended unexpectedly)";
+
+            if is_json {
+                let err_json = serde_json::json!({
+                    "type": "error",
+                    "message": err_msg,
+                    "session_id": session_id,
+                });
+                eprintln!("{}", serde_json::to_string(&err_json)?);
+            } else {
+                eprintln!(
+                    "{}Error:{} {}",
+                    TermColor::Red.ansi_code(),
+                    TermColor::Default.ansi_code(),
+                    err_msg
+                );
+            }
+        }
+
         // Handle JSON result output
         if matches!(self.format, OutputFormat::Json) {
             let result = serde_json::json!({