@siftd/connect-agent 0.2.2 → 0.2.4

package/dist/heartbeat.js

@@ -10,7 +10,7 @@ import { hostname } from 'os';
 import { createHash } from 'crypto';
 import { getServerUrl, getAgentToken, getUserId, isCloudMode } from './config.js';
 const HEARTBEAT_INTERVAL = 10000; // 10 seconds
-const VERSION = '0.2.2'; // Should match package.json
+const VERSION = '0.2.4'; // Should match package.json
 const state = {
     intervalId: null,
     runnerId: null,

package/dist/orchestrator.js

@@ -690,12 +690,20 @@ Be specific about what you want done.`,
         const maxRetries = 2;
         const id = `worker_${Date.now()}_${++this.jobCounter}`;
         const cwd = workingDir || this.workspaceDir;
-        // Build prompt for worker
+        // Build prompt for worker with checkpoint instructions
         let prompt = task;
         if (context) {
             prompt = `Context: ${context}\n\nTask: ${task}`;
         }
-        prompt += '\n\nKeep your response concise and focused on results.';
+        // Add checkpoint instructions to prevent data loss on timeout
+        prompt += `
+
+IMPORTANT - Progress Saving:
+- Output your findings as you go, don't wait until the end
+- Print discoveries, file paths, and insights immediately as you find them
+- If you're exploring multiple files, report on each one before moving to the next
+- If you hit a timeout, the orchestrator should still have your partial findings
+- Keep responses concise but don't batch output until the very end`;
         console.log(`[ORCHESTRATOR] Delegating to worker ${id}: ${task.slice(0, 80)}...`);
         return new Promise((resolve) => {
             const job = {
@@ -722,10 +730,20 @@ Be specific about what you want done.`,
                 if (job.status === 'running') {
                     job.status = 'timeout';
                     job.endTime = Date.now();
-                    child.kill('SIGTERM');
+                    // Send SIGINT first to allow graceful shutdown and output flush
+                    child.kill('SIGINT');
+                    // Give 5 seconds for graceful shutdown before SIGTERM
+                    setTimeout(() => {
+                        if (!child.killed) {
+                            child.kill('SIGTERM');
+                        }
+                    }, 5000);
+                    const partialOutput = job.output.trim();
                     resolve({
                         success: false,
-                        output: `Worker timed out after 5 minutes.\nPartial output:\n${job.output.slice(-1000)}`
+                        output: partialOutput
+                            ? `Worker timed out after 5 minutes. Partial findings:\n${partialOutput.slice(-2000)}`
+                            : 'Worker timed out after 5 minutes with no output. The task may be too complex - try breaking it into smaller steps.'
                     });
                 }
             }, 5 * 60 * 1000);

package/dist/websocket.d.ts

@@ -9,7 +9,7 @@
 export type MessageHandler = (message: WebSocketMessage) => Promise<void>;
 export type StreamHandler = (chunk: string) => void;
 export interface WebSocketMessage {
-    type: 'message' | 'interrupt' | 'ping' | 'connected';
+    type: 'message' | 'interrupt' | 'ping' | 'pong' | 'connected';
     id?: string;
     content?: string;
     timestamp?: number;

package/dist/websocket.js

@@ -184,6 +184,9 @@ export class AgentWebSocket {
                 case 'ping':
                     this.sendToServer({ type: 'pong' });
                     break;
+                case 'pong':
+                    // Expected response to our ping, ignore silently
+                    break;
                 case 'connected':
                     console.log('[WS] Server confirmed connection');
                     break;

package/dist/workers/manager.js

@@ -70,8 +70,16 @@ export class WorkerManager {
         this.saveJob(job);
         // Spawn Claude CLI process
         try {
+            // Add checkpoint instructions to prevent data loss on timeout
+            const enhancedTask = `${task}
+
+IMPORTANT - Progress Saving:
+- Output findings as you go, don't wait until the end
+- Print discoveries and insights immediately as you find them
+- Report on each file/step before moving to the next
+- Keep responses concise but don't batch output`;
             // Escape single quotes in task for shell safety
-            const escapedTask = task.replace(/'/g, "'\\''");
+            const escapedTask = enhancedTask.replace(/'/g, "'\\''");
             // Spawn using bash -l -c to ensure proper PATH resolution (NVM, etc.)
             // This is more reliable than shell: true which has quote escaping issues
             const child = spawn('bash', ['-l', '-c', `claude -p '${escapedTask}' --output-format text --dangerously-skip-permissions`], {
@@ -101,11 +109,18 @@ export class WorkerManager {
             // Set up timeout
             const timeoutHandle = setTimeout(() => {
                 if (child.pid && !child.killed) {
-                    child.kill('SIGTERM');
+                    // Send SIGINT first to allow graceful shutdown and output flush
+                    child.kill('SIGINT');
+                    // Give 5 seconds for graceful shutdown before SIGTERM
+                    setTimeout(() => {
+                        if (!child.killed) {
+                            child.kill('SIGTERM');
+                        }
+                    }, 5000);
                     job.status = 'timeout';
                     job.error = `Job timed out after ${effectiveTimeout}ms`;
                     job.completed = new Date().toISOString();
-                    job.result = stdout || stderr;
+                    job.result = stdout.trim() || stderr.trim() || 'No output captured before timeout';
                     this.saveJob(job);
                     this.activeWorkers.delete(jobId);
                 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@siftd/connect-agent",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "description": "Master orchestrator agent - control Claude Code remotely via web",
   "type": "module",
   "main": "dist/index.js",