@posthog/agent 2.3.256 → 2.3.261

package/dist/agent.js

@@ -245,7 +245,7 @@ import { v7 as uuidv7 } from "uuid";
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.256",
+  version: "2.3.261",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -341,6 +341,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",
@@ -2984,6 +2985,7 @@ function buildSessionOptions(params) {
       params.settingsManager,
       params.logger
     ),
+    outputFormat: params.outputFormat,
     abortController: getAbortController(
       params.userProvidedOptions?.abortController
     ),
@@ -3571,6 +3573,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,
@@ -3796,6 +3803,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,
@@ -3819,6 +3827,7 @@ var ClaudeAcpAgent = class extends BaseAcpAgent {
         ...meta?.additionalRoots ?? []
       ],
       disableBuiltInTools: meta?.disableBuiltInTools,
+      outputFormat,
       settingsManager,
       onModeChange: this.createOnModeChange(),
       onProcessSpawned: this.options?.onProcessSpawned,
@@ -4781,7 +4790,10 @@ function createClaudeConnection(config) {
   const agentStream = ndJsonStream2(agentWritable, streams.agent.readable);
   let agent = null;
   const agentConnection = new 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);
@@ -4974,6 +4986,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(
@@ -5499,6 +5520,7 @@ var Agent = class {
       deviceType: "local",
       logger: this.logger,
       processCallbacks: options.processCallbacks,
+      onStructuredOutput: options.onStructuredOutput,
       allowedModelIds,
       codexOptions: options.adapter === "codex" && gatewayConfig ? {
         cwd: options.repositoryPath,