sparkecoder 0.1.67 → 0.1.68

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';
+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-Dm6wGcYv.js';
 import '../search-CCffrVJE.js';
 import 'drizzle-orm/sqlite-core';
 import 'zod';

package/dist/agent/index.js

@@ -5922,6 +5922,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 +6180,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 +6193,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 +6214,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 +6241,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 +6255,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 +6272,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 +6412,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);