@posthog/agent 2.3.520 → 2.3.524

package/dist/server/bin.cjs

@@ -3945,7 +3945,7 @@ function isSupportedReasoningEffort(adapter, modelId, value) {
 
 // src/server/agent-server.ts
 var import_promises6 = require("fs/promises");
-var import_node_path8 = require("path");
+var import_node_path9 = require("path");
 var import_node_url2 = require("url");
 var import_sdk5 = require("@agentclientprotocol/sdk");
 var import_node_server = require("@hono/node-server");
@@ -8755,7 +8755,7 @@ var import_zod3 = require("zod");
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.520",
+  version: "2.3.524",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -8872,7 +8872,6 @@ var package_default = {
   },
   dependencies: {
     "@agentclientprotocol/sdk": "0.19.0",
-    ajv: "^8.17.1",
     "@anthropic-ai/claude-agent-sdk": "0.2.112",
     "@anthropic-ai/sdk": "0.89.0",
     "@hono/node-server": "^1.19.9",
@@ -8886,6 +8885,7 @@ var package_default = {
     hono: "^4.11.7",
     jsonwebtoken: "^9.0.2",
     minimatch: "^10.0.3",
+    "@modelcontextprotocol/sdk": "1.29.0",
     tar: "^7.5.0",
     uuid: "13.0.0",
     "yoga-wasm-web": "^0.3.3",
@@ -17841,9 +17841,26 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
 };
 
 // src/adapters/codex/codex-agent.ts
+var import_node_fs3 = require("fs");
+var import_node_path5 = require("path");
 var import_sdk3 = require("@agentclientprotocol/sdk");
 
+// src/adapters/codex/structured-output-constants.ts
+var STRUCTURED_OUTPUT_MCP_NAME = "posthog_output";
+var STRUCTURED_OUTPUT_TOOL_NAME = "create_output";
+
 // src/adapters/codex/codex-client.ts
+function isStructuredOutputToolCall(title) {
+  if (!title) return false;
+  if (title === STRUCTURED_OUTPUT_TOOL_NAME) return true;
+  return title.includes(STRUCTURED_OUTPUT_MCP_NAME) && title.includes(STRUCTURED_OUTPUT_TOOL_NAME);
+}
+function toRecord(value) {
+  if (value && typeof value === "object" && !Array.isArray(value)) {
+    return value;
+  }
+  return null;
+}
 var AUTO_APPROVED_KINDS = {
   default: /* @__PURE__ */ new Set(["read", "search", "fetch", "think"]),
   acceptEdits: /* @__PURE__ */ new Set(["read", "edit", "search", "fetch", "think"]),
@@ -17882,6 +17899,7 @@ function shouldAutoApprove(mode, kind) {
 }
 function createCodexClient(upstreamClient, logger, sessionState, callbacks) {
   const terminalHandles = /* @__PURE__ */ new Map();
+  const structuredOutputState = /* @__PURE__ */ new Map();
   return {
     async requestPermission(params) {
       const kind = params.toolCall?.kind;
@@ -17905,6 +17923,26 @@ function createCodexClient(upstreamClient, logger, sessionState, callbacks) {
     },
     async sessionUpdate(params) {
       const update = params.update;
+      if (callbacks?.onStructuredOutput && (update?.sessionUpdate === "tool_call" || update?.sessionUpdate === "tool_call_update")) {
+        const toolCallId = update.toolCallId;
+        const title = update.title;
+        if (toolCallId && isStructuredOutputToolCall(title)) {
+          const entry = structuredOutputState.get(toolCallId) ?? {
+            fired: false
+          };
+          const rawInput = toRecord(update.rawInput);
+          if (rawInput) entry.rawInput = rawInput;
+          structuredOutputState.set(toolCallId, entry);
+          if (update.status === "completed" && !entry.fired && entry.rawInput) {
+            entry.fired = true;
+            try {
+              await callbacks.onStructuredOutput(entry.rawInput);
+            } catch (err2) {
+              logger.warn("onStructuredOutput callback threw", { error: err2 });
+            }
+          }
+        }
+      }
       if (update?.sessionUpdate === "usage_update") {
         const used = update.used;
         const size = update.size;
@@ -18194,6 +18232,7 @@ function spawnCodexProcess(options) {
 }
 
 // src/adapters/codex/codex-agent.ts
+var import_meta2 = {};
 function toCodexPermissionMode(mode) {
   if (mode && (isCodexNativeMode(mode) || isCodeExecutionMode(mode))) {
     return mode;
@@ -18232,6 +18271,33 @@ function getCurrentPermissionMode(currentModeId, fallbackMode) {
   }
   return toCodexPermissionMode(fallbackMode);
 }
+var STRUCTURED_OUTPUT_INSTRUCTIONS = `
+
+When you have completed the task, call the \`${STRUCTURED_OUTPUT_TOOL_NAME}\` tool with the final structured result. The tool's input schema matches the required output format for this task. Do not describe the result in a plain message \u2014 submitting it via the tool is required for the task to be considered complete.`;
+function resolveStructuredOutputMcpScript() {
+  const rel = "adapters/codex/structured-output-mcp-server.js";
+  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;
+    dir = (0, import_node_path5.resolve)(dir, "..");
+  }
+  throw new Error(
+    `Could not locate ${rel} relative to ${import_meta2.dirname ?? __dirname}.`
+  );
+}
+function buildStructuredOutputMcpServer(jsonSchema) {
+  const scriptPath = resolveStructuredOutputMcpScript();
+  const schemaBase64 = Buffer.from(JSON.stringify(jsonSchema)).toString(
+    "base64"
+  );
+  return {
+    name: STRUCTURED_OUTPUT_MCP_NAME,
+    command: process.execPath,
+    args: [scriptPath],
+    env: [{ name: "POSTHOG_OUTPUT_SCHEMA", value: schemaBase64 }]
+  };
+}
 var CodexAcpAgent = class extends BaseAcpAgent {
   adapterName = "codex";
   codexProcess;
@@ -18250,6 +18316,7 @@ var CodexAcpAgent = class extends BaseAcpAgent {
   promptMutex = Promise.resolve();
   codexProcessOptions;
   processCallbacks;
+  onStructuredOutput;
   // Snapshot of the initialize() request so refreshSession can replay the
   // same handshake against a respawned codex-acp subprocess.
   lastInitRequest;
@@ -18261,6 +18328,7 @@ var CodexAcpAgent = class extends BaseAcpAgent {
     const settingsManager = new CodexSettingsManager(cwd);
     this.codexProcessOptions = options.codexProcessOptions;
     this.processCallbacks = options.processCallbacks;
+    this.onStructuredOutput = options.onStructuredOutput;
     this.codexProcess = spawnCodexProcess({
       ...options.codexProcessOptions,
       settings: settingsManager.getSettings(),
@@ -18282,7 +18350,8 @@ var CodexAcpAgent = class extends BaseAcpAgent {
     this.enrichment = createEnrichment(options.posthogApiConfig, this.logger);
     this.codexConnection = new import_sdk3.ClientSideConnection(
       (_agent) => createCodexClient(this.client, this.logger, this.sessionState, {
-        enrichmentDeps: this.enrichment?.deps
+        enrichmentDeps: this.enrichment?.deps,
+        onStructuredOutput: this.onStructuredOutput
       }),
       codexStream
     );
@@ -18316,7 +18385,8 @@ var CodexAcpAgent = class extends BaseAcpAgent {
   async newSession(params) {
     const meta = params._meta;
     const requestedPermissionMode = toCodexPermissionMode(meta?.permissionMode);
-    const response = await this.codexConnection.newSession(params);
+    const injectedParams = this.applyStructuredOutput(params, meta);
+    const response = await this.codexConnection.newSession(injectedParams);
     response.configOptions = normalizeCodexConfigOptions(
       response.configOptions
     );
@@ -18348,11 +18418,12 @@ var CodexAcpAgent = class extends BaseAcpAgent {
     return response;
   }
   async loadSession(params) {
-    const response = await this.codexConnection.loadSession(params);
+    const meta = params._meta;
+    const injectedParams = this.applyStructuredOutput(params, meta);
+    const response = await this.codexConnection.loadSession(injectedParams);
     response.configOptions = normalizeCodexConfigOptions(
       response.configOptions
     );
-    const meta = params._meta;
     const currentPermissionMode = getCurrentPermissionMode(
       response.modes?.currentModeId,
       meta?.permissionMode
@@ -18375,15 +18446,20 @@ var CodexAcpAgent = class extends BaseAcpAgent {
     return response;
   }
   async unstable_resumeSession(params) {
-    const loadResponse = await this.codexConnection.loadSession({
-      sessionId: params.sessionId,
-      cwd: params.cwd,
-      mcpServers: params.mcpServers ?? []
-    });
+    const meta = params._meta;
+    const injectedParams = this.applyStructuredOutput(
+      {
+        sessionId: params.sessionId,
+        cwd: params.cwd,
+        mcpServers: params.mcpServers ?? [],
+        _meta: params._meta
+      },
+      meta
+    );
+    const loadResponse = await this.codexConnection.loadSession(injectedParams);
     loadResponse.configOptions = normalizeCodexConfigOptions(
       loadResponse.configOptions
     );
-    const meta = params._meta;
     const currentPermissionMode = getCurrentPermissionMode(
       loadResponse.modes?.currentModeId,
       meta?.permissionMode
@@ -18410,15 +18486,19 @@ var CodexAcpAgent = class extends BaseAcpAgent {
     };
   }
   async unstable_forkSession(params) {
-    const newResponse = await this.codexConnection.newSession({
-      cwd: params.cwd,
-      mcpServers: params.mcpServers ?? [],
-      _meta: params._meta
-    });
+    const meta = params._meta;
+    const injectedParams = this.applyStructuredOutput(
+      {
+        cwd: params.cwd,
+        mcpServers: params.mcpServers ?? [],
+        _meta: params._meta
+      },
+      meta
+    );
+    const newResponse = await this.codexConnection.newSession(injectedParams);
     newResponse.configOptions = normalizeCodexConfigOptions(
       newResponse.configOptions
     );
-    const meta = params._meta;
     const requestedPermissionMode = toCodexPermissionMode(meta?.permissionMode);
     this.sessionState = createSessionState(newResponse.sessionId, params.cwd, {
       taskRunId: meta?.taskRunId,
@@ -18435,6 +18515,30 @@ var CodexAcpAgent = class extends BaseAcpAgent {
     );
     return newResponse;
   }
+  /**
+   * When the caller wires up `onStructuredOutput` and provides a JSON schema
+   * via `_meta.jsonSchema`, inject the stdio MCP server that exposes
+   * `create_output` and append instructions telling the model to use it.
+   *
+   * Codex has no native equivalent of Claude's `outputFormat`, so we lean on
+   * MCP tool-calling to get validated structured output back.
+   */
+  applyStructuredOutput(request, meta) {
+    if (!meta?.jsonSchema || !this.onStructuredOutput) {
+      return request;
+    }
+    const mcpServer = buildStructuredOutputMcpServer(meta.jsonSchema);
+    const existingMeta = request._meta ?? {};
+    const existingSystemPrompt = typeof existingMeta.systemPrompt === "string" ? existingMeta.systemPrompt : "";
+    return {
+      ...request,
+      mcpServers: [...request.mcpServers ?? [], mcpServer],
+      _meta: {
+        ...existingMeta,
+        systemPrompt: existingSystemPrompt + STRUCTURED_OUTPUT_INSTRUCTIONS
+      }
+    };
+  }
   async applyInitialPermissionMode(sessionId, permissionMode, currentModeId) {
     if (!permissionMode) {
       return;
@@ -18593,7 +18697,9 @@ var CodexAcpAgent = class extends BaseAcpAgent {
     const codexStream = (0, import_sdk3.ndJsonStream)(codexWritable, codexReadable);
     const newAbortController = new AbortController();
     const newConnection = new import_sdk3.ClientSideConnection(
-      (_agent) => createCodexClient(this.client, this.logger, this.sessionState),
+      (_agent) => createCodexClient(this.client, this.logger, this.sessionState, {
+        onStructuredOutput: this.onStructuredOutput
+      }),
       codexStream
     );
     const initRequest = this.lastInitRequest ?? {
@@ -18773,7 +18879,8 @@ function createCodexConnection(config) {
     agent = new CodexAcpAgent(client, {
       codexProcessOptions: config.codexOptions ?? {},
       processCallbacks: config.processCallbacks,
-      posthogApiConfig: resolveEnricherApiConfig(config)
+      posthogApiConfig: resolveEnricherApiConfig(config),
+      onStructuredOutput: config.onStructuredOutput
     });
     return agent;
   }, agentStream);
@@ -18803,13 +18910,13 @@ function createCodexConnection(config) {
 // src/handoff-checkpoint.ts
 var import_promises3 = require("fs/promises");
 var import_node_os2 = require("os");
-var import_node_path6 = require("path");
+var import_node_path7 = require("path");
 
 // ../git/dist/handoff.js
 var import_node_child_process4 = require("child_process");
 var import_promises2 = require("fs/promises");
 var import_node_os = require("os");
-var import_node_path5 = __toESM(require("path"), 1);
+var import_node_path6 = __toESM(require("path"), 1);
 
 // ../git/dist/sagas/checkpoint.js
 var import_node_crypto2 = require("crypto");
@@ -19310,7 +19417,7 @@ var GitHandoffTracker = class {
         packBaseline ? `^${packBaseline}` : null
       ].filter((ref) => !!ref);
       const headRef = checkpoint.head ? `${HANDOFF_HEAD_REF_PREFIX}${checkpoint.checkpointId}` : void 0;
-      const packPrefix = import_node_path5.default.join(tempDir, checkpoint.checkpointId);
+      const packPrefix = import_node_path6.default.join(tempDir, checkpoint.checkpointId);
       const [headPack, indexFile, tracking] = await Promise.all([
         this.captureObjectPack(packPrefix, packRefs),
         this.statFileArtifact(reconciledIndex.indexFilePath),
@@ -19385,7 +19492,7 @@ var GitHandoffTracker = class {
   }
   async reconcileHandoffIndex(git, head, indexTree, tempDir, checkpointId) {
     const realIndexPath = await this.getGitPath(git, "index");
-    const tempIndexPath = import_node_path5.default.join(tempDir, `${checkpointId}.index`);
+    const tempIndexPath = import_node_path6.default.join(tempDir, `${checkpointId}.index`);
     await (0, import_promises2.copyFile)(realIndexPath, tempIndexPath);
     const largePaths = await this.listLargeBlobsInTree(indexTree, MAX_HANDOFF_FILE_BYTES);
     if (largePaths.length === 0) {
@@ -19607,7 +19714,7 @@ var GitHandoffTracker = class {
   async getGitPath(git, gitPath) {
     const raw = await git.raw(["rev-parse", "--git-path", gitPath]);
     const resolved = raw.trim();
-    return import_node_path5.default.isAbsolute(resolved) ? resolved : import_node_path5.default.resolve(this.repositoryPath, resolved);
+    return import_node_path6.default.isAbsolute(resolved) ? resolved : import_node_path6.default.resolve(this.repositoryPath, resolved);
   }
   async getFileSize(filePath) {
     return (await (0, import_promises2.stat)(filePath)).size;
@@ -19697,7 +19804,7 @@ var GitHandoffTracker = class {
   }
 };
 function joinTempPrefix(checkpointId) {
-  return import_node_path5.default.join((0, import_node_os.tmpdir)(), `posthog-code-handoff-${checkpointId}-`);
+  return import_node_path6.default.join((0, import_node_os.tmpdir)(), `posthog-code-handoff-${checkpointId}-`);
 }
 async function getCurrentBranchName(git) {
   try {
@@ -19785,7 +19892,7 @@ var HandoffCheckpointTracker = class {
         indexArtifactPath: uploads.index?.storagePath
       };
     } finally {
-      const tempDir = capture.headPack?.path ? (0, import_node_path6.dirname)(capture.headPack.path) : (0, import_node_path6.dirname)(capture.indexFile.path);
+      const tempDir = capture.headPack?.path ? (0, import_node_path7.dirname)(capture.headPack.path) : (0, import_node_path7.dirname)(capture.indexFile.path);
       await this.removeIfPresent(capture.headPack?.path);
       await this.removeIfPresent(capture.indexFile.path);
       await (0, import_promises3.rm)(tempDir, { recursive: true, force: true }).catch(() => {
@@ -19800,10 +19907,10 @@ var HandoffCheckpointTracker = class {
     }
     const gitTracker = this.createGitTracker();
     const tmpDir = await (0, import_promises3.mkdtemp)(
-      (0, import_node_path6.join)((0, import_node_os2.tmpdir)(), `posthog-code-handoff-${checkpoint.checkpointId}-`)
+      (0, import_node_path7.join)((0, import_node_os2.tmpdir)(), `posthog-code-handoff-${checkpoint.checkpointId}-`)
     );
-    const packPath = (0, import_node_path6.join)(tmpDir, `${checkpoint.checkpointId}.pack`);
-    const indexPath = (0, import_node_path6.join)(tmpDir, `${checkpoint.checkpointId}.index`);
+    const packPath = (0, import_node_path7.join)(tmpDir, `${checkpoint.checkpointId}.pack`);
+    const indexPath = (0, import_node_path7.join)(tmpDir, `${checkpoint.checkpointId}.index`);
     try {
       const downloads = await this.downloadArtifacts([
         {
@@ -20491,9 +20598,9 @@ ${toolSummary}`);
 }
 
 // src/session-log-writer.ts
-var import_node_fs3 = __toESM(require("fs"), 1);
+var import_node_fs4 = __toESM(require("fs"), 1);
 var import_promises4 = __toESM(require("fs/promises"), 1);
-var import_node_path7 = __toESM(require("path"), 1);
+var import_node_path8 = __toESM(require("path"), 1);
 var SessionLogWriter = class _SessionLogWriter {
   static FLUSH_DEBOUNCE_MS = 500;
   static FLUSH_MAX_INTERVAL_MS = 5e3;
@@ -20529,13 +20636,13 @@ var SessionLogWriter = class _SessionLogWriter {
     this.sessions.set(sessionId, { context, currentTurnMessages: [] });
     this.lastFlushAttemptTime.set(sessionId, Date.now());
     if (this.localCachePath) {
-      const sessionDir = import_node_path7.default.join(
+      const sessionDir = import_node_path8.default.join(
         this.localCachePath,
         "sessions",
         context.runId
       );
       try {
-        import_node_fs3.default.mkdirSync(sessionDir, { recursive: true });
+        import_node_fs4.default.mkdirSync(sessionDir, { recursive: true });
       } catch (error) {
         this.logger.warn("Failed to create local cache directory", {
           sessionDir,
@@ -20787,14 +20894,14 @@ var SessionLogWriter = class _SessionLogWriter {
     if (!this.localCachePath) return;
     const session = this.sessions.get(sessionId);
     if (!session) return;
-    const logPath = import_node_path7.default.join(
+    const logPath = import_node_path8.default.join(
       this.localCachePath,
       "sessions",
       session.context.runId,
       "logs.ndjson"
     );
     try {
-      import_node_fs3.default.appendFileSync(logPath, `${JSON.stringify(entry)}
+      import_node_fs4.default.appendFileSync(logPath, `${JSON.stringify(entry)}
 `);
     } catch (error) {
       this.logger.warn("Failed to write to local cache", {
@@ -20806,13 +20913,13 @@ var SessionLogWriter = class _SessionLogWriter {
     }
   }
   static async cleanupOldSessions(localCachePath) {
-    const sessionsDir = import_node_path7.default.join(localCachePath, "sessions");
+    const sessionsDir = import_node_path8.default.join(localCachePath, "sessions");
     let deleted = 0;
     try {
       const entries = await import_promises4.default.readdir(sessionsDir);
       const now = Date.now();
       for (const entry of entries) {
-        const entryPath = import_node_path7.default.join(sessionsDir, entry);
+        const entryPath = import_node_path8.default.join(sessionsDir, entry);
         try {
           const stats = await import_promises4.default.stat(entryPath);
           if (stats.isDirectory() && now - stats.birthtimeMs > _SessionLogWriter.SESSIONS_MAX_AGE_MS) {
@@ -22104,7 +22211,7 @@ Continue from where you left off. The user is waiting for your response.`
       throw new Error(`Failed to download artifact ${artifact.name}`);
     }
     const safeName = this.getSafeArtifactName(artifact.name);
-    const artifactDir = (0, import_node_path8.join)(
+    const artifactDir = (0, import_node_path9.join)(
       this.config.repositoryPath ?? "/tmp/workspace",
       ".posthog",
       "attachments",
@@ -22112,7 +22219,7 @@ Continue from where you left off. The user is waiting for your response.`
       artifact.id ?? safeName
     );
     await (0, import_promises6.mkdir)(artifactDir, { recursive: true });
-    const artifactPath = (0, import_node_path8.join)(artifactDir, safeName);
+    const artifactPath = (0, import_node_path9.join)(artifactDir, safeName);
     await (0, import_promises6.writeFile)(artifactPath, Buffer.from(data));
     return resourceLink((0, import_node_url2.pathToFileURL)(artifactPath).toString(), artifact.name, {
       ...artifact.content_type ? { mimeType: artifact.content_type } : {},
@@ -22120,7 +22227,7 @@ Continue from where you left off. The user is waiting for your response.`
     });
   }
   getSafeArtifactName(name2) {
-    const baseName = (0, import_node_path8.basename)(name2).trim();
+    const baseName = (0, import_node_path9.basename)(name2).trim();
     const normalizedName = baseName.replace(/[^\w.-]/g, "_");
     return normalizedName.length > 0 ? normalizedName : "attachment";
   }