@posthog/agent 2.3.213 → 2.3.216

package/package.json

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

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;
   }