@posthog/agent 2.1.114 → 2.1.118

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.114",
+  version: "2.1.118",
   repository: "https://github.com/PostHog/twig",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -3353,12 +3353,10 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
   toolUseCache;
   backgroundTerminals = {};
   clientCapabilities;
-  logWriter;
   options;
   lastSentConfigOptions;
-  constructor(client, logWriter, options) {
+  constructor(client, options) {
     super(client);
-    this.logWriter = logWriter;
     this.options = options;
     this.toolUseCache = {};
     this.logger = new Logger({ debug: true, prefix: "[ClaudeAcpAgent]" });
@@ -3403,7 +3401,14 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
   async newSession(params) {
     this.checkAuthStatus();
     const meta = params._meta;
+    const taskId = meta?.persistence?.taskId;
     const sessionId = (0, import_uuid.v7)();
+    this.logger.info("Creating new session", {
+      sessionId,
+      taskId,
+      taskRunId: meta?.taskRunId,
+      cwd: params.cwd
+    });
     const permissionMode = meta?.permissionMode && TWIG_EXECUTION_MODES.includes(meta.permissionMode) ? meta.permissionMode : "default";
     const mcpServers = parseMcpServers(params);
     const options = buildSessionOptions({
@@ -3432,7 +3437,6 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
       options.abortController
     );
     session.taskRunId = meta?.taskRunId;
-    this.registerPersistence(sessionId, meta);
     if (meta?.taskRunId) {
       await this.client.extNotification("_posthog/sdk_session", {
         taskRunId: meta.taskRunId,
@@ -3458,6 +3462,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
   }
   async resumeSession(params) {
     const meta = params._meta;
+    const taskId = meta?.persistence?.taskId;
     const sessionId = meta?.sessionId;
     if (!sessionId) {
       throw new Error("Cannot resume session without sessionId");
@@ -3465,6 +3470,12 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
     if (this.sessionId === sessionId) {
       return {};
     }
+    this.logger.info("Resuming session", {
+      sessionId,
+      taskId,
+      taskRunId: meta?.taskRunId,
+      cwd: params.cwd
+    });
     const mcpServers = parseMcpServers(params);
     const permissionMode = meta?.permissionMode && TWIG_EXECUTION_MODES.includes(meta.permissionMode) ? meta.permissionMode : "default";
     const { query: q, session } = await this.initializeQuery({
@@ -3477,15 +3488,36 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
       isResume: true,
       additionalDirectories: meta?.claudeCode?.options?.additionalDirectories
     });
+    this.logger.info("Session query initialized, awaiting resumption", {
+      sessionId,
+      taskId,
+      taskRunId: meta?.taskRunId
+    });
     session.taskRunId = meta?.taskRunId;
-    this.registerPersistence(sessionId, meta);
-    const validation = await withTimeout(
-      q.initializationResult(),
-      SESSION_VALIDATION_TIMEOUT_MS
-    );
-    if (validation.result === "timeout") {
-      throw new Error("Session validation timed out");
+    try {
+      const result = await withTimeout(
+        q.initializationResult(),
+        SESSION_VALIDATION_TIMEOUT_MS
+      );
+      if (result.result === "timeout") {
+        throw new Error(
+          `Session resumption timed out for sessionId=${sessionId}`
+        );
+      }
+    } catch (err) {
+      this.logger.error("Session resumption failed", {
+        sessionId,
+        taskId,
+        taskRunId: meta?.taskRunId,
+        error: err instanceof Error ? err.message : String(err)
+      });
+      throw err;
     }
+    this.logger.info("Session resumed successfully", {
+      sessionId,
+      taskId,
+      taskRunId: meta?.taskRunId
+    });
     this.deferBackgroundFetches(q, sessionId);
     const configOptions = await this.buildConfigOptions();
     return { configOptions };
@@ -3684,12 +3716,6 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
       this.logger.warn("Failed to fetch deferred session data", { err });
     });
   }
-  registerPersistence(sessionId, meta) {
-    const persistence = meta?.persistence;
-    if (persistence && this.logWriter) {
-      this.logWriter.register(sessionId, persistence);
-    }
-  }
   sendAvailableCommandsUpdate(sessionId, availableCommands) {
     setTimeout(() => {
       this.client.sessionUpdate({
@@ -3977,7 +4003,7 @@ function createClaudeConnection(config) {
   const agentStream = (0, import_sdk3.ndJsonStream)(agentWritable, streams.agent.readable);
   let agent = null;
   const agentConnection = new import_sdk3.AgentSideConnection((client) => {
-    agent = new ClaudeAcpAgent(client, logWriter, config.processCallbacks);
+    agent = new ClaudeAcpAgent(client, config.processCallbacks);
     logger.info(`Created ${agent.adapterName} agent`);
     return agent;
   }, agentStream);
@@ -4221,17 +4247,17 @@ function createCodexConnection(config) {
 }
 
 // src/utils/gateway.ts
-function getLlmGatewayUrl(posthogHost) {
+function getLlmGatewayUrl(posthogHost, product = "twig") {
   const url = new URL(posthogHost);
   const hostname = url.hostname;
   if (hostname === "localhost" || hostname === "127.0.0.1") {
-    return `${url.protocol}//localhost:3308/twig`;
+    return `${url.protocol}//localhost:3308/${product}`;
   }
   if (hostname === "host.docker.internal") {
-    return `${url.protocol}//host.docker.internal:3308/twig`;
+    return `${url.protocol}//host.docker.internal:3308/${product}`;
   }
   const region = hostname.match(/^(us|eu)\.posthog\.com$/)?.[1] ?? "us";
-  return `https://gateway.${region}.posthog.com/twig`;
+  return `https://gateway.${region}.posthog.com/${product}`;
 }
 
 // src/posthog-api.ts
@@ -4417,11 +4443,15 @@ var SessionLogWriter = class _SessionLogWriter {
   }
   async flushAll() {
     const sessionIds = [...this.sessions.keys()];
-    const pendingCounts = sessionIds.map((id) => ({
-      id,
-      pending: this.pendingEntries.get(id)?.length ?? 0,
-      messages: this.messageCounts.get(id) ?? 0
-    }));
+    const pendingCounts = sessionIds.map((id) => {
+      const session = this.sessions.get(id);
+      return {
+        taskId: session?.context.taskId,
+        runId: session?.context.runId,
+        pending: this.pendingEntries.get(id)?.length ?? 0,
+        messages: this.messageCounts.get(id) ?? 0
+      };
+    });
     this.logger.info("flushAll called", {
       sessions: sessionIds.length,
       pending: pendingCounts
@@ -4437,8 +4467,8 @@ var SessionLogWriter = class _SessionLogWriter {
       return;
     }
     this.logger.info("Session registered", {
-      sessionId,
-      taskId: context.taskId
+      taskId: context.taskId,
+      runId: context.runId
     });
     this.sessions.set(sessionId, { context });
     this.lastFlushAttemptTime.set(sessionId, Date.now());
@@ -4472,7 +4502,11 @@ var SessionLogWriter = class _SessionLogWriter {
     const count = (this.messageCounts.get(sessionId) ?? 0) + 1;
     this.messageCounts.set(sessionId, count);
     if (count % 10 === 1) {
-      this.logger.info("Messages received", { count, sessionId });
+      this.logger.info("Messages received", {
+        count,
+        taskId: session.context.taskId,
+        runId: session.context.runId
+      });
     }
     try {
       const message = JSON.parse(line);
@@ -4503,7 +4537,8 @@ var SessionLogWriter = class _SessionLogWriter {
       }
     } catch {
       this.logger.warn("Failed to parse raw line for persistence", {
-        sessionId,
+        taskId: session.context.taskId,
+        runId: session.context.runId,
         lineLength: line.length
       });
     }
@@ -4518,7 +4553,8 @@ var SessionLogWriter = class _SessionLogWriter {
     const pending = this.pendingEntries.get(sessionId);
     if (!this.posthogAPI || !pending?.length) {
       this.logger.info("flush: nothing to persist", {
-        sessionId,
+        taskId: session.context.taskId,
+        runId: session.context.runId,
         hasPosthogAPI: !!this.posthogAPI,
         pendingCount: pending?.length ?? 0
       });
@@ -4539,7 +4575,8 @@ var SessionLogWriter = class _SessionLogWriter {
       );
       this.retryCounts.set(sessionId, 0);
       this.logger.info("Flushed session logs", {
-        sessionId,
+        taskId: session.context.taskId,
+        runId: session.context.runId,
         entryCount: pending.length
       });
     } catch (error) {
@@ -4548,7 +4585,11 @@ var SessionLogWriter = class _SessionLogWriter {
       if (retryCount >= _SessionLogWriter.MAX_FLUSH_RETRIES) {
         this.logger.error(
           `Dropping ${pending.length} session log entries after ${retryCount} failed flush attempts`,
-          { sessionId, error }
+          {
+            taskId: session.context.taskId,
+            runId: session.context.runId,
+            error
+          }
         );
         this.retryCounts.set(sessionId, 0);
       } else {
@@ -4637,7 +4678,12 @@ var SessionLogWriter = class _SessionLogWriter {
       import_node_fs2.default.appendFileSync(logPath, `${JSON.stringify(entry)}
 `);
     } catch (error) {
-      this.logger.warn("Failed to write to local cache", { logPath, error });
+      this.logger.warn("Failed to write to local cache", {
+        taskId: session.context.taskId,
+        runId: session.context.runId,
+        logPath,
+        error
+      });
     }
   }
 };
@@ -10678,7 +10724,8 @@ Important:
   }
   configureEnvironment() {
     const { apiKey, apiUrl, projectId } = this.config;
-    const gatewayUrl = process.env.LLM_GATEWAY_URL || getLlmGatewayUrl(apiUrl);
+    const product = this.config.mode === "background" ? "background_agents" : "twig";
+    const gatewayUrl = process.env.LLM_GATEWAY_URL || getLlmGatewayUrl(apiUrl, product);
     const openaiBaseUrl = gatewayUrl.endsWith("/v1") ? gatewayUrl : `${gatewayUrl}/v1`;
     Object.assign(process.env, {
       // PostHog