sparkecoder 0.1.67 → 0.1.69

package/dist/agent/index.d.ts

@@ -1,6 +1,6 @@
 import 'ai';
-import '../schema-XcP0dedO.js';
-export { A as Agent, a as AgentOptions, b as AgentRunOptions, c as AgentStreamResult, C as ContextManager, M as MessageAttachment, d as buildSystemPrompt, e as buildTaskPromptAddendum } from '../index-DHyVVhJY.js';
-import '../search-CCffrVJE.js';
+import '../schema-Bq4tID-f.js';
+export { A as Agent, a as AgentOptions, b as AgentRunOptions, c as AgentStreamResult, C as ContextManager, M as MessageAttachment, d as buildSystemPrompt, e as buildTaskPromptAddendum } from '../index-DqaHLgSC.js';
+import '../search-BRnGaIl-.js';
 import 'drizzle-orm/sqlite-core';
 import 'zod';

package/dist/agent/index.js

@@ -1417,10 +1417,15 @@ import WebSocket from "ws";
 import { EventEmitter } from "events";
 function getOrCreateProxy(sessionId, port) {
   const existing = activeProxies.get(sessionId);
-  if (existing) return existing;
+  if (existing) {
+    console.log(`[BROWSER-WS] Reusing existing proxy for session ${sessionId} (connected=${existing.connected})`);
+    return existing;
+  }
+  console.log(`[BROWSER-WS] Creating new proxy for session ${sessionId} on port ${port} (active proxies: ${activeProxies.size})`);
   const proxy = new BrowserStreamProxy(port);
   activeProxies.set(sessionId, proxy);
   proxy.on("close", () => {
+    console.log(`[BROWSER-WS] Proxy closed for session ${sessionId}, removing from registry`);
     activeProxies.delete(sessionId);
   });
   proxy.connect();
@@ -1432,8 +1437,11 @@ function getProxy(sessionId) {
 function destroyProxy(sessionId) {
   const proxy = activeProxies.get(sessionId);
   if (proxy) {
+    console.log(`[BROWSER-WS] destroyProxy() called for session ${sessionId}`);
     proxy.destroy();
     activeProxies.delete(sessionId);
+  } else {
+    console.log(`[BROWSER-WS] destroyProxy() called but no proxy exists for session ${sessionId}`);
   }
 }
 var RECONNECT_DELAY_MS, MAX_RECONNECT_ATTEMPTS, FRAME_THROTTLE_MS, BrowserStreamProxy, activeProxies;
@@ -1464,18 +1472,22 @@ var init_stream_proxy = __esm({
       }
       connect() {
         if (this.destroyed) return;
+        console.log(`[BROWSER-WS] connect() called for port ${this.port}`);
         this.doConnect();
       }
       doConnect() {
         if (this.destroyed) return;
         const url = `ws://localhost:${this.port}`;
+        console.log(`[BROWSER-WS] Attempting WebSocket connection to ${url} (attempt ${this.reconnectAttempts + 1}/${MAX_RECONNECT_ATTEMPTS})`);
         try {
           this.ws = new WebSocket(url);
-        } catch {
+        } catch (err) {
+          console.warn(`[BROWSER-WS] WebSocket constructor threw for ${url}:`, err);
           this.scheduleReconnect();
           return;
         }
         this.ws.on("open", () => {
+          console.log(`[BROWSER-WS] Connected to ${url} (after ${this.reconnectAttempts} retries)`);
           this.reconnectAttempts = 0;
           this._connected = true;
           this.emit("status", {
@@ -1487,12 +1499,14 @@ var init_stream_proxy = __esm({
           try {
             const msg = JSON.parse(typeof raw === "string" ? raw : raw.toString("utf8"));
             this.handleMessage(msg);
-          } catch {
+          } catch (err) {
+            console.warn(`[BROWSER-WS] Malformed message from ${url}:`, err);
           }
         });
-        this.ws.on("close", () => {
+        this.ws.on("close", (code, reason) => {
           const wasConnected = this._connected;
           this._connected = false;
+          console.log(`[BROWSER-WS] Connection closed: code=${code} reason="${reason?.toString() || ""}" wasConnected=${wasConnected} destroyed=${this.destroyed}`);
           if (wasConnected) {
             this.emit("status", { connected: false, screencasting: false });
           }
@@ -1500,14 +1514,26 @@ var init_stream_proxy = __esm({
             this.scheduleReconnect();
           }
         });
-        this.ws.on("error", () => {
+        this.ws.on("error", (err) => {
+          console.warn(`[BROWSER-WS] WebSocket error on port ${this.port}:`, err.message);
         });
       }
+      frameCount = 0;
+      throttledCount = 0;
+      lastFrameLogTime = 0;
       handleMessage(msg) {
         if (msg.type === "frame") {
           const now = Date.now();
-          if (now - this.lastFrameTime < FRAME_THROTTLE_MS) return;
+          if (now - this.lastFrameTime < FRAME_THROTTLE_MS) {
+            this.throttledCount++;
+            return;
+          }
           this.lastFrameTime = now;
+          this.frameCount++;
+          if (now - this.lastFrameLogTime > 5e3) {
+            console.log(`[BROWSER-WS] Frame stats: emitted=${this.frameCount} throttled=${this.throttledCount} listeners=${this.listenerCount("frame")} dataSize=${msg.data?.length ?? 0}`);
+            this.lastFrameLogTime = now;
+          }
           const frame = {
             data: msg.data,
             metadata: msg.metadata ?? {
@@ -1523,21 +1549,26 @@ var init_stream_proxy = __esm({
           this._latestFrame = frame;
           this.emit("frame", frame);
         } else if (msg.type === "status") {
+          console.log(`[BROWSER-WS] Status message received:`, JSON.stringify(msg));
           this.emit("status", {
             connected: msg.connected ?? true,
             screencasting: msg.screencasting ?? true,
             viewportWidth: msg.viewportWidth,
             viewportHeight: msg.viewportHeight
           });
+        } else {
+          console.log(`[BROWSER-WS] Unknown message type: ${msg.type}`);
         }
       }
       scheduleReconnect() {
         if (this.destroyed || this.reconnectAttempts >= MAX_RECONNECT_ATTEMPTS) {
+          console.log(`[BROWSER-WS] Giving up reconnection: destroyed=${this.destroyed} attempts=${this.reconnectAttempts}/${MAX_RECONNECT_ATTEMPTS}`);
           this.emit("close");
           return;
         }
         this.reconnectAttempts++;
         const delay = this.reconnectAttempts <= 5 ? RECONNECT_DELAY_MS : RECONNECT_DELAY_MS * (this.reconnectAttempts - 4);
+        console.log(`[BROWSER-WS] Scheduling reconnect in ${delay}ms (attempt ${this.reconnectAttempts}/${MAX_RECONNECT_ATTEMPTS})`);
         this.reconnectTimer = setTimeout(() => this.doConnect(), delay);
       }
       /**
@@ -1549,6 +1580,7 @@ var init_stream_proxy = __esm({
         }
       }
       destroy() {
+        console.log(`[BROWSER-WS] Destroying proxy for port ${this.port} (emitted ${this.frameCount} frames, throttled ${this.throttledCount})`);
         this.destroyed = true;
         if (this.reconnectTimer) {
           clearTimeout(this.reconnectTimer);
@@ -5922,6 +5954,21 @@ async function sendWebhook(url, event) {
 }
 
 // src/agent/index.ts
+var MAX_SSE_FIELD_LENGTH = 8 * 1024;
+var SSE_PREVIEW_LENGTH = 2 * 1024;
+function truncateWriteFileInput(input) {
+  const out = { ...input };
+  for (const key of ["content", "old_string", "new_string"]) {
+    const val = out[key];
+    if (typeof val === "string" && val.length > MAX_SSE_FIELD_LENGTH) {
+      out[key] = `${val.slice(0, SSE_PREVIEW_LENGTH)}
+... (truncated)`;
+      out[`${key}Truncated`] = true;
+      out[`${key}Length`] = val.length;
+    }
+  }
+  return out;
+}
 var approvalResolvers = /* @__PURE__ */ new Map();
 var Agent = class _Agent {
   session;
@@ -6165,8 +6212,11 @@ ${prompt}` });
     };
     let taskRecorder = null;
     const sessionId = this.session.id;
+    const emit = options.writeSSE;
     const bashProgressHandler = (progress) => {
       options.onToolProgress?.({ toolName: "bash", data: progress });
+      if (emit) emit(JSON.stringify({ type: "tool-progress", toolName: "bash", data: progress })).catch(() => {
+      });
       const port = progress.browserStreamPort;
       if (port && progress.status === "started") {
         Promise.resolve().then(() => (init_stream_proxy(), stream_proxy_exports)).then(({ getOrCreateProxy: getOrCreateProxy2 }) => {
@@ -6175,7 +6225,17 @@ ${prompt}` });
             Promise.resolve().then(() => (init_recorder(), recorder_exports)).then(({ FrameRecorder: FrameRecorder2 }) => {
               taskRecorder = new FrameRecorder2(sessionId);
               taskRecorder.start();
-              proxy.on("frame", (frame) => taskRecorder?.addFrame(frame));
+            });
+          }
+          if (proxy.listenerCount("frame") === 0) {
+            proxy.on("frame", (frame) => {
+              taskRecorder?.addFrame(frame);
+              if (emit) emit(JSON.stringify({ type: "browser-frame", data: frame.data, metadata: frame.metadata })).catch(() => {
+              });
+            });
+            proxy.on("status", (s) => {
+              if (emit) emit(JSON.stringify({ type: "browser-status", ...s })).catch(() => {
+              });
             });
           }
         });
@@ -6186,8 +6246,16 @@ ${prompt}` });
       workingDirectory: this.session.workingDirectory,
       skillsDirectories: config.resolvedSkillsDirectories,
       onBashProgress: bashProgressHandler,
-      onWriteFileProgress: options.onToolProgress ? (progress) => options.onToolProgress({ toolName: "write_file", data: progress }) : void 0,
-      onSearchProgress: options.onToolProgress ? (progress) => options.onToolProgress({ toolName: "explore_agent", data: progress }) : void 0,
+      onWriteFileProgress: (progress) => {
+        options.onToolProgress?.({ toolName: "write_file", data: progress });
+        if (emit) emit(JSON.stringify({ type: "tool-progress", toolName: "write_file", data: progress })).catch(() => {
+        });
+      },
+      onSearchProgress: (progress) => {
+        options.onToolProgress?.({ toolName: "explore_agent", data: progress });
+        if (emit) emit(JSON.stringify({ type: "tool-progress", toolName: "explore_agent", data: progress })).catch(() => {
+        });
+      },
       taskTools: {
         outputSchema: options.taskConfig.outputSchema,
         onComplete
@@ -6205,6 +6273,9 @@ ${prompt}` });
 
 ${taskAddendum}`;
     fireWebhook("task.started", { prompt: options.prompt });
+    if (emit) {
+      await emit(JSON.stringify({ type: "data-user-message", data: { id: `user_${Date.now()}`, content: options.prompt } }));
+    }
     await this.context.addUserMessage(options.prompt);
     let iteration = 0;
     while (iteration < maxIterations) {
@@ -6216,7 +6287,15 @@ ${taskAddendum}`;
       }
       const messages = await this.context.getMessages();
       const useAnthropic = isAnthropicModel(this.session.model);
-      const result = await generateText3({
+      if (emit) {
+        await emit(JSON.stringify({ type: "start", messageId: `msg_${Date.now()}` }));
+      }
+      let textStarted = false;
+      let textId = `text_${Date.now()}`;
+      let reasoningId = `reasoning_${Date.now()}`;
+      let reasoningStarted = false;
+      const toolCallStarts = /* @__PURE__ */ new Set();
+      const iterStream = streamText2({
         model: resolveModel(this.session.model),
         system: systemPrompt,
         messages,
@@ -6225,21 +6304,94 @@ ${taskAddendum}`;
         abortSignal: options.abortSignal,
         providerOptions: useAnthropic ? {
           anthropic: {
+            toolStreaming: true,
             thinking: { type: "enabled", budgetTokens: 1e4 }
           }
         } : void 0,
-        onStepFinish: (step) => {
+        onStepFinish: async (step) => {
           options.onStepFinish?.(step);
           fireWebhook("task.step_finished", { iteration, text: step.text });
+          if (emit) {
+            if (textStarted) {
+              await emit(JSON.stringify({ type: "text-end", id: textId }));
+              textStarted = false;
+              textId = `text_${Date.now()}`;
+            }
+            await emit(JSON.stringify({ type: "finish-step" }));
+          }
         }
       });
-      const responseMessages = result.response.messages;
+      for await (const part of iterStream.fullStream) {
+        if (part.type === "text-delta") {
+          if (emit) {
+            if (!textStarted) {
+              await emit(JSON.stringify({ type: "text-start", id: textId }));
+              textStarted = true;
+            }
+            await emit(JSON.stringify({ type: "text-delta", id: textId, delta: part.text }));
+          }
+        } else if (part.type === "reasoning-start") {
+          if (emit) {
+            await emit(JSON.stringify({ type: "reasoning-start", id: reasoningId }));
+            reasoningStarted = true;
+          }
+        } else if (part.type === "reasoning-delta") {
+          if (emit) {
+            await emit(JSON.stringify({ type: "reasoning-delta", id: reasoningId, delta: part.text }));
+          }
+        } else if (part.type === "reasoning-end") {
+          if (emit && reasoningStarted) {
+            await emit(JSON.stringify({ type: "reasoning-end", id: reasoningId }));
+            reasoningStarted = false;
+            reasoningId = `reasoning_${Date.now()}`;
+          }
+        } else if (part.type === "tool-call-streaming-start") {
+          if (emit) {
+            const p = part;
+            await emit(JSON.stringify({ type: "tool-input-start", toolCallId: p.toolCallId, toolName: p.toolName }));
+            toolCallStarts.add(p.toolCallId);
+          }
+        } else if (part.type === "tool-call-delta") {
+          if (emit) {
+            const p = part;
+            await emit(JSON.stringify({ type: "tool-input-delta", toolCallId: p.toolCallId, argsTextDelta: p.argsTextDelta }));
+          }
+        } else if (part.type === "tool-call") {
+          if (emit) {
+            if (!toolCallStarts.has(part.toolCallId)) {
+              await emit(JSON.stringify({ type: "tool-input-start", toolCallId: part.toolCallId, toolName: part.toolName }));
+              toolCallStarts.add(part.toolCallId);
+            }
+            const safeInput = part.toolName === "write_file" && part.input && typeof part.input === "object" ? truncateWriteFileInput(part.input) : part.input;
+            await emit(JSON.stringify({ type: "tool-input-available", toolCallId: part.toolCallId, toolName: part.toolName, input: safeInput }));
+          }
+        } else if (part.type === "tool-result") {
+          if (emit) {
+            await emit(JSON.stringify({ type: "tool-output-available", toolCallId: part.toolCallId, output: part.output }));
+          }
+        } else if (part.type === "error") {
+          console.error("Task stream error:", part.error);
+          if (emit) {
+            await emit(JSON.stringify({ type: "error", errorText: String(part.error) }));
+          }
+        }
+      }
+      if (emit && textStarted) {
+        await emit(JSON.stringify({ type: "text-end", id: textId }));
+      }
+      if (emit && reasoningStarted) {
+        await emit(JSON.stringify({ type: "reasoning-end", id: reasoningId }));
+      }
+      const iterResponse = await iterStream.response;
+      const responseMessages = iterResponse.messages;
       await this.context.addResponseMessages(responseMessages);
-      if (result.text) {
-        options.onText?.(result.text);
-        fireWebhook("task.message", { iteration, text: result.text });
+      const resultText = await iterStream.text;
+      const resultSteps = await iterStream.steps;
+      if (resultText) {
+        options.onText?.(resultText);
+        fireWebhook("task.message", { iteration, text: resultText });
       }
-      for (const step of result.steps) {
+      for (const step of resultSteps) {
         if (step.toolCalls) {
           for (const tc of step.toolCalls) {
             options.onToolCall?.({ toolCallId: tc.toolCallId, toolName: tc.toolName, input: tc.args });
@@ -6292,9 +6444,11 @@ ${taskAddendum}`;
           iterations: iteration
         };
       }
-      await this.context.addUserMessage(
-        "Continue working on the task. Before calling `complete_task`, VERIFY your work is correct \u2014 re-read edited files, run the linter, run tests if applicable, and check the browser/server if you made UI or API changes. Make sure you searched the right directories and found everything relevant. When fully verified, call `complete_task` with the result. If you cannot complete it, call `task_failed` with a reason."
-      );
+      const continuationPrompt = "Continue working on the task. Before calling `complete_task`, VERIFY your work is correct \u2014 re-read edited files, run the linter, run tests if applicable, and check the browser/server if you made UI or API changes. Make sure you searched the right directories and found everything relevant. When fully verified, call `complete_task` with the result. If you cannot complete it, call `task_failed` with a reason.";
+      if (emit) {
+        await emit(JSON.stringify({ type: "data-user-message", data: { id: `user_${Date.now()}`, content: continuationPrompt } }));
+      }
+      await this.context.addUserMessage(continuationPrompt);
     }
     const timeoutError = `Task did not complete within ${maxIterations} iterations`;
     const timeoutRecordingUrls = await this.finishTaskRecording(taskRecorder);