@posthog/agent 2.1.148 → 2.1.152

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.148",
+  version: "2.1.152",
   repository: "https://github.com/PostHog/twig",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -1810,8 +1810,14 @@ function toolInfoFromToolUse(toolUse, options) {
       };
     case "Edit": {
       const path9 = input?.file_path ? String(input.file_path) : void 0;
-      const oldText = input?.old_string ? String(input.old_string) : null;
-      const newText = input?.new_string ? String(input.new_string) : "";
+      let oldText = input?.old_string ? String(input.old_string) : null;
+      let newText = input?.new_string ? String(input.new_string) : "";
+      if (path9 && options?.cachedFileContent && path9 in options.cachedFileContent) {
+        const oldContent = options.cachedFileContent[path9];
+        const newContent = input?.replace_all ? oldContent.replaceAll(oldText ?? "", newText) : oldContent.replace(oldText ?? "", newText);
+        oldText = oldContent;
+        newText = newContent;
+      }
       return {
         title: path9 ? `Edit \`${path9}\`` : "Edit",
         kind: "edit",
@@ -1831,7 +1837,8 @@ function toolInfoFromToolUse(toolUse, options) {
       const filePath = input?.file_path ? String(input.file_path) : void 0;
       const contentStr = input?.content ? String(input.content) : void 0;
       if (filePath) {
-        contentResult = toolContent().diff(filePath, null, contentStr ?? "").build();
+        const oldContent = options?.cachedFileContent && filePath in options.cachedFileContent ? options.cachedFileContent[filePath] : null;
+        contentResult = toolContent().diff(filePath, oldContent, contentStr ?? "").build();
       } else if (contentStr) {
         contentResult = toolContent().text(contentStr).build();
       }
@@ -2325,7 +2332,8 @@ function handleToolUseChunk(chunk, ctx) {
   }
   const toolInfo = toolInfoFromToolUse(chunk, {
     supportsTerminalOutput: ctx.supportsTerminalOutput,
-    toolUseId: chunk.id
+    toolUseId: chunk.id,
+    cachedFileContent: ctx.fileContentCache
   });
   const meta = {
     ...toolMeta(chunk.name, void 0, ctx.parentToolCallId)
@@ -2351,6 +2359,47 @@ function handleToolUseChunk(chunk, ctx) {
     ...toolInfo
   };
 }
+function extractTextFromContent(content) {
+  if (Array.isArray(content)) {
+    const parts = [];
+    for (const item of content) {
+      if (typeof item === "object" && item !== null && "text" in item && typeof item.text === "string") {
+        parts.push(item.text);
+      }
+    }
+    return parts.length > 0 ? parts.join("") : null;
+  }
+  if (typeof content === "string") {
+    return content;
+  }
+  return null;
+}
+function stripCatLineNumbers(text2) {
+  return text2.replace(/^ *\d+[\t→]/gm, "");
+}
+function updateFileContentCache(toolUse, chunk, ctx) {
+  const input = toolUse.input;
+  const filePath = input?.file_path ? String(input.file_path) : void 0;
+  if (!filePath) return;
+  if (toolUse.name === "Read" && !input?.limit && !input?.offset) {
+    const fileText = extractTextFromContent(chunk.content);
+    if (fileText !== null) {
+      ctx.fileContentCache[filePath] = stripCatLineNumbers(fileText);
+    }
+  } else if (toolUse.name === "Write") {
+    const content = input?.content;
+    if (typeof content === "string") {
+      ctx.fileContentCache[filePath] = content;
+    }
+  } else if (toolUse.name === "Edit") {
+    const oldString = input?.old_string;
+    const newString = input?.new_string;
+    if (typeof oldString === "string" && typeof newString === "string" && filePath in ctx.fileContentCache) {
+      const current = ctx.fileContentCache[filePath];
+      ctx.fileContentCache[filePath] = input?.replace_all ? current.replaceAll(oldString, newString) : current.replace(oldString, newString);
+    }
+  }
+}
 function handleToolResultChunk(chunk, ctx) {
   const toolUse = ctx.toolUseCache[chunk.tool_use_id];
   if (!toolUse) {
@@ -2362,12 +2411,16 @@ function handleToolResultChunk(chunk, ctx) {
   if (toolUse.name === "TodoWrite") {
     return [];
   }
+  if (!chunk.is_error) {
+    updateFileContentCache(toolUse, chunk, ctx);
+  }
   const { _meta: resultMeta, ...toolUpdate } = toolUpdateFromToolResult(
     chunk,
     toolUse,
     {
       supportsTerminalOutput: ctx.supportsTerminalOutput,
-      toolUseId: chunk.tool_use_id
+      toolUseId: chunk.tool_use_id,
+      cachedFileContent: ctx.fileContentCache
     }
   );
   const updates = [];
@@ -3416,7 +3469,7 @@ function getAbortController(userProvidedController) {
   }
   return controller;
 }
-function buildSpawnWrapper(sessionId, onProcessSpawned, onProcessExited) {
+function buildSpawnWrapper(sessionId, onProcessSpawned, onProcessExited, logger) {
   return (spawnOpts) => {
     const child = (0, import_node_child_process.spawn)(spawnOpts.command, spawnOpts.args, {
       cwd: spawnOpts.cwd,
@@ -3430,6 +3483,12 @@ function buildSpawnWrapper(sessionId, onProcessSpawned, onProcessExited) {
         sessionId
       });
     }
+    child.stderr?.on("data", (data) => {
+      const msg = data.toString().trim();
+      if (msg && logger) {
+        logger.debug(`[claude-code:${child.pid}] stderr: ${msg}`);
+      }
+    });
     if (onProcessExited) {
       child.on("exit", () => {
         if (child.pid) {
@@ -3521,7 +3580,8 @@ function buildSessionOptions(params) {
       spawnClaudeCodeProcess: buildSpawnWrapper(
         params.sessionId,
         params.onProcessSpawned,
-        params.onProcessExited
+        params.onProcessExited,
+        params.logger
       )
     }
   };
@@ -4524,7 +4584,7 @@ function spawnCodexProcess(options) {
     detached: process.platform !== "win32"
   });
   child.stderr?.on("data", (data) => {
-    logger.debug("codex-acp stderr:", data.toString());
+    logger.warn("codex-acp stderr:", data.toString());
   });
   child.on("error", (err) => {
     logger.error("codex-acp process error:", err);
@@ -5087,12 +5147,14 @@ var PostHogAPIClient = class {
 
 // src/session-log-writer.ts
 var import_node_fs2 = __toESM(require("fs"), 1);
+var import_promises = __toESM(require("fs/promises"), 1);
 var import_node_path2 = __toESM(require("path"), 1);
 var SessionLogWriter = class _SessionLogWriter {
   static FLUSH_DEBOUNCE_MS = 500;
   static FLUSH_MAX_INTERVAL_MS = 5e3;
   static MAX_FLUSH_RETRIES = 10;
   static MAX_RETRY_DELAY_MS = 3e4;
+  static SESSIONS_MAX_AGE_MS = 30 * 24 * 60 * 60 * 1e3;
   posthogAPI;
   pendingEntries = /* @__PURE__ */ new Map();
   flushTimeouts = /* @__PURE__ */ new Map();
@@ -5229,10 +5291,16 @@ var SessionLogWriter = class _SessionLogWriter {
         );
         this.retryCounts.set(sessionId, 0);
       } else {
-        this.logger.error(
-          `Failed to persist session logs (attempt ${retryCount}/${_SessionLogWriter.MAX_FLUSH_RETRIES}):`,
-          error
-        );
+        if (retryCount === 1) {
+          this.logger.warn(
+            `Failed to persist session logs, will retry (up to ${_SessionLogWriter.MAX_FLUSH_RETRIES} attempts)`,
+            {
+              taskId: session.context.taskId,
+              runId: session.context.runId,
+              error: error instanceof Error ? error.message : String(error)
+            }
+          );
+        }
         const currentPending = this.pendingEntries.get(sessionId) ?? [];
         this.pendingEntries.set(sessionId, [...pending, ...currentPending]);
         this.scheduleFlush(sessionId);
@@ -5346,6 +5414,27 @@ var SessionLogWriter = class _SessionLogWriter {
       });
     }
   }
+  static async cleanupOldSessions(localCachePath) {
+    const sessionsDir = import_node_path2.default.join(localCachePath, "sessions");
+    let deleted = 0;
+    try {
+      const entries = await import_promises.default.readdir(sessionsDir);
+      const now = Date.now();
+      for (const entry of entries) {
+        const entryPath = import_node_path2.default.join(sessionsDir, entry);
+        try {
+          const stats = await import_promises.default.stat(entryPath);
+          if (stats.isDirectory() && now - stats.birthtimeMs > _SessionLogWriter.SESSIONS_MAX_AGE_MS) {
+            await import_promises.default.rm(entryPath, { recursive: true, force: true });
+            deleted++;
+          }
+        } catch {
+        }
+      }
+    } catch {
+    }
+    return deleted;
+  }
 };
 
 // ../git/dist/queries.js
@@ -9913,7 +10002,7 @@ function createGitClient(baseDir, options) {
 
 // ../git/dist/lock-detector.js
 var import_node_child_process3 = require("child_process");
-var import_promises = __toESM(require("fs/promises"), 1);
+var import_promises2 = __toESM(require("fs/promises"), 1);
 var import_node_path4 = __toESM(require("path"), 1);
 var import_node_util = require("util");
 var execFileAsync = (0, import_node_util.promisify)(import_node_child_process3.execFile);
@@ -9928,7 +10017,7 @@ async function getIndexLockPath(repoPath) {
 async function getLockInfo(repoPath) {
   const lockPath = await getIndexLockPath(repoPath);
   try {
-    const stat = await import_promises.default.stat(lockPath);
+    const stat = await import_promises2.default.stat(lockPath);
     return {
       path: lockPath,
       ageMs: Date.now() - stat.mtimeMs
@@ -9939,7 +10028,7 @@ async function getLockInfo(repoPath) {
 }
 async function removeLock(repoPath) {
   const lockPath = await getIndexLockPath(repoPath);
-  await import_promises.default.rm(lockPath, { force: true });
+  await import_promises2.default.rm(lockPath, { force: true });
 }
 async function isLocked(repoPath) {
   return await getLockInfo(repoPath) !== null;
@@ -10105,7 +10194,7 @@ async function getHeadSha(baseDir, options) {
 }
 
 // src/sagas/apply-snapshot-saga.ts
-var import_promises2 = require("fs/promises");
+var import_promises3 = require("fs/promises");
 var import_node_path5 = require("path");
 
 // ../shared/dist/index.js
@@ -10136,12 +10225,12 @@ var Saga = class {
     this.currentStepName = "unknown";
     this.stepTimings = [];
     const sagaStart = performance.now();
-    this.log.info("Starting saga", { sagaName: this.constructor.name });
+    this.log.info("Starting saga", { sagaName: this.sagaName });
     try {
       const result = await this.execute(input);
       const totalDuration = performance.now() - sagaStart;
       this.log.debug("Saga completed successfully", {
-        sagaName: this.constructor.name,
+        sagaName: this.sagaName,
         stepsCompleted: this.completedSteps.length,
         totalDurationMs: Math.round(totalDuration),
         stepTimings: this.stepTimings
@@ -10149,7 +10238,7 @@ var Saga = class {
       return { success: true, data: result };
     } catch (error) {
       this.log.error("Saga failed, initiating rollback", {
-        sagaName: this.constructor.name,
+        sagaName: this.sagaName,
         failedStep: this.currentStepName,
         error: error instanceof Error ? error.message : String(error)
       });
@@ -10261,6 +10350,7 @@ var GitSaga = class extends Saga {
 
 // ../git/dist/sagas/tree.js
 var CaptureTreeSaga = class extends GitSaga {
+  sagaName = "CaptureTreeSaga";
   tempIndexPath = null;
   async executeGitOperations(input) {
     const { baseDir, lastTreeHash, archivePath, signal } = input;
@@ -10381,6 +10471,7 @@ var CaptureTreeSaga = class extends GitSaga {
   }
 };
 var ApplyTreeSaga = class extends GitSaga {
+  sagaName = "ApplyTreeSaga";
   originalHead = null;
   originalBranch = null;
   extractedFiles = [];
@@ -10515,6 +10606,7 @@ var ApplyTreeSaga = class extends GitSaga {
 
 // src/sagas/apply-snapshot-saga.ts
 var ApplySnapshotSaga = class extends Saga {
+  sagaName = "ApplySnapshotSaga";
   archivePath = null;
   async execute(input) {
     const { snapshot, repositoryPath, apiClient, taskId, runId } = input;
@@ -10525,7 +10617,7 @@ var ApplySnapshotSaga = class extends Saga {
     const archiveUrl = snapshot.archiveUrl;
     await this.step({
       name: "create_tmp_dir",
-      execute: () => (0, import_promises2.mkdir)(tmpDir, { recursive: true }),
+      execute: () => (0, import_promises3.mkdir)(tmpDir, { recursive: true }),
       rollback: async () => {
       }
     });
@@ -10544,11 +10636,11 @@ var ApplySnapshotSaga = class extends Saga {
         }
         const base64Content = Buffer.from(arrayBuffer).toString("utf-8");
         const binaryContent = Buffer.from(base64Content, "base64");
-        await (0, import_promises2.writeFile)(archivePath, binaryContent);
+        await (0, import_promises3.writeFile)(archivePath, binaryContent);
       },
       rollback: async () => {
         if (this.archivePath) {
-          await (0, import_promises2.rm)(this.archivePath, { force: true }).catch(() => {
+          await (0, import_promises3.rm)(this.archivePath, { force: true }).catch(() => {
           });
         }
       }
@@ -10564,7 +10656,7 @@ var ApplySnapshotSaga = class extends Saga {
     if (!applyResult.success) {
       throw new Error(`Failed to apply tree: ${applyResult.error}`);
     }
-    await (0, import_promises2.rm)(this.archivePath, { force: true }).catch(() => {
+    await (0, import_promises3.rm)(this.archivePath, { force: true }).catch(() => {
     });
     this.log.info("Tree snapshot applied", {
       treeHash: snapshot.treeHash,
@@ -10577,9 +10669,10 @@ var ApplySnapshotSaga = class extends Saga {
 
 // src/sagas/capture-tree-saga.ts
 var import_node_fs4 = require("fs");
-var import_promises3 = require("fs/promises");
+var import_promises4 = require("fs/promises");
 var import_node_path6 = require("path");
 var CaptureTreeSaga2 = class extends Saga {
+  sagaName = "CaptureTreeSaga";
   async execute(input) {
     const {
       repositoryPath,
@@ -10626,7 +10719,7 @@ var CaptureTreeSaga2 = class extends Saga {
           runId
         );
       } finally {
-        await (0, import_promises3.rm)(createdArchivePath, { force: true }).catch(() => {
+        await (0, import_promises4.rm)(createdArchivePath, { force: true }).catch(() => {
         });
       }
     }
@@ -10650,7 +10743,7 @@ var CaptureTreeSaga2 = class extends Saga {
     const archiveUrl = await this.step({
       name: "upload_archive",
       execute: async () => {
-        const archiveContent = await (0, import_promises3.readFile)(archivePath);
+        const archiveContent = await (0, import_promises4.readFile)(archivePath);
         const base64Content = archiveContent.toString("base64");
         const artifacts = await apiClient.uploadTaskArtifacts(taskId, runId, [
           {
@@ -10670,7 +10763,7 @@ var CaptureTreeSaga2 = class extends Saga {
         return void 0;
       },
       rollback: async () => {
-        await (0, import_promises3.rm)(archivePath, { force: true }).catch(() => {
+        await (0, import_promises4.rm)(archivePath, { force: true }).catch(() => {
         });
       }
     });
@@ -10978,6 +11071,24 @@ var AgentServer = class {
   posthogAPI;
   questionRelayedToSlack = false;
   detectedPrUrl = null;
+  emitConsoleLog = (level, _scope, message, data) => {
+    if (!this.session) return;
+    const formatted = data !== void 0 ? `${message} ${JSON.stringify(data)}` : message;
+    const notification = {
+      jsonrpc: "2.0",
+      method: POSTHOG_NOTIFICATIONS.CONSOLE,
+      params: { level, message: formatted }
+    };
+    this.broadcastEvent({
+      type: "notification",
+      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+      notification
+    });
+    this.session.logWriter.appendRawLine(
+      this.session.payload.run_id,
+      JSON.stringify(notification)
+    );
+  };
   constructor(config) {
     this.config = config;
     this.logger = new Logger({ debug: true, prefix: "[AgentServer]" });
@@ -11329,6 +11440,14 @@ You MUST NOT create a new branch, close the existing PR, or create a new PR.`
       deviceInfo,
       logWriter
     };
+    this.logger = new Logger({
+      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.posthogAPI.updateTaskRun(payload.task_id, payload.run_id, {
       status: "in_progress"
@@ -11711,15 +11830,25 @@ Important:
           ...snapshot,
           device: this.session.deviceInfo
         };
+        const notification = {
+          jsonrpc: "2.0",
+          method: POSTHOG_NOTIFICATIONS.TREE_SNAPSHOT,
+          params: snapshotWithDevice
+        };
         this.broadcastEvent({
           type: "notification",
           timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-          notification: {
-            jsonrpc: "2.0",
-            method: POSTHOG_NOTIFICATIONS.TREE_SNAPSHOT,
-            params: snapshotWithDevice
-          }
+          notification
         });
+        const { archiveUrl: _, ...paramsWithoutArchive } = snapshotWithDevice;
+        const logNotification = {
+          ...notification,
+          params: paramsWithoutArchive
+        };
+        this.session.logWriter.appendRawLine(
+          this.session.payload.run_id,
+          JSON.stringify(logNotification)
+        );
       }
     } catch (error) {
       this.logger.error("Failed to capture tree state", error);