@posthog/agent 2.3.233 → 2.3.259

package/dist/server/bin.cjs

@@ -5683,7 +5683,7 @@ var import_hono = require("hono");
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.233",
+  version: "2.3.259",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -5779,6 +5779,7 @@ var package_default = {
   },
   dependencies: {
     "@agentclientprotocol/sdk": "0.16.1",
+    ajv: "^8.17.1",
     "@anthropic-ai/claude-agent-sdk": "0.2.76",
     "@anthropic-ai/sdk": "^0.78.0",
     "@hono/node-server": "^1.19.9",
@@ -8630,6 +8631,7 @@ function buildSessionOptions(params) {
       params.settingsManager,
       params.logger
     ),
+    outputFormat: params.outputFormat,
     abortController: getAbortController(
       params.userProvidedOptions?.abortController
     ),
@@ -9217,6 +9219,11 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
             };
             const result = handleResultMessage(message);
             if (result.error) throw result.error;
+            if (message.subtype === "success" && message.structured_output != null && this.options?.onStructuredOutput) {
+              await this.options.onStructuredOutput(
+                message.structured_output
+              );
+            }
             if (isLocalOnlyCommand && message.subtype === "success" && message.result) {
               await this.client.sessionUpdate({
                 sessionId: params.sessionId,
@@ -9442,6 +9449,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
     const earlyModelId = settingsManager.getSettings().model || meta?.model || "";
     const mcpServers = supportsMcpInjection(earlyModelId) ? parseMcpServers(params) : {};
     const systemPrompt = buildSystemPrompt(meta?.systemPrompt);
+    const outputFormat = meta?.jsonSchema && this.options?.onStructuredOutput ? { type: "json_schema", schema: meta.jsonSchema } : void 0;
     this.logger.info(isResume ? "Resuming session" : "Creating new session", {
       sessionId,
       taskId,
@@ -9465,6 +9473,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
         ...meta?.additionalRoots ?? []
       ],
       disableBuiltInTools: meta?.disableBuiltInTools,
+      outputFormat,
       settingsManager,
       onModeChange: this.createOnModeChange(),
       onProcessSpawned: this.options?.onProcessSpawned,
@@ -10424,7 +10433,10 @@ function createClaudeConnection(config) {
   const agentStream = (0, import_sdk4.ndJsonStream)(agentWritable, streams.agent.readable);
   let agent = null;
   const agentConnection = new import_sdk4.AgentSideConnection((client) => {
-    agent = new ClaudeAcpAgent(client, config.processCallbacks);
+    agent = new ClaudeAcpAgent(client, {
+      ...config.processCallbacks,
+      onStructuredOutput: config.onStructuredOutput
+    });
     logger.info(`Created ${agent.adapterName} agent`);
     return agent;
   }, agentStream);
@@ -10656,6 +10668,15 @@ var PostHogAPIClient = class {
       }
     );
   }
+  async setTaskRunOutput(taskId, runId, output) {
+    return this.apiRequest(
+      `/api/projects/${this.getTeamId()}/tasks/${taskId}/runs/${runId}/set_output/`,
+      {
+        method: "PATCH",
+        body: JSON.stringify(output)
+      }
+    );
+  }
   async appendTaskRunLog(taskId, runId, entries) {
     const teamId = this.getTeamId();
     return this.apiRequest(
@@ -12704,7 +12725,16 @@ You MUST NOT create a new branch, close the existing PR, or create a new PR.`
       taskRunId: payload.run_id,
       taskId: payload.task_id,
       deviceType: deviceInfo.type,
-      logWriter
+      logWriter,
+      onStructuredOutput: async (output) => {
+        await this.posthogAPI.setTaskRunOutput(
+          payload.task_id,
+          payload.run_id,
+          {
+            output
+          }
+        );
+      }
     });
     const onAcpMessage = (message) => {
       this.broadcastEvent({
@@ -12732,18 +12762,23 @@ You MUST NOT create a new branch, close the existing PR, or create a new PR.`
       protocolVersion: import_sdk5.PROTOCOL_VERSION,
       clientCapabilities: {}
     });
-    let preTaskRun = null;
-    try {
-      preTaskRun = await this.posthogAPI.getTaskRun(
-        payload.task_id,
-        payload.run_id
-      );
-    } catch {
-      this.logger.warn("Failed to fetch task run for session context", {
-        taskId: payload.task_id,
-        runId: payload.run_id
-      });
-    }
+    const [preTaskRun, preTask] = await Promise.all([
+      this.posthogAPI.getTaskRun(payload.task_id, payload.run_id).catch((err) => {
+        this.logger.warn("Failed to fetch task run for session context", {
+          taskId: payload.task_id,
+          runId: payload.run_id,
+          error: err
+        });
+        return null;
+      }),
+      this.posthogAPI.getTask(payload.task_id).catch((err) => {
+        this.logger.warn("Failed to fetch task for session context", {
+          taskId: payload.task_id,
+          error: err
+        });
+        return null;
+      })
+    ]);
     const prUrl = typeof preTaskRun?.state?.slack_notified_pr_url === "string" ? (preTaskRun?.state).slack_notified_pr_url : null;
     if (prUrl) {
       this.detectedPrUrl = prUrl;
@@ -12756,6 +12791,7 @@ You MUST NOT create a new branch, close the existing PR, or create a new PR.`
         taskRunId: payload.run_id,
         systemPrompt: this.buildSessionSystemPrompt(prUrl),
         allowedDomains: this.config.allowedDomains,
+        jsonSchema: preTask?.json_schema ?? null,
         ...this.config.claudeCode?.plugins?.length && {
           claudeCode: {
             options: {