@posthog/agent 2.1.62 → 2.1.69

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/agent",
-  "version": "2.1.62",
+  "version": "2.1.69",
   "repository": "https://github.com/PostHog/twig",
   "description": "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   "exports": {

package/src/adapters/claude/claude-agent.ts

@@ -33,7 +33,7 @@ import { v7 as uuidv7 } from "uuid";
 import packageJson from "../../../package.json" with { type: "json" };
 import type { SessionContext } from "../../otel-log-writer.js";
 import type { SessionLogWriter } from "../../session-log-writer.js";
-import { unreachable } from "../../utils/common.js";
+import { unreachable, withTimeout } from "../../utils/common.js";
 import { Logger } from "../../utils/logger.js";
 import { Pushable } from "../../utils/streams.js";
 import { BaseAcpAgent } from "../base-acp-agent.js";
@@ -48,7 +48,7 @@ import { fetchMcpToolMetadata } from "./mcp/tool-metadata.js";
 import { canUseTool } from "./permissions/permission-handlers.js";
 import { getAvailableSlashCommands } from "./session/commands.js";
 import { parseMcpServers } from "./session/mcp-config.js";
-import { toSdkModelId } from "./session/models.js";
+import { DEFAULT_MODEL, toSdkModelId } from "./session/models.js";
 import {
   buildSessionOptions,
   buildSystemPrompt,
@@ -66,6 +66,8 @@ import type {
   ToolUseCache,
 } from "./types.js";
 
+const SESSION_VALIDATION_TIMEOUT_MS = 10_000;
+
 export interface ClaudeAcpAgentOptions {
   onProcessSpawned?: (info: ProcessSpawnedInfo) => void;
   onProcessExited?: (pid: number) => void;
@@ -162,6 +164,8 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
     });
 
     const input = new Pushable<SDKUserMessage>();
+    // Pass default model at construction to avoid expensive post-hoc setModel IPC
+    options.model = DEFAULT_MODEL;
     const q = query({ prompt: input, options });
 
     const session = this.createSession(
@@ -191,7 +195,12 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
     this.deferBackgroundFetches(q, sessionId, mcpServers);
 
     session.modelId = modelOptions.currentModelId;
-    await this.trySetModel(q, modelOptions.currentModelId);
+    // Only call setModel if the resolved model differs from the default we
+    // already baked into the query options — avoids a ~2s IPC round-trip.
+    const resolvedSdkModel = toSdkModelId(modelOptions.currentModelId);
+    if (resolvedSdkModel !== DEFAULT_MODEL) {
+      await this.trySetModel(q, modelOptions.currentModelId);
+    }
 
     const configOptions = await this.buildConfigOptions(modelOptions);
 
@@ -240,6 +249,16 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
 
     this.registerPersistence(sessionId, meta as Record<string, unknown>);
 
+    // Validate the resumed session is alive. For stale sessions this throws
+    // (e.g. "No conversation found"), preventing a broken session.
+    const validation = await withTimeout(
+      q.initializationResult(),
+      SESSION_VALIDATION_TIMEOUT_MS,
+    );
+    if (validation.result === "timeout") {
+      throw new Error("Session validation timed out");
+    }
+
     // Deferred: slash commands + MCP metadata (not needed to return configOptions)
     this.deferBackgroundFetches(q, sessionId, mcpServers);
 

package/src/agent.ts

@@ -18,12 +18,10 @@ export class Agent {
   private acpConnection?: InProcessAcpConnection;
   private taskRunId?: string;
   private sessionLogWriter?: SessionLogWriter;
-  public debug: boolean;
 
   constructor(config: AgentConfig) {
-    this.debug = config.debug || false;
     this.logger = new Logger({
-      debug: this.debug,
+      debug: config.debug || false,
       prefix: "[PostHog Agent]",
       onLog: config.onLog,
     });

package/src/utils/common.ts

@@ -1,5 +1,24 @@
 import type { Logger } from "./logger.js";
 
+/**
+ * Races an operation against a timeout.
+ * Returns success with the value if the operation completes in time,
+ * or timeout if the operation takes longer than the specified duration.
+ */
+export async function withTimeout<T>(
+  operation: Promise<T>,
+  timeoutMs: number,
+): Promise<{ result: "success"; value: T } | { result: "timeout" }> {
+  const timeoutPromise = new Promise<{ result: "timeout" }>((resolve) =>
+    setTimeout(() => resolve({ result: "timeout" }), timeoutMs),
+  );
+  const operationPromise = operation.then((value) => ({
+    result: "success" as const,
+    value,
+  }));
+  return Promise.race([operationPromise, timeoutPromise]);
+}
+
 export const IS_ROOT =
   typeof process !== "undefined" &&
   (process.geteuid?.() ?? process.getuid?.()) === 0;