Merge pull request #37 from stakpak/patch/delay

Patch: Delay

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "stakpak"
-version = "0.1.37"
+version = "0.1.38"
 edition = "2021"
 
 [dependencies]

src/commands/agent/mod.rs

@@ -208,7 +208,7 @@ impl Action {
                                     // Ctrl+P
                                     return Err("re-prompt".to_string());
                                 }
-                                print(line);
+                                println!("{}", line);
                                 lines.push(line.to_string());
                             }
                             Err(e) => return Err(format!("Failed to read input: {}", e)),
@@ -254,7 +254,7 @@ impl Action {
                 }
 
                 let confirmation = input.trim().to_lowercase();
-                print(confirmation.as_str());
+                println!("{}", confirmation.as_str());
 
                 match confirmation.as_str() {
                     "edit" => {
@@ -470,7 +470,7 @@ impl<'a> AgentOutputListener<'a> {
         let subscription_complete = Arc::new(AtomicBool::new(false));
 
         for retry in 0.. {
-            sleep(Duration::from_millis(100 * (retry + 1))).await;
+            sleep(Duration::from_millis(200 * (retry + 1))).await;
 
             let subscription_complete_clone = Arc::clone(&subscription_complete);
             let ack_callback =
@@ -490,7 +490,7 @@ impl<'a> AgentOutputListener<'a> {
                 )
                 .await
             {
-                if retry >= 5 {
+                if retry >= 9 {
                     return Err(format!("Failed to subscribe to session: {}", e));
                 }
             }

src/commands/agent/run_agent.rs

@@ -66,6 +66,10 @@ pub async fn run_agent(
             };
         },
         false => {
+            println!(
+                "\x1b[33m[NOTE: This agent is in non-interactive mode. Use the \x1b[1;32m-i\x1b[33m flag to enable interactive mode.]\x1b[0m"
+            );
+
             if let Some(flow_ref) = &session.flow_ref {
                 let url = format!("{}?session_id={}", flow_ref.to_url(), session.id);
                 println!("\x1b[1;34m{}\x1b[0m", "━".repeat(80));

src/commands/flow/sync.rs

@@ -325,7 +325,7 @@ async fn wait_for_subscription(
     let subscription_complete = Arc::new(AtomicBool::new(false));
 
     for retry in 0.. {
-        sleep(Duration::from_millis(100 * (retry + 1))).await;
+        sleep(Duration::from_millis(200 * (retry + 1))).await;
 
         let subscription_complete_clone = Arc::clone(&subscription_complete);
         let ack_callback = move |_: Payload, _: SocketClient| -> BoxFuture<'static, ()> {
@@ -344,7 +344,7 @@ async fn wait_for_subscription(
             )
             .await
         {
-            if retry >= 5 {
+            if retry >= 9 {
                 return Err(format!("Failed to subscribe to session: {}", e));
             }
         }

src/utils/output.rs

@@ -69,15 +69,15 @@ pub async fn setup_output_handler(
                 "session_id": session_id
             });
 
-            for retry in 0..5 {
+            for retry in 0..10 {
                 match socket_client.emit("publish", payload.clone()).await {
                     Ok(_) => break,
                     Err(e) => {
-                        if retry == 4 {
+                        if retry == 9 {
                             eprintln!("Failed to publish message: {}", e);
                             break;
                         }
-                        tokio::time::sleep(std::time::Duration::from_millis(100 * (retry + 1)))
+                        tokio::time::sleep(std::time::Duration::from_millis(200 * (retry + 1)))
                             .await;
                     }
                 }