@posthog/agent 2.3.213 → 2.3.233

package/package.json

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

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

@@ -108,9 +108,15 @@ export class CodexAcpAgent extends BaseAcpAgent {
     super(client);
     this.logger = new Logger({ debug: true, prefix: "[CodexAcpAgent]" });
 
+    // Load user codex settings before spawning so spawnCodexProcess can
+    // filter out any [mcp_servers.*] entries from ~/.codex/config.toml.
+    const cwd = options.codexProcessOptions.cwd ?? process.cwd();
+    const settingsManager = new CodexSettingsManager(cwd);
+
     // Spawn the codex-acp subprocess
     this.codexProcess = spawnCodexProcess({
       ...options.codexProcessOptions,
+      settings: settingsManager.getSettings(),
       logger: this.logger,
       processCallbacks: options.processCallbacks,
     });
@@ -120,9 +126,6 @@ export class CodexAcpAgent extends BaseAcpAgent {
     const codexWritable = nodeWritableToWebWritable(this.codexProcess.stdin);
     const codexStream = ndJsonStream(codexWritable, codexReadable);
 
-    // Set up session with CodexSettingsManager
-    const cwd = options.codexProcessOptions.cwd ?? process.cwd();
-    const settingsManager = new CodexSettingsManager(cwd);
     const abortController = new AbortController();
     this.session = {
       abortController,

package/src/adapters/codex/settings.ts

@@ -13,6 +13,8 @@ export interface CodexSettings {
   personality?: string;
   modelReasoningEffort?: string;
   trustLevel?: string;
+  // Names of every `[mcp_servers.<name>]` section declared in the user's config.toml
+  mcpServerNames: string[];
 }
 
 /**
@@ -24,32 +26,29 @@ export interface CodexSettings {
  */
 export class CodexSettingsManager {
   private cwd: string;
-  private settings: CodexSettings = {};
-  private initialized = false;
+  private settings: CodexSettings = { mcpServerNames: [] };
 
   constructor(cwd: string) {
     this.cwd = cwd;
+    this.loadSettings();
   }
 
   async initialize(): Promise<void> {
-    if (this.initialized) {
-      return;
-    }
-    await this.loadSettings();
-    this.initialized = true;
+    // No-op: settings are loaded in the constructor. Kept async to
+    // satisfy the BaseSettingsManager interface.
   }
 
   private getConfigPath(): string {
     return path.join(os.homedir(), ".codex", "config.toml");
   }
 
-  private async loadSettings(): Promise<void> {
+  private loadSettings(): void {
     const configPath = this.getConfigPath();
     try {
-      const content = await fs.promises.readFile(configPath, "utf-8");
+      const content = fs.readFileSync(configPath, "utf-8");
       this.settings = parseCodexToml(content, this.cwd);
     } catch {
-      this.settings = {};
+      this.settings = { mcpServerNames: [] };
     }
   }
 
@@ -62,17 +61,13 @@ export class CodexSettingsManager {
   }
 
   async setCwd(cwd: string): Promise<void> {
-    if (this.cwd === cwd) {
-      return;
-    }
-    this.dispose();
+    if (this.cwd === cwd) return;
     this.cwd = cwd;
-    this.initialized = false;
-    await this.initialize();
+    this.loadSettings();
   }
 
   dispose(): void {
-    this.initialized = false;
+    // No-op: no resources to release. Kept to satisfy the BaseSettingsManager interface.
   }
 }
 
@@ -82,7 +77,8 @@ export class CodexSettingsManager {
  * Does NOT handle full TOML spec — only what codex config uses.
  */
 function parseCodexToml(content: string, cwd: string): CodexSettings {
-  const settings: CodexSettings = {};
+  const settings: CodexSettings = { mcpServerNames: [] };
+  const mcpServerNames = new Set<string>();
   let currentSection = "";
 
   for (const line of content.split("\n")) {
@@ -93,6 +89,9 @@ function parseCodexToml(content: string, cwd: string): CodexSettings {
     const sectionMatch = trimmed.match(/^\[(.+)\]$/);
     if (sectionMatch) {
       currentSection = sectionMatch[1] ?? "";
+      if (currentSection.startsWith("mcp_servers.")) {
+        mcpServerNames.add(currentSection.slice("mcp_servers.".length));
+      }
       continue;
     }
 
@@ -123,5 +122,6 @@ function parseCodexToml(content: string, cwd: string): CodexSettings {
     }
   }
 
+  settings.mcpServerNames = Array.from(mcpServerNames);
   return settings;
 }

package/src/adapters/codex/spawn.ts

@@ -4,6 +4,7 @@ import { delimiter, dirname } from "node:path";
 import type { Readable, Writable } from "node:stream";
 import type { ProcessSpawnedCallback } from "../../types";
 import { Logger } from "../../utils/logger";
+import type { CodexSettings } from "./settings";
 
 export interface CodexProcessOptions {
   cwd?: string;
@@ -14,6 +15,7 @@ export interface CodexProcessOptions {
   binaryPath?: string;
   logger?: Logger;
   processCallbacks?: ProcessSpawnedCallback;
+  settings?: CodexSettings;
 }
 
 export interface CodexProcess {
@@ -28,6 +30,13 @@ function buildConfigArgs(options: CodexProcessOptions): string[] {
 
   args.push("-c", `features.remote_models=false`);
 
+  // Disable the user's local MCPs one-by-one so Codex only uses the MCPs we
+  // provide via ACP. We can't use `-c mcp_servers={}` because that makes Codex
+  // ignore MCPs entirely, including the ones we inject later.
+  for (const name of options.settings?.mcpServerNames ?? []) {
+    args.push("-c", `mcp_servers.${name}.enabled=false`);
+  }
+
   if (options.apiBaseUrl) {
     args.push("-c", `model_provider="posthog"`);
     args.push("-c", `model_providers.posthog.name="PostHog Gateway"`);

package/src/server/agent-server.ts

@@ -5,6 +5,7 @@ import {
   PROTOCOL_VERSION,
 } from "@agentclientprotocol/sdk";
 import { type ServerType, serve } from "@hono/node-server";
+import { getCurrentBranch } from "@posthog/git/queries";
 import { Hono } from "hono";
 import packageJson from "../../package.json" with { type: "json" };
 import { POSTHOG_NOTIFICATIONS } from "../acp-extensions";
@@ -167,6 +168,7 @@ export class AgentServer {
   private posthogAPI: PostHogAPIClient;
   private questionRelayedToSlack = false;
   private detectedPrUrl: string | null = null;
+  private lastReportedBranch: string | null = null;
   private resumeState: ResumeState | null = null;
   // Guards against concurrent session initialization. autoInitializeSession() and
   // the GET /events SSE handler can both call initializeSession() — the SSE connection
@@ -524,6 +526,10 @@ export class AgentServer {
           stopReason: result.stopReason,
         });
 
+        if (result.stopReason === "end_turn") {
+          void this.syncCloudBranchMetadata(this.session.payload);
+        }
+
         this.broadcastTurnComplete(result.stopReason);
 
         if (result.stopReason === "end_turn") {
@@ -879,6 +885,10 @@ export class AgentServer {
         stopReason: result.stopReason,
       });
 
+      if (result.stopReason === "end_turn") {
+        void this.syncCloudBranchMetadata(payload);
+      }
+
       this.broadcastTurnComplete(result.stopReason);
 
       if (result.stopReason === "end_turn") {
@@ -964,6 +974,10 @@ export class AgentServer {
         stopReason: result.stopReason,
       });
 
+      if (result.stopReason === "end_turn") {
+        void this.syncCloudBranchMetadata(payload);
+      }
+
       this.broadcastTurnComplete(result.stopReason);
 
       if (result.stopReason === "end_turn") {
@@ -1168,6 +1182,44 @@ ${attributionInstructions}
 `;
   }
 
+  private async getCurrentGitBranch(): Promise<string | null> {
+    if (!this.config.repositoryPath) {
+      return null;
+    }
+
+    try {
+      return await getCurrentBranch(this.config.repositoryPath);
+    } catch (error) {
+      this.logger.warn("Failed to determine current git branch", {
+        repositoryPath: this.config.repositoryPath,
+        error,
+      });
+      return null;
+    }
+  }
+
+  private async syncCloudBranchMetadata(payload: JwtPayload): Promise<void> {
+    const branchName = await this.getCurrentGitBranch();
+    if (!branchName || branchName === this.lastReportedBranch) {
+      return;
+    }
+
+    try {
+      await this.posthogAPI.updateTaskRun(payload.task_id, payload.run_id, {
+        branch: branchName,
+        output: { head_branch: branchName },
+      });
+      this.lastReportedBranch = branchName;
+    } catch (error) {
+      this.logger.warn("Failed to attach current branch to task run", {
+        taskId: payload.task_id,
+        runId: payload.run_id,
+        branchName,
+        error,
+      });
+    }
+  }
+
   private async signalTaskComplete(
     payload: JwtPayload,
     stopReason: string,
@@ -1556,6 +1608,7 @@ ${attributionInstructions}
     }
 
     this.pendingEvents = [];
+    this.lastReportedBranch = null;
     this.session = null;
   }