@posthog/agent 2.3.548 → 2.3.616

package/dist/server/bin.cjs

@@ -3951,6 +3951,7 @@ var import_sdk5 = require("@agentclientprotocol/sdk");
 var import_node_server = require("@hono/node-server");
 
 // ../git/dist/queries.js
+var import_node_fs = require("fs");
 var fs3 = __toESM(require("fs/promises"), 1);
 var path2 = __toESM(require("path"), 1);
 
@@ -8502,13 +8503,20 @@ init_git_response_error();
 var simpleGit = gitInstanceFactory;
 
 // ../git/dist/client.js
+var PERFORMANCE_CONFIG = [
+  "core.untrackedCache=true",
+  "core.fsmonitor=true",
+  "core.preloadIndex=true"
+];
 function createGitClient(baseDir, options) {
-  const { abortSignal: signal, ...rest } = options ?? {};
+  const { abortSignal: signal, config: callerConfig, ...rest } = options ?? {};
+  const config = callerConfig ? [...PERFORMANCE_CONFIG, ...callerConfig] : PERFORMANCE_CONFIG;
   return simpleGit({
     baseDir,
     maxConcurrentProcesses: 6,
     trimmed: true,
     abort: signal,
+    config,
     ...rest
   });
 }
@@ -8661,14 +8669,18 @@ var GitOperationManagerImpl = class _GitOperationManagerImpl {
   }
   async executeRead(repoPath, operation, options) {
     const state = this.getRepoState(repoPath);
+    const env = {
+      ...getCleanEnv(),
+      GIT_OPTIONAL_LOCKS: "0",
+      ...options?.env
+    };
     if (options?.signal) {
       const scopedGit = createGitClient(repoPath, {
         abortSignal: options.signal
       });
-      return operation(scopedGit.env({ ...getCleanEnv(), GIT_OPTIONAL_LOCKS: "0" }));
+      return operation(scopedGit.env(env));
     }
-    const git = state.client.env({ ...getCleanEnv(), GIT_OPTIONAL_LOCKS: "0" });
-    return operation(git);
+    return operation(state.client.env(env));
   }
   async executeWrite(repoPath, operation, options) {
     const state = this.getRepoState(repoPath);
@@ -8678,15 +8690,16 @@ var GitOperationManagerImpl = class _GitOperationManagerImpl {
         throw new Error(`Git repository is locked: ${repoPath}`);
       }
     }
+    const env = { ...getCleanEnv(), ...options?.env };
     await state.lock.acquireWrite();
     try {
       if (options?.signal) {
         const scopedGit = createGitClient(repoPath, {
           abortSignal: options.signal
         });
-        return await operation(scopedGit.env(getCleanEnv()));
+        return await operation(scopedGit.env(env));
       }
-      return await operation(state.client.env(getCleanEnv()));
+      return await operation(state.client.env(env));
     } catch (error) {
       if (options?.signal?.aborted) {
         await removeLock(repoPath).catch(() => {
@@ -8713,6 +8726,9 @@ function getGitOperationManager() {
   return instance2;
 }
 
+// ../git/dist/status-stream.js
+var import_node_child_process2 = require("child_process");
+
 // ../git/dist/queries.js
 async function getCurrentBranch(baseDir, options) {
   const manager = getGitOperationManager();
@@ -8747,6 +8763,44 @@ async function listWorktrees(baseDir, options) {
     return worktrees;
   }, { signal: options?.abortSignal });
 }
+async function inspectGitBusyState(git) {
+  const toplevel = (await git.raw(["rev-parse", "--show-toplevel"])).trim();
+  const resolveGitPath = async (gitPath) => {
+    const relative = (await git.raw(["rev-parse", "--git-path", gitPath])).trim();
+    return path2.isAbsolute(relative) ? relative : path2.resolve(toplevel, relative);
+  };
+  const pathExists = async (gitPath) => {
+    const resolved = await resolveGitPath(gitPath);
+    try {
+      await fs3.access(resolved);
+      return true;
+    } catch {
+      return false;
+    }
+  };
+  const dirExists = async (gitPath) => {
+    const resolved = await resolveGitPath(gitPath);
+    try {
+      const stat4 = await fs3.stat(resolved);
+      return stat4.isDirectory();
+    } catch {
+      return false;
+    }
+  };
+  if (await dirExists("rebase-merge") || await dirExists("rebase-apply")) {
+    return { busy: true, operation: "rebase" };
+  }
+  if (await pathExists("MERGE_HEAD")) {
+    return { busy: true, operation: "merge" };
+  }
+  if (await pathExists("CHERRY_PICK_HEAD")) {
+    return { busy: true, operation: "cherry-pick" };
+  }
+  if (await pathExists("REVERT_HEAD")) {
+    return { busy: true, operation: "revert" };
+  }
+  return { busy: false };
+}
 
 // src/server/agent-server.ts
 var import_hono = require("hono");
@@ -8755,7 +8809,7 @@ var import_zod3 = require("zod");
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.548",
+  version: "2.3.616",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -11222,6 +11276,7 @@ var ParserManager = class {
   languages = /* @__PURE__ */ new Map();
   languageKeys = /* @__PURE__ */ new WeakMap();
   queryCache = /* @__PURE__ */ new Map();
+  failedQueries = /* @__PURE__ */ new Set();
   maxCacheSize = 256;
   initPromise = null;
   wasmDir = resolveGrammarsDir();
@@ -11229,6 +11284,7 @@ var ParserManager = class {
   updateConfig(config) {
     this.config = config;
     this.queryCache.clear();
+    this.failedQueries.clear();
   }
   async ensureInitialized() {
     if (!this.initPromise) {
@@ -11290,6 +11346,9 @@ var ParserManager = class {
     }
     const langKey = this.languageKeys.get(lang) ?? lang.toString();
     const cacheKey = `${langKey}:${queryStr}`;
+    if (this.failedQueries.has(cacheKey)) {
+      return null;
+    }
     let query2 = this.queryCache.get(cacheKey);
     if (query2) {
       this.queryCache.delete(cacheKey);
@@ -11306,8 +11365,8 @@ var ParserManager = class {
       }
       this.queryCache.set(cacheKey, query2);
       return query2;
-    } catch (err2) {
-      warn("Query compilation failed", err2);
+    } catch {
+      this.failedQueries.add(cacheKey);
       return null;
     }
   }
@@ -11318,6 +11377,7 @@ var ParserManager = class {
     this.languages.clear();
     this.languageKeys = /* @__PURE__ */ new WeakMap();
     this.queryCache.clear();
+    this.failedQueries.clear();
   }
 };
 async function findVariantBranches(pm, source, languageId) {
@@ -12843,6 +12903,33 @@ var PostHogApi = class {
     );
     return data.results.filter((f) => !f.deleted);
   }
+  // Keys absent from the returned map have NOT been called in the window.
+  async getFlagLastCalled(flagKeys, daysBack = 30) {
+    if (flagKeys.length === 0) return /* @__PURE__ */ new Map();
+    const days = Math.max(1, Math.min(365, Math.floor(daysBack)));
+    const query2 = `
+      SELECT
+        properties.$feature_flag AS flag_key,
+        max(timestamp) AS last_called_at
+      FROM events
+      WHERE event = '$feature_flag_called'
+        AND properties.$feature_flag IN {flagKeys}
+        AND timestamp >= now() - INTERVAL ${days} DAY
+      GROUP BY flag_key
+    `;
+    const data = await this.post("/query/", {
+      query: {
+        kind: "HogQLQuery",
+        query: query2,
+        values: { flagKeys }
+      }
+    });
+    const lastCalled = /* @__PURE__ */ new Map();
+    for (const [flagKey, lastCalledAt] of data.results) {
+      if (lastCalledAt) lastCalled.set(flagKey, lastCalledAt);
+    }
+    return lastCalled;
+  }
   async getExperiments() {
     const data = await this.get(
       "/experiments/?limit=500"
@@ -13664,6 +13751,23 @@ function tryParsePartialJson(s) {
   return null;
 }
 
+// src/adapters/error-classification.ts
+var UPSTREAM_PROVIDER_ERROR_STATUS_PATTERN = /API Error:\s*(?:429|5\d\d)\b/i;
+function classifyAgentError(result) {
+  if (!result) return "agent_error";
+  const text2 = result.trim();
+  if (/API Error:\s*terminated\b/i.test(text2)) {
+    return "upstream_stream_terminated";
+  }
+  if (/API Error:\s*Connection error\b/i.test(text2)) {
+    return "upstream_connection_error";
+  }
+  if (UPSTREAM_PROVIDER_ERROR_STATUS_PATTERN.test(text2)) {
+    return "upstream_provider_failure";
+  }
+  return "agent_error";
+}
+
 // src/adapters/claude/permissions/posthog-exec-gate.ts
 var POSTHOG_EXEC_TOOL_RE = /^mcp__posthog(?:_[^_]+)*__exec$/;
 var POSTHOG_CALL_COMMAND_RE = /^\s*call\s+(?:--json\s+)?([a-zA-Z0-9_-]+)/;
@@ -13860,7 +13964,7 @@ var createPreToolUseHook = (settingsManager, logger) => async (input, _toolUseID
 };
 
 // src/adapters/claude/conversion/tool-use-to-acp.ts
-var import_node_fs = __toESM(require("fs"), 1);
+var import_node_fs2 = __toESM(require("fs"), 1);
 var import_node_path3 = __toESM(require("path"), 1);
 
 // src/adapters/claude/mcp/tool-metadata.ts
@@ -14094,7 +14198,7 @@ function toolInfoFromToolUse(toolUse, options) {
           oldContent = options.cachedFileContent[writeFilePath];
         } else {
           try {
-            oldContent = import_node_fs.default.readFileSync(writeFilePath, "utf-8");
+            oldContent = import_node_fs2.default.readFileSync(writeFilePath, "utf-8");
           } catch {
           }
         }
@@ -14508,7 +14612,7 @@ function resolveFileContent(filePath, oldText, cachedFileContent) {
     }
   }
   try {
-    const content = import_node_fs.default.readFileSync(filePath, "utf-8");
+    const content = import_node_fs2.default.readFileSync(filePath, "utf-8");
     if (content.includes(oldText)) {
       return content;
     }
@@ -14980,17 +15084,6 @@ async function handleSystemMessage(message, context) {
       break;
   }
 }
-function classifyAgentError(result) {
-  if (!result) return "agent_error";
-  const text2 = result.trim();
-  if (/API Error:\s*terminated\b/i.test(text2)) {
-    return "upstream_stream_terminated";
-  }
-  if (/API Error:\s*Connection error\b/i.test(text2)) {
-    return "upstream_connection_error";
-  }
-  return "agent_error";
-}
 function handleResultMessage(message) {
   const usage = extractUsageFromResult(message);
   switch (message.subtype) {
@@ -16052,7 +16145,7 @@ function parseMcpServers(params) {
 }
 
 // src/adapters/claude/session/options.ts
-var import_node_child_process2 = require("child_process");
+var import_node_child_process3 = require("child_process");
 var fs7 = __toESM(require("fs"), 1);
 var os3 = __toESM(require("os"), 1);
 var path9 = __toESM(require("path"), 1);
@@ -16114,6 +16207,10 @@ function buildMcpServers(userServers, acpServers, projectScopedServers) {
   };
 }
 function buildEnvironment() {
+  const bedrockFallbackHeader = "x-posthog-use-bedrock-fallback: true";
+  const existingCustomHeaders = process.env.ANTHROPIC_CUSTOM_HEADERS;
+  const customHeaders = existingCustomHeaders ? `${existingCustomHeaders}
+${bedrockFallbackHeader}` : bedrockFallbackHeader;
   return {
     ...process.env,
     ELECTRON_RUN_AS_NODE: "1",
@@ -16121,7 +16218,9 @@ function buildEnvironment() {
     // Offload all MCP tools by default
     ENABLE_TOOL_SEARCH: "auto:0",
     // Enable idle state as end-of-turn signal (required for SDK 0.2.114+)
-    CLAUDE_CODE_EMIT_SESSION_STATE_EVENTS: "1"
+    CLAUDE_CODE_EMIT_SESSION_STATE_EVENTS: "1",
+    // Route to AWS Bedrock as a fallback when Anthropic returns 5xx
+    ANTHROPIC_CUSTOM_HEADERS: customHeaders
   };
 }
 function buildHooks(userHooks, onModeChange, settingsManager, logger, enrichmentDeps, enrichedReadCache, registeredAgents) {
@@ -16194,7 +16293,7 @@ function getAbortController(userProvidedController) {
 }
 function buildSpawnWrapper(sessionId, onProcessSpawned, onProcessExited, logger) {
   return (spawnOpts) => {
-    const child = (0, import_node_child_process2.spawn)(spawnOpts.command, spawnOpts.args, {
+    const child = (0, import_node_child_process3.spawn)(spawnOpts.command, spawnOpts.args, {
       cwd: spawnOpts.cwd,
       env: spawnOpts.env,
       stdio: ["pipe", "pipe", "pipe"]
@@ -17867,7 +17966,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
 };
 
 // src/adapters/codex/codex-agent.ts
-var import_node_fs3 = require("fs");
+var import_node_fs4 = require("fs");
 var import_node_path5 = require("path");
 var import_sdk3 = require("@agentclientprotocol/sdk");
 
@@ -18152,8 +18251,8 @@ function parseCodexToml(content, cwd) {
 }
 
 // src/adapters/codex/spawn.ts
-var import_node_child_process3 = require("child_process");
-var import_node_fs2 = require("fs");
+var import_node_child_process4 = require("child_process");
+var import_node_fs3 = require("fs");
 var import_node_path4 = require("path");
 function buildConfigArgs(options) {
   const args2 = [];
@@ -18185,7 +18284,7 @@ function buildConfigArgs(options) {
 }
 function findCodexBinary(options) {
   const configArgs = buildConfigArgs(options);
-  if (options.binaryPath && (0, import_node_fs2.existsSync)(options.binaryPath)) {
+  if (options.binaryPath && (0, import_node_fs3.existsSync)(options.binaryPath)) {
     return { command: options.binaryPath, args: configArgs };
   }
   if (options.binaryPath) {
@@ -18204,7 +18303,7 @@ function spawnCodexProcess(options) {
     env.POSTHOG_GATEWAY_API_KEY = options.apiKey;
   }
   const { command, args: args2 } = findCodexBinary(options);
-  if (options.binaryPath && (0, import_node_fs2.existsSync)(options.binaryPath)) {
+  if (options.binaryPath && (0, import_node_fs3.existsSync)(options.binaryPath)) {
     const binDir = (0, import_node_path4.dirname)(options.binaryPath);
     env.PATH = `${binDir}${import_node_path4.delimiter}${env.PATH ?? ""}`;
   }
@@ -18216,7 +18315,7 @@ function spawnCodexProcess(options) {
     hasApiKey: !!options.apiKey,
     binaryPath: options.binaryPath
   });
-  const child = (0, import_node_child_process3.spawn)(command, args2, {
+  const child = (0, import_node_child_process4.spawn)(command, args2, {
     cwd: options.cwd,
     env,
     stdio: ["pipe", "pipe", "pipe"],
@@ -18275,6 +18374,17 @@ function prependPrContext(params) {
     prompt: [{ type: "text", text: prContext }, ...params.prompt]
   };
 }
+function classifyPromptError(error) {
+  const message = error instanceof Error ? error.message : String(error ?? "");
+  const classification = classifyAgentError(message);
+  if (classification === "agent_error") {
+    return error;
+  }
+  return import_sdk3.RequestError.internalError(
+    { classification, result: message },
+    message
+  );
+}
 var CODEX_NATIVE_MODE = {
   auto: "auto",
   default: "auto",
@@ -18305,7 +18415,7 @@ function resolveStructuredOutputMcpScript() {
   let dir = import_meta2.dirname ?? __dirname;
   for (let i2 = 0; i2 < 5; i2++) {
     const candidate = (0, import_node_path5.resolve)(dir, rel);
-    if ((0, import_node_fs3.existsSync)(candidate)) return candidate;
+    if ((0, import_node_fs4.existsSync)(candidate)) return candidate;
     dir = (0, import_node_path5.resolve)(dir, "..");
   }
   throw new Error(
@@ -18604,6 +18714,8 @@ var CodexAcpAgent = class extends BaseAcpAgent {
     let response;
     try {
       response = await this.codexConnection.prompt(prependPrContext(params));
+    } catch (error) {
+      throw classifyPromptError(error);
     } finally {
       this.session.promptRunning = false;
     }
@@ -18939,7 +19051,7 @@ var import_node_os2 = require("os");
 var import_node_path7 = require("path");
 
 // ../git/dist/handoff.js
-var import_node_child_process4 = require("child_process");
+var import_node_child_process5 = require("child_process");
 var import_promises2 = require("fs/promises");
 var import_node_os = require("os");
 var import_node_path6 = __toESM(require("path"), 1);
@@ -19105,7 +19217,7 @@ var GitSaga = class extends Saga {
     return manager.executeWrite(input.baseDir, async (git) => {
       this._git = git;
       return this.executeGitOperations(input);
-    }, { signal: input.signal });
+    }, { signal: input.signal, env: input.env });
   }
 };
 
@@ -19217,42 +19329,7 @@ async function hasUnmergedEntries(git) {
   return output.trim().length > 0;
 }
 async function getGitBusyState(git) {
-  const toplevel = (await git.raw(["rev-parse", "--show-toplevel"])).trim();
-  const resolveGitPath = async (gitPath) => {
-    const relative = (await git.raw(["rev-parse", "--git-path", gitPath])).trim();
-    return path13.isAbsolute(relative) ? relative : path13.resolve(toplevel, relative);
-  };
-  const pathExists = async (gitPath) => {
-    const resolved = await resolveGitPath(gitPath);
-    try {
-      await fs11.access(resolved);
-      return true;
-    } catch {
-      return false;
-    }
-  };
-  const dirExists = async (gitPath) => {
-    const resolved = await resolveGitPath(gitPath);
-    try {
-      const stat4 = await fs11.stat(resolved);
-      return stat4.isDirectory();
-    } catch {
-      return false;
-    }
-  };
-  if (await dirExists("rebase-merge") || await dirExists("rebase-apply")) {
-    return { busy: true, operation: "rebase" };
-  }
-  if (await pathExists("MERGE_HEAD")) {
-    return { busy: true, operation: "merge" };
-  }
-  if (await pathExists("CHERRY_PICK_HEAD")) {
-    return { busy: true, operation: "cherry-pick" };
-  }
-  if (await pathExists("REVERT_HEAD")) {
-    return { busy: true, operation: "revert" };
-  }
-  return { busy: false };
+  return inspectGitBusyState(git);
 }
 var MAX_WORKTREE_FILE_BYTES = 1024 * 1024;
 async function createWorktreeTree(git, baseDir, head) {
@@ -19754,7 +19831,7 @@ var GitHandoffTracker = class {
   }
   async runGitProcessAllowingFailure(args2) {
     return new Promise((resolve7, reject) => {
-      const child = (0, import_node_child_process4.spawn)("git", args2, {
+      const child = (0, import_node_child_process5.spawn)("git", args2, {
         cwd: this.repositoryPath,
         stdio: ["ignore", "ignore", "pipe"]
       });
@@ -19778,7 +19855,7 @@ var GitHandoffTracker = class {
   }
   async runGitWithEnv(env, args2) {
     return new Promise((resolve7, reject) => {
-      const child = (0, import_node_child_process4.spawn)("git", args2, {
+      const child = (0, import_node_child_process5.spawn)("git", args2, {
         cwd: this.repositoryPath,
         stdio: ["ignore", "pipe", "pipe"],
         env
@@ -19803,7 +19880,7 @@ var GitHandoffTracker = class {
   }
   runGitProcess(args2, input) {
     return new Promise((resolve7, reject) => {
-      const child = (0, import_node_child_process4.spawn)("git", args2, {
+      const child = (0, import_node_child_process5.spawn)("git", args2, {
         cwd: this.repositoryPath,
         stdio: "pipe"
       });
@@ -20655,7 +20732,7 @@ ${toolSummary}`);
 }
 
 // src/session-log-writer.ts
-var import_node_fs4 = __toESM(require("fs"), 1);
+var import_node_fs5 = __toESM(require("fs"), 1);
 var import_promises4 = __toESM(require("fs/promises"), 1);
 var import_node_path8 = __toESM(require("path"), 1);
 var SessionLogWriter = class _SessionLogWriter {
@@ -20699,7 +20776,7 @@ var SessionLogWriter = class _SessionLogWriter {
         context.runId
       );
       try {
-        import_node_fs4.default.mkdirSync(sessionDir, { recursive: true });
+        import_node_fs5.default.mkdirSync(sessionDir, { recursive: true });
       } catch (error) {
         this.logger.warn("Failed to create local cache directory", {
           sessionDir,
@@ -20958,7 +21035,7 @@ var SessionLogWriter = class _SessionLogWriter {
       "logs.ndjson"
     );
     try {
-      import_node_fs4.default.appendFileSync(logPath, `${JSON.stringify(entry)}
+      import_node_fs5.default.appendFileSync(logPath, `${JSON.stringify(entry)}
 `);
     } catch (error) {
       this.logger.warn("Failed to write to local cache", {
@@ -20993,11 +21070,11 @@ var SessionLogWriter = class _SessionLogWriter {
 };
 
 // src/server/agentsh-runtime.ts
-var import_node_child_process5 = require("child_process");
+var import_node_child_process6 = require("child_process");
 var import_promises5 = require("fs/promises");
 var import_node_util2 = require("util");
 var AGENTSH_SESSION_ID_FILE = "/tmp/agentsh-session-id";
-var execFileAsync2 = (0, import_node_util2.promisify)(import_node_child_process5.execFile);
+var execFileAsync2 = (0, import_node_util2.promisify)(import_node_child_process6.execFile);
 function errorMessage(error) {
   return error instanceof Error ? error.message : String(error);
 }
@@ -21217,8 +21294,15 @@ function validateCommandParams(method, params) {
 var agentErrorClassificationSchema = import_zod3.z.enum([
   "upstream_stream_terminated",
   "upstream_connection_error",
+  "upstream_provider_failure",
   "agent_error"
 ]);
+var UPSTREAM_PROVIDER_FAILURE_MESSAGE = "The upstream AI provider failed to process the request. Please retry the task in a few minutes.";
+var upstreamProviderFailureClassifications = /* @__PURE__ */ new Set([
+  "upstream_stream_terminated",
+  "upstream_connection_error",
+  "upstream_provider_failure"
+]);
 var errorWithClassificationSchema = import_zod3.z.object({
   data: import_zod3.z.object({ classification: agentErrorClassificationSchema })
 });
@@ -21962,7 +22046,9 @@ var AgentServer = class {
   }
   classifyAndSignalFailure(payload, phase, error) {
     const { classification, message } = this.extractErrorClassification(error);
-    const errorMessage2 = classification === "upstream_stream_terminated" ? "Upstream LLM stream terminated" : classification === "upstream_connection_error" ? "Upstream LLM connection error" : message || "Agent error";
+    const errorMessage2 = upstreamProviderFailureClassifications.has(
+      classification
+    ) ? UPSTREAM_PROVIDER_FAILURE_MESSAGE : message || "Agent error";
     this.logger.error(`send_${phase}_task_message_failed`, {
       classification,
       message