@posthog/agent 2.3.353 → 2.3.356

package/dist/server/agent-server.js

@@ -8599,11 +8599,12 @@ async function getHeadSha(baseDir, options) {
 
 // src/server/agent-server.ts
 import { Hono } from "hono";
+import { z as z4 } from "zod";
 
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.353",
+  version: "2.3.356",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -8764,6 +8765,8 @@ var POSTHOG_NOTIFICATIONS = {
   CLOSE: "_posthog/close",
   /** Agent status update (thinking, working, etc.) */
   STATUS: "_posthog/status",
+  /** Structured backend progress notification; events in the same turn group into one card on the client */
+  PROGRESS: "_posthog/progress",
   /** Task-level notification (progress, milestones) */
   TASK_NOTIFICATION: "_posthog/task_notification",
   /** Marks a boundary for log compaction */
@@ -13764,6 +13767,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) {
@@ -13779,9 +13793,13 @@ function handleResultMessage(message) {
         return { shouldStop: true, stopReason: "max_tokens", usage };
       }
       if (message.is_error) {
+        const classification = classifyAgentError(message.result);
         return {
           shouldStop: true,
-          error: RequestError.internalError(void 0, message.result),
+          error: RequestError.internalError(
+            { classification, result: message.result },
+            message.result
+          ),
           usage
         };
       }
@@ -19106,6 +19124,14 @@ function validateCommandParams(method, params) {
 }
 
 // src/server/agent-server.ts
+var agentErrorClassificationSchema = z4.enum([
+  "upstream_stream_terminated",
+  "upstream_connection_error",
+  "agent_error"
+]);
+var errorWithClassificationSchema = z4.object({
+  data: z4.object({ classification: agentErrorClassificationSchema })
+});
 var NdJsonTap = class {
   constructor(onMessage) {
     this.onMessage = onMessage;
@@ -19405,7 +19431,9 @@ var AgentServer = class _AgentServer {
           port: this.config.port
         },
         () => {
-          this.logger.info(`HTTP server listening on port ${this.config.port}`);
+          this.logger.debug(
+            `HTTP server listening on port ${this.config.port}`
+          );
           resolve4();
         }
       );
@@ -19414,10 +19442,10 @@ var AgentServer = class _AgentServer {
   }
   async autoInitializeSession() {
     const { taskId, runId, mode, projectId } = this.config;
-    this.logger.info("Auto-initializing session", { taskId, runId, mode });
+    this.logger.debug("Auto-initializing session", { taskId, runId, mode });
     const resumeRunId = process.env.POSTHOG_RESUME_RUN_ID;
     if (resumeRunId) {
-      this.logger.info("Resuming from previous run", {
+      this.logger.debug("Resuming from previous run", {
         resumeRunId,
         currentRunId: runId
       });
@@ -19429,13 +19457,13 @@ var AgentServer = class _AgentServer {
           apiClient: this.posthogAPI,
           logger: new Logger({ debug: true, prefix: "[Resume]" })
         });
-        this.logger.info("Resume state loaded", {
+        this.logger.debug("Resume state loaded", {
           conversationTurns: this.resumeState.conversation.length,
           snapshotApplied: this.resumeState.snapshotApplied,
           logEntries: this.resumeState.logEntryCount
         });
       } catch (error) {
-        this.logger.warn("Failed to load resume state, starting fresh", {
+        this.logger.debug("Failed to load resume state, starting fresh", {
           error
         });
         this.resumeState = null;
@@ -19453,7 +19481,7 @@ var AgentServer = class _AgentServer {
     await this.initializeSession(payload, null);
   }
   async stop() {
-    this.logger.info("Stopping agent server...");
+    this.logger.debug("Stopping agent server...");
     if (this.session) {
       await this.cleanupSession();
     }
@@ -19461,7 +19489,7 @@ var AgentServer = class _AgentServer {
       this.server.close();
       this.server = null;
     }
-    this.logger.info("Agent server stopped");
+    this.logger.debug("Agent server stopped");
   }
   authenticateRequest(getHeader) {
     if (!this.config.jwtPublicKey) {
@@ -19504,7 +19532,7 @@ var AgentServer = class _AgentServer {
           blockTypes: prompt.map((block) => block.type)
         });
         const promptPreview = promptBlocksToText(prompt);
-        this.logger.info(
+        this.logger.debug(
           `Processing user message (detectedPrUrl=${this.detectedPrUrl ?? "none"}): ${promptPreview.substring(0, 100)}...`
         );
         this.session.logWriter.resetTurnMessages(this.session.payload.run_id);
@@ -19519,7 +19547,7 @@ var AgentServer = class _AgentServer {
             }
           }
         });
-        this.logger.info("User message completed", {
+        this.logger.debug("User message completed", {
           stopReason: result.stopReason
         });
         if (result.stopReason === "end_turn") {
@@ -19528,7 +19556,7 @@ var AgentServer = class _AgentServer {
         this.broadcastTurnComplete(result.stopReason);
         if (result.stopReason === "end_turn") {
           this.relayAgentResponse(this.session.payload).catch(
-            (err2) => this.logger.warn("Failed to relay follow-up response", err2)
+            (err2) => this.logger.debug("Failed to relay follow-up response", err2)
           );
         }
         let assistantMessage;
@@ -19540,7 +19568,7 @@ var AgentServer = class _AgentServer {
             this.session.payload.run_id
           );
         } catch {
-          this.logger.warn("Failed to extract assistant message from logs");
+          this.logger.debug("Failed to extract assistant message from logs");
         }
         return {
           stopReason: result.stopReason,
@@ -19549,7 +19577,7 @@ var AgentServer = class _AgentServer {
       }
       case POSTHOG_NOTIFICATIONS.CANCEL:
       case "cancel": {
-        this.logger.info("Cancel requested", {
+        this.logger.debug("Cancel requested", {
           acpSessionId: this.session.acpSessionId
         });
         await this.session.clientConnection.cancel({
@@ -19559,7 +19587,7 @@ var AgentServer = class _AgentServer {
       }
       case POSTHOG_NOTIFICATIONS.CLOSE:
       case "close": {
-        this.logger.info("Close requested");
+        this.logger.debug("Close requested");
         await this.cleanupSession();
         return { closed: true };
       }
@@ -19567,7 +19595,7 @@ var AgentServer = class _AgentServer {
       case "set_config_option": {
         const configId = params.configId;
         const value = params.value;
-        this.logger.info("Set config option requested", { configId, value });
+        this.logger.debug("Set config option requested", { configId, value });
         const result = await this.session.clientConnection.setSessionConfigOption({
           sessionId: this.session.acpSessionId,
           configId,
@@ -19595,7 +19623,7 @@ var AgentServer = class _AgentServer {
         const optionId = params.optionId;
         const customInput = params.customInput;
         const answers = params.answers;
-        this.logger.info("Permission response received", {
+        this.logger.debug("Permission response received", {
           requestId,
           optionId
         });
@@ -19618,7 +19646,7 @@ var AgentServer = class _AgentServer {
   }
   async initializeSession(payload, sseController) {
     if (this.initializationPromise) {
-      this.logger.info("Waiting for in-progress initialization", {
+      this.logger.debug("Waiting for in-progress initialization", {
         runId: payload.run_id
       });
       await this.initializationPromise;
@@ -19643,7 +19671,7 @@ var AgentServer = class _AgentServer {
     if (this.session) {
       await this.cleanupSession();
     }
-    this.logger.info("Initializing session", {
+    this.logger.debug("Initializing session", {
       runId: payload.run_id,
       taskId: payload.task_id
     });
@@ -19654,7 +19682,7 @@ var AgentServer = class _AgentServer {
     this.configureEnvironment();
     const [preTaskRun, preTask] = await Promise.all([
       this.posthogAPI.getTaskRun(payload.task_id, payload.run_id).catch((err2) => {
-        this.logger.warn("Failed to fetch task run for session context", {
+        this.logger.debug("Failed to fetch task run for session context", {
           taskId: payload.task_id,
           runId: payload.run_id,
           error: err2
@@ -19662,7 +19690,7 @@ var AgentServer = class _AgentServer {
         return null;
       }),
       this.posthogAPI.getTask(payload.task_id).catch((err2) => {
-        this.logger.warn("Failed to fetch task for session context", {
+        this.logger.debug("Failed to fetch task for session context", {
           taskId: payload.task_id,
           error: err2
         });
@@ -19772,7 +19800,7 @@ var AgentServer = class _AgentServer {
       }
     });
     const acpSessionId = sessionResponse.sessionId;
-    this.logger.info("ACP session created", {
+    this.logger.debug("ACP session created", {
       acpSessionId,
       runId: payload.run_id
     });
@@ -19792,22 +19820,38 @@ var AgentServer = class _AgentServer {
       debug: true,
       prefix: "[AgentServer]",
       onLog: (level, scope, message, data) => {
-        const _formatted = data !== void 0 ? `${message} ${JSON.stringify(data)}` : message;
         this.emitConsoleLog(level, scope, message, data);
       }
     });
-    this.logger.info("Session initialized successfully");
-    this.logger.info(
+    this.logger.debug("Session initialized successfully");
+    this.logger.debug(
       `Agent version: ${this.config.version ?? package_default.version}`
     );
-    this.logger.info(`Initial permission mode: ${initialPermissionMode}`);
+    this.logger.debug(`Initial permission mode: ${initialPermissionMode}`);
     this.posthogAPI.updateTaskRun(payload.task_id, payload.run_id, {
       status: "in_progress"
     }).catch(
-      (err2) => this.logger.warn("Failed to set task run to in_progress", err2)
+      (err2) => this.logger.debug("Failed to set task run to in_progress", err2)
     );
     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;
@@ -19818,7 +19862,7 @@ var AgentServer = class _AgentServer {
           payload.run_id
         );
       } catch (error) {
-        this.logger.warn("Failed to fetch task run", {
+        this.logger.debug("Failed to fetch task run", {
           taskId: payload.task_id,
           runId: payload.run_id,
           error
@@ -19828,7 +19872,7 @@ var AgentServer = class _AgentServer {
     if (!this.resumeState) {
       const resumeRunId = this.getResumeRunId(taskRun);
       if (resumeRunId) {
-        this.logger.info("Resuming from previous run (via TaskRun state)", {
+        this.logger.debug("Resuming from previous run (via TaskRun state)", {
           resumeRunId,
           currentRunId: payload.run_id
         });
@@ -19840,13 +19884,13 @@ var AgentServer = class _AgentServer {
             apiClient: this.posthogAPI,
             logger: new Logger({ debug: true, prefix: "[Resume]" })
           });
-          this.logger.info("Resume state loaded (via TaskRun state)", {
+          this.logger.debug("Resume state loaded (via TaskRun state)", {
             conversationTurns: this.resumeState.conversation.length,
             snapshotApplied: this.resumeState.snapshotApplied,
             logEntries: this.resumeState.logEntryCount
           });
         } catch (error) {
-          this.logger.warn("Failed to load resume state, starting fresh", {
+          this.logger.debug("Failed to load resume state, starting fresh", {
             error
           });
           this.resumeState = null;
@@ -19870,10 +19914,10 @@ var AgentServer = class _AgentServer {
         initialPrompt = [{ type: "text", text: task.description }];
       }
       if (initialPrompt.length === 0) {
-        this.logger.warn("Task has no description, skipping initial message");
+        this.logger.debug("Task has no description, skipping initial message");
         return;
       }
-      this.logger.info("Sending initial task message", {
+      this.logger.debug("Sending initial task message", {
         taskId: payload.task_id,
         descriptionLength: promptBlocksToText(initialPrompt).length,
         usedInitialPromptOverride: !!initialPromptOverride,
@@ -19884,7 +19928,7 @@ var AgentServer = class _AgentServer {
         sessionId: this.session.acpSessionId,
         prompt: initialPrompt
       });
-      this.logger.info("Initial task message completed", {
+      this.logger.debug("Initial task message completed", {
         stopReason: result.stopReason
       });
       await this.clearPendingInitialPromptState(payload, taskRun);
@@ -19900,7 +19944,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) {
@@ -19946,7 +19990,7 @@ Continue from where you left off. The user is waiting for your response.`
           }
         ];
       }
-      this.logger.info("Sending resume message", {
+      this.logger.debug("Sending resume message", {
         taskId: payload.task_id,
         conversationTurns: this.resumeState.conversation.length,
         promptLength: promptBlocksToText(resumePromptBlocks).length,
@@ -19959,7 +20003,7 @@ Continue from where you left off. The user is waiting for your response.`
         sessionId: this.session.acpSessionId,
         prompt: resumePromptBlocks
       });
-      this.logger.info("Resume message completed", {
+      this.logger.debug("Resume message completed", {
         stopReason: result.stopReason
       });
       if (result.stopReason === "end_turn") {
@@ -19974,7 +20018,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;
@@ -20305,7 +20349,7 @@ ${attributionInstructions}
     try {
       return await getCurrentBranch(this.config.repositoryPath);
     } catch (error) {
-      this.logger.warn("Failed to determine current git branch", {
+      this.logger.debug("Failed to determine current git branch", {
         repositoryPath: this.config.repositoryPath,
         error
       });
@@ -20324,7 +20368,7 @@ ${attributionInstructions}
       });
       this.lastReportedBranch = branchName;
     } catch (error) {
-      this.logger.warn("Failed to attach current branch to task run", {
+      this.logger.debug("Failed to attach current branch to task run", {
         taskId: payload.task_id,
         runId: payload.run_id,
         branchName,
@@ -20332,14 +20376,14 @@ ${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, {
           coalesce: true
         });
       } catch (error) {
-        this.logger.warn("Failed to flush session logs before completion", {
+        this.logger.debug("Failed to flush session logs before completion", {
           taskId: payload.task_id,
           runId: payload.run_id,
           error
@@ -20347,7 +20391,7 @@ ${attributionInstructions}
       }
     }
     if (stopReason !== "error") {
-      this.logger.info("Skipping status update for non-error stop reason", {
+      this.logger.debug("Skipping status update for non-error stop reason", {
         stopReason
       });
       return;
@@ -20356,9 +20400,9 @@ ${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 });
+      this.logger.debug("Task completion signaled", { status, stopReason });
     } catch (error) {
       this.logger.error("Failed to signal task completion", error);
     }
@@ -20437,7 +20481,7 @@ ${attributionInstructions}
           const sessionPermissionMode = this.getSessionPermissionMode();
           const needsDesktopApproval = isQuestion || this.shouldRelayPermissionToClient(sessionPermissionMode);
           if (isPlanApproval || needsDesktopApproval && this.session?.hasDesktopConnected) {
-            this.logger.info("Relaying permission request", {
+            this.logger.debug("Relaying permission request", {
               kind: params.toolCall?.kind,
               isQuestion,
               hasDesktopConnected: this.session?.hasDesktopConnected ?? false,
@@ -20467,7 +20511,7 @@ ${attributionInstructions}
       sessionUpdate: async (params) => {
         if (params.update?.sessionUpdate === "current_mode_update" && typeof params.update?.currentModeId === "string" && this.session) {
           this.session.permissionMode = params.update.currentModeId;
-          this.logger.info("Permission mode updated", {
+          this.logger.debug("Permission mode updated", {
             mode: params.update.currentModeId
           });
         }
@@ -20496,7 +20540,7 @@ ${attributionInstructions}
     try {
       await this.session.logWriter.flush(payload.run_id, { coalesce: true });
     } catch (error) {
-      this.logger.warn("Failed to flush logs before Slack relay", {
+      this.logger.debug("Failed to flush logs before Slack relay", {
         taskId: payload.task_id,
         runId: payload.run_id,
         error
@@ -20504,7 +20548,7 @@ ${attributionInstructions}
     }
     const message = this.session.logWriter.getFullAgentResponse(payload.run_id);
     if (!message) {
-      this.logger.warn("No agent message found for Slack relay", {
+      this.logger.debug("No agent message found for Slack relay", {
         taskId: payload.task_id,
         runId: payload.run_id,
         sessionRegistered: this.session.logWriter.isRegistered(payload.run_id)
@@ -20518,7 +20562,7 @@ ${attributionInstructions}
         message
       );
     } catch (error) {
-      this.logger.warn("Failed to relay initial agent response to Slack", {
+      this.logger.debug("Failed to relay initial agent response to Slack", {
         taskId: payload.task_id,
         runId: payload.run_id,
         error
@@ -20545,7 +20589,7 @@ ${attributionInstructions}
     message += "\nReply in this thread with your choice.";
     this.questionRelayedToSlack = true;
     this.posthogAPI.relayMessage(payload.task_id, payload.run_id, message).catch(
-      (err2) => this.logger.warn("Failed to relay question to Slack", { err: err2 })
+      (err2) => this.logger.debug("Failed to relay question to Slack", { err: err2 })
     );
   }
   getFirstQuestionMeta(toolMeta2) {
@@ -20607,14 +20651,14 @@ ${attributionInstructions}
       if (!prUrlMatch) return;
       const prUrl = prUrlMatch[0];
       this.detectedPrUrl = prUrl;
-      this.logger.info("Detected PR URL in bash output", {
+      this.logger.debug("Detected PR URL in bash output", {
         runId: payload.run_id,
         prUrl
       });
       this.posthogAPI.updateTaskRun(payload.task_id, payload.run_id, {
         output: { pr_url: prUrl }
       }).then(() => {
-        this.logger.info("PR URL attached to task run", {
+        this.logger.debug("PR URL attached to task run", {
           taskId: payload.task_id,
           runId: payload.run_id,
           prUrl
@@ -20636,7 +20680,7 @@ ${attributionInstructions}
   }
   async cleanupSession() {
     if (!this.session) return;
-    this.logger.info("Cleaning up session");
+    this.logger.debug("Cleaning up session");
     try {
       await this.captureTreeState();
     } catch (error) {