@posthog/agent 2.1.156 → 2.1.167

package/dist/server/bin.cjs

@@ -904,7 +904,7 @@ var import_hono = require("hono");
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.1.156",
+  version: "2.1.167",
   repository: "https://github.com/PostHog/twig",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -992,7 +992,7 @@ var package_default = {
   },
   dependencies: {
     "@agentclientprotocol/sdk": "^0.14.0",
-    "@anthropic-ai/claude-agent-sdk": "0.2.63",
+    "@anthropic-ai/claude-agent-sdk": "0.2.68",
     "@anthropic-ai/sdk": "^0.78.0",
     "@hono/node-server": "^1.19.9",
     "@opentelemetry/api-logs": "^0.208.0",
@@ -2632,6 +2632,9 @@ function handleResultMessage(message) {
           usage
         };
       }
+      if (message.stop_reason === "max_tokens") {
+        return { shouldStop: true, stopReason: "max_tokens", usage };
+      }
       if (message.is_error) {
         return {
           shouldStop: true,
@@ -2642,6 +2645,9 @@ function handleResultMessage(message) {
       return { shouldStop: true, stopReason: "end_turn", usage };
     }
     case "error_during_execution":
+      if (message.stop_reason === "max_tokens") {
+        return { shouldStop: true, stopReason: "max_tokens", usage };
+      }
       if (message.is_error) {
         return {
           shouldStop: true,
@@ -4104,14 +4110,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
             };
             const result = handleResultMessage(message);
             if (result.error) throw result.error;
-            switch (message.subtype) {
-              case "error_max_budget_usd":
-              case "error_max_turns":
-              case "error_max_structured_output_retries":
-                return { stopReason: "max_turn_requests", usage };
-              default:
-                return { stopReason: "end_turn", usage };
-            }
+            return { stopReason: result.stopReason ?? "end_turn", usage };
           }
           case "stream_event":
             await handleStreamEvent(message, context);
@@ -4132,6 +4131,9 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
                 return { stopReason: "end_turn" };
               }
             }
+            if ("isReplay" in message && message.isReplay) {
+              break;
+            }
             if ("usage" in message.message && message.parent_tool_use_id === null) {
               const usage = message.message.usage;
               lastAssistantTotalUsage = usage.input_tokens + usage.output_tokens + usage.cache_read_input_tokens + usage.cache_creation_input_tokens;
@@ -4146,6 +4148,8 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
           case "tool_progress":
           case "auth_status":
           case "tool_use_summary":
+          case "prompt_suggestion":
+          case "rate_limit_event":
             break;
           default:
             unreachable(message, this.logger);
@@ -4153,6 +4157,22 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
         }
       }
       throw new Error("Session did not end in result");
+    } catch (error) {
+      if (error instanceof import_sdk2.RequestError || !(error instanceof Error)) {
+        throw error;
+      }
+      const msg = error.message;
+      if (msg.includes("ProcessTransport") || msg.includes("terminated process") || msg.includes("process exited with") || msg.includes("process terminated by signal") || msg.includes("Failed to write to process stdin")) {
+        this.logger.error(`Process died: ${msg}`, {
+          sessionId: this.sessionId
+        });
+        this.session.input.end();
+        throw import_sdk2.RequestError.internalError(
+          void 0,
+          "The Claude Agent process exited unexpectedly. Please start a new session."
+        );
+      }
+      throw error;
     } finally {
       if (!handedOff) {
         this.session.promptRunning = false;
@@ -4336,6 +4356,9 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
       }
     } catch (err) {
       settingsManager.dispose();
+      if (isResume && err instanceof Error && err.message === "Query closed before response received") {
+        throw import_sdk2.RequestError.resourceNotFound(sessionId);
+      }
       this.logger.error(
         isResume ? forkSession ? "Session fork failed" : "Session resumption failed" : "Session initialization failed",
         {
@@ -11552,7 +11575,7 @@ After completing the requested changes:
 1. Create a new branch with a descriptive name based on the work done
 2. Stage and commit all changes with a clear commit message
 3. Push the branch to origin
-4. Create a draft pull request using \`gh pr create --draft\` with a descriptive title and body
+4. Create a draft pull request using \`gh pr create --draft${this.config.baseBranch ? ` --base ${this.config.baseBranch}` : ""}\` with a descriptive title and body
 
 Important:
 - Always create the PR as a draft. Do not ask for confirmation.
@@ -11898,7 +11921,7 @@ program.name("agent-server").description("PostHog cloud agent server - runs in s
 ).requiredOption("--repositoryPath <path>", "Path to the repository").requiredOption("--taskId <id>", "Task ID").requiredOption("--runId <id>", "Task run ID").option(
   "--mcpServers <json>",
   "MCP servers config as JSON array (ACP McpServer[] format)"
-).action(async (options) => {
+).option("--baseBranch <branch>", "Base branch for PR creation").action(async (options) => {
   const envResult = envSchema.safeParse(process.env);
   if (!envResult.success) {
     const errors = envResult.error.issues.map((issue) => `  - ${issue.message}`).join("\n");
@@ -11938,7 +11961,8 @@ ${errors}`
     mode,
     taskId: options.taskId,
     runId: options.runId,
-    mcpServers
+    mcpServers,
+    baseBranch: options.baseBranch
   });
   process.on("SIGINT", async () => {
     await server.stop();