@posthog/agent 2.3.353 → 2.3.354

package/dist/server/bin.cjs

@@ -8723,11 +8723,12 @@ async function getHeadSha(baseDir, options) {
 
 // src/server/agent-server.ts
 var import_hono = require("hono");
+var import_zod3 = require("zod");
 
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.353",
+  version: "2.3.354",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -13883,6 +13884,17 @@ async function handleSystemMessage(message, context) {
       break;
   }
 }
+function classifyAgentError(result) {
+  if (!result) return "agent_error";
+  const text2 = result.trim();
+  if (/API Error:\s*terminated\b/i.test(text2)) {
+    return "upstream_stream_terminated";
+  }
+  if (/API Error:\s*Connection error\b/i.test(text2)) {
+    return "upstream_connection_error";
+  }
+  return "agent_error";
+}
 function handleResultMessage(message) {
   const usage = extractUsageFromResult(message);
   switch (message.subtype) {
@@ -13898,9 +13910,13 @@ function handleResultMessage(message) {
         return { shouldStop: true, stopReason: "max_tokens", usage };
       }
       if (message.is_error) {
+        const classification = classifyAgentError(message.result);
         return {
           shouldStop: true,
-          error: import_sdk.RequestError.internalError(void 0, message.result),
+          error: import_sdk.RequestError.internalError(
+            { classification, result: message.result },
+            message.result
+          ),
           usage
         };
       }
@@ -19114,6 +19130,14 @@ function validateCommandParams(method, params) {
 }
 
 // src/server/agent-server.ts
+var agentErrorClassificationSchema = import_zod3.z.enum([
+  "upstream_stream_terminated",
+  "upstream_connection_error",
+  "agent_error"
+]);
+var errorWithClassificationSchema = import_zod3.z.object({
+  data: import_zod3.z.object({ classification: agentErrorClassificationSchema })
+});
 var NdJsonTap = class {
   constructor(onMessage) {
     this.onMessage = onMessage;
@@ -19816,6 +19840,23 @@ var AgentServer = class _AgentServer {
     );
     await this.sendInitialTaskMessage(payload, preTaskRun);
   }
+  extractErrorClassification(error) {
+    const message = error instanceof Error ? error.message : String(error ?? "");
+    const parsed = errorWithClassificationSchema.safeParse(error);
+    if (parsed.success) {
+      return { classification: parsed.data.data.classification, message };
+    }
+    return { classification: classifyAgentError(message), message };
+  }
+  classifyAndSignalFailure(payload, phase, error) {
+    const { classification, message } = this.extractErrorClassification(error);
+    const errorMessage = classification === "upstream_stream_terminated" ? "Upstream LLM stream terminated" : classification === "upstream_connection_error" ? "Upstream LLM connection error" : message || "Agent error";
+    this.logger.error(`send_${phase}_task_message_failed`, {
+      classification,
+      message
+    });
+    return this.signalTaskComplete(payload, "error", errorMessage);
+  }
   async sendInitialTaskMessage(payload, prefetchedRun) {
     if (!this.session) return;
     let taskRun = prefetchedRun ?? null;
@@ -19908,7 +19949,7 @@ var AgentServer = class _AgentServer {
       if (this.session) {
         await this.session.logWriter.flushAll();
       }
-      await this.signalTaskComplete(payload, "error");
+      await this.classifyAndSignalFailure(payload, "initial", error);
     }
   }
   async sendResumeMessage(payload, taskRun) {
@@ -19982,7 +20023,7 @@ Continue from where you left off. The user is waiting for your response.`
       if (this.session) {
         await this.session.logWriter.flushAll();
       }
-      await this.signalTaskComplete(payload, "error");
+      await this.classifyAndSignalFailure(payload, "resume", error);
     }
   }
   static RESUME_HISTORY_TOKEN_BUDGET = 5e4;
@@ -20340,7 +20381,7 @@ ${attributionInstructions}
       });
     }
   }
-  async signalTaskComplete(payload, stopReason) {
+  async signalTaskComplete(payload, stopReason, errorMessage) {
     if (this.session?.payload.run_id === payload.run_id) {
       try {
         await this.session.logWriter.flush(payload.run_id, {
@@ -20364,7 +20405,7 @@ ${attributionInstructions}
     try {
       await this.posthogAPI.updateTaskRun(payload.task_id, payload.run_id, {
         status,
-        error_message: stopReason === "error" ? "Agent error" : void 0
+        error_message: errorMessage ?? "Agent error"
       });
       this.logger.info("Task completion signaled", { status, stopReason });
     } catch (error) {