devbonzai 2.1.5 → 2.1.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "devbonzai",
-  "version": "2.1.5",
+  "version": "2.1.6",
   "description": "Quickly set up a local file server in any repository for browser-based file access",
   "main": "cli.js",
   "bin": {

package/templates/receiver.js

@@ -1709,15 +1709,15 @@ app.post('/prompt_agent_stream', (req, res) => {
     }
     
     // Always try to send complete event when process finishes
-    // Only skip if we've already sent it
+    // Check actual response stream state rather than relying on responseSent flag
+    // Only skip if we've already sent it (responseSent flag prevents duplicates)
     if (!responseSent) {
       try {
-        // Check if response is still writable - try to send even if clientDisconnected flag is set
-        // because the response stream might still be open
+        // Check if response is still writable - check actual stream state
         if (res.destroyed || res.closed) {
           console.log('⚠️ [prompt_agent_stream] Response already closed, cannot send complete event');
         } else {
-          responseSent = true;
+          // Send events and only set flag after successful send
           sendEvent('complete', {
             code,
             stdout,
@@ -1729,6 +1729,7 @@ app.post('/prompt_agent_stream', (req, res) => {
           // Send stop event after complete
           sendEvent('stop', {});
           res.end();
+          responseSent = true;
           console.log('✅ [prompt_agent_stream] Sent complete and stop events');
         }
       } catch (e) {
@@ -1745,8 +1746,7 @@ app.post('/prompt_agent_stream', (req, res) => {
   req.on('close', () => {
     console.log('🔵 [prompt_agent_stream] Client disconnected (process continues in background)');
     // Don't kill the process - let it complete
-    // Just mark that we shouldn't try to send more events
-    responseSent = true;
+    // Don't set responseSent here - let proc.on('close') check actual stream state
   });
 });