@posthog/agent 2.3.207 → 2.3.213

package/dist/server/bin.cjs

@@ -904,7 +904,7 @@ var import_hono = require("hono");
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.207",
+  version: "2.3.213",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -4354,6 +4354,14 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
             if (message.subtype === "compact_boundary") {
               lastAssistantTotalUsage = 0;
               promptReplayed = true;
+              await this.client.sessionUpdate({
+                sessionId: params.sessionId,
+                update: {
+                  sessionUpdate: "usage_update",
+                  used: 0,
+                  size: lastContextWindowSize
+                }
+              });
             }
             if (message.subtype === "local_command_output") {
               promptReplayed = true;
@@ -4469,7 +4477,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
             }
             if ("usage" in message.message && message.parent_tool_use_id === null) {
               const usage = message.message.usage;
-              lastAssistantTotalUsage = usage.input_tokens + usage.cache_read_input_tokens + usage.cache_creation_input_tokens;
+              lastAssistantTotalUsage = usage.input_tokens + usage.output_tokens + usage.cache_read_input_tokens + usage.cache_creation_input_tokens;
               await this.client.sessionUpdate({
                 sessionId: params.sessionId,
                 update: {
@@ -5970,6 +5978,27 @@ var PostHogAPIClient = class {
 };
 
 // ../shared/dist/index.js
+var CLOUD_PROMPT_PREFIX = "__twig_cloud_prompt_v1__:";
+function deserializeCloudPrompt(value) {
+  const trimmed2 = value.trim();
+  if (!trimmed2) {
+    return [];
+  }
+  if (!trimmed2.startsWith(CLOUD_PROMPT_PREFIX)) {
+    return [{ type: "text", text: trimmed2 }];
+  }
+  try {
+    const parsed = JSON.parse(trimmed2.slice(CLOUD_PROMPT_PREFIX.length));
+    if (Array.isArray(parsed.blocks) && parsed.blocks.length > 0) {
+      return parsed.blocks;
+    }
+  } catch {
+  }
+  return [{ type: "text", text: trimmed2 }];
+}
+function promptBlocksToText(blocks) {
+  return blocks.filter((b) => b.type === "text").map((block) => block.text).join("").trim();
+}
 var consoleLogger = {
   info: (_message, _data) => {
   },
@@ -12065,6 +12094,14 @@ var AsyncMutex = class {
   }
 };
 
+// src/server/cloud-prompt.ts
+function normalizeCloudPromptContent(content) {
+  if (typeof content === "string") {
+    return deserializeCloudPrompt(content);
+  }
+  return content;
+}
+
 // src/server/jwt.ts
 var import_jsonwebtoken = __toESM(require("jsonwebtoken"), 1);
 var import_zod2 = require("zod");
@@ -12148,7 +12185,10 @@ var jsonRpcRequestSchema = import_v4.z.object({
   id: import_v4.z.union([import_v4.z.string(), import_v4.z.number()]).optional()
 });
 var userMessageParamsSchema = import_v4.z.object({
-  content: import_v4.z.string().min(1, "Content is required")
+  content: import_v4.z.union([
+    import_v4.z.string().min(1, "Content is required"),
+    import_v4.z.array(import_v4.z.record(import_v4.z.string(), import_v4.z.unknown())).min(1, "Content is required")
+  ])
 });
 var commandParamsSchemas = {
   user_message: userMessageParamsSchema,
@@ -12529,14 +12569,17 @@ var AgentServer = class _AgentServer {
     switch (method) {
       case POSTHOG_NOTIFICATIONS.USER_MESSAGE:
       case "user_message": {
-        const content = params.content;
+        const prompt = normalizeCloudPromptContent(
+          params.content
+        );
+        const promptPreview = promptBlocksToText(prompt);
         this.logger.info(
-          `Processing user message (detectedPrUrl=${this.detectedPrUrl ?? "none"}): ${content.substring(0, 100)}...`
+          `Processing user message (detectedPrUrl=${this.detectedPrUrl ?? "none"}): ${promptPreview.substring(0, 100)}...`
         );
         this.session.logWriter.resetTurnMessages(this.session.payload.run_id);
         const result = await this.session.clientConnection.prompt({
           sessionId: this.session.acpSessionId,
-          prompt: [{ type: "text", text: content }],
+          prompt,
           ...this.detectedPrUrl && {
             _meta: {
               prContext: `IMPORTANT \u2014 OVERRIDE PREVIOUS INSTRUCTIONS ABOUT CREATING BRANCHES/PRs.
@@ -12796,20 +12839,29 @@ You MUST NOT create a new branch, close the existing PR, or create a new PR.`
     try {
       const task = await this.posthogAPI.getTask(payload.task_id);
       const initialPromptOverride = taskRun ? this.getInitialPromptOverride(taskRun) : null;
-      const initialPrompt = initialPromptOverride ?? task.description;
-      if (!initialPrompt) {
+      const pendingUserPrompt = this.getPendingUserPrompt(taskRun);
+      let initialPrompt = [];
+      if (pendingUserPrompt?.length) {
+        initialPrompt = pendingUserPrompt;
+      } else if (initialPromptOverride) {
+        initialPrompt = [{ type: "text", text: initialPromptOverride }];
+      } else if (task.description) {
+        initialPrompt = [{ type: "text", text: task.description }];
+      }
+      if (initialPrompt.length === 0) {
         this.logger.warn("Task has no description, skipping initial message");
         return;
       }
       this.logger.info("Sending initial task message", {
         taskId: payload.task_id,
-        descriptionLength: initialPrompt.length,
-        usedInitialPromptOverride: !!initialPromptOverride
+        descriptionLength: promptBlocksToText(initialPrompt).length,
+        usedInitialPromptOverride: !!initialPromptOverride,
+        usedPendingUserMessage: !!pendingUserPrompt?.length
       });
       this.session.logWriter.resetTurnMessages(payload.run_id);
       const result = await this.session.clientConnection.prompt({
         sessionId: this.session.acpSessionId,
-        prompt: [{ type: "text", text: initialPrompt }]
+        prompt: initialPrompt
       });
       this.logger.info("Initial task message completed", {
         stopReason: result.stopReason
@@ -12832,11 +12884,14 @@ You MUST NOT create a new branch, close the existing PR, or create a new PR.`
       const conversationSummary = this.formatConversationForResume(
         this.resumeState.conversation
       );
-      const pendingUserMessage = this.getPendingUserMessage(taskRun);
+      const pendingUserPrompt = this.getPendingUserPrompt(taskRun);
       const sandboxContext = this.resumeState.snapshotApplied ? `The workspace environment (all files, packages, and code changes) has been fully restored from where you left off.` : `The workspace files from the previous session were not restored (the file snapshot may have expired), so you are starting with a fresh environment. Your conversation history is fully preserved below.`;
-      let resumePrompt;
-      if (pendingUserMessage) {
-        resumePrompt = `You are resuming a previous conversation. ${sandboxContext}
+      let resumePromptBlocks;
+      if (pendingUserPrompt?.length) {
+        resumePromptBlocks = [
+          {
+            type: "text",
+            text: `You are resuming a previous conversation. ${sandboxContext}
 
 Here is the conversation history from the previous session:
 
@@ -12844,30 +12899,40 @@ ${conversationSummary}
 
 The user has sent a new message:
 
-${pendingUserMessage}
-
-Respond to the user's new message above. You have full context from the previous session.`;
+`
+          },
+          ...pendingUserPrompt,
+          {
+            type: "text",
+            text: "\n\nRespond to the user's new message above. You have full context from the previous session."
+          }
+        ];
       } else {
-        resumePrompt = `You are resuming a previous conversation. ${sandboxContext}
+        resumePromptBlocks = [
+          {
+            type: "text",
+            text: `You are resuming a previous conversation. ${sandboxContext}
 
 Here is the conversation history from the previous session:
 
 ${conversationSummary}
 
-Continue from where you left off. The user is waiting for your response.`;
+Continue from where you left off. The user is waiting for your response.`
+          }
+        ];
       }
       this.logger.info("Sending resume message", {
         taskId: payload.task_id,
         conversationTurns: this.resumeState.conversation.length,
-        promptLength: resumePrompt.length,
-        hasPendingUserMessage: !!pendingUserMessage,
+        promptLength: promptBlocksToText(resumePromptBlocks).length,
+        hasPendingUserMessage: !!pendingUserPrompt?.length,
         snapshotApplied: this.resumeState.snapshotApplied
       });
       this.resumeState = null;
       this.session.logWriter.resetTurnMessages(payload.run_id);
       const result = await this.session.clientConnection.prompt({
         sessionId: this.session.acpSessionId,
-        prompt: [{ type: "text", text: resumePrompt }]
+        prompt: resumePromptBlocks
       });
       this.logger.info("Resume message completed", {
         stopReason: result.stopReason
@@ -12927,15 +12992,15 @@ ${toolSummary}`);
     const trimmed2 = override.trim();
     return trimmed2.length > 0 ? trimmed2 : null;
   }
-  getPendingUserMessage(taskRun) {
+  getPendingUserPrompt(taskRun) {
     if (!taskRun) return null;
     const state = taskRun.state;
     const message = state?.pending_user_message;
     if (typeof message !== "string") {
       return null;
     }
-    const trimmed2 = message.trim();
-    return trimmed2.length > 0 ? trimmed2 : null;
+    const prompt = deserializeCloudPrompt(message);
+    return prompt.length > 0 ? prompt : null;
   }
   getResumeRunId(taskRun) {
     const envRunId = process.env.POSTHOG_RESUME_RUN_ID;