@posthog/agent 2.1.13 → 2.1.15

package/dist/server/agent-server.d.ts

@@ -29,9 +29,11 @@ declare class AgentServer {
     private executeCommand;
     private initializeSession;
     private sendInitialTaskMessage;
+    private buildCloudSystemPrompt;
     private signalTaskComplete;
     private configureEnvironment;
     private createCloudClient;
+    private detectAndAttachPrUrl;
     private cleanupSession;
     private captureTreeState;
     private broadcastEvent;

package/dist/server/agent-server.js

@@ -1183,7 +1183,7 @@ import { v7 as uuidv7 } from "uuid";
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.1.13",
+  version: "2.1.15",
   repository: "https://github.com/PostHog/twig",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -10310,7 +10310,11 @@ var AgentServer = class {
     await clientConnection.newSession({
       cwd: this.config.repositoryPath,
       mcpServers: [],
-      _meta: { sessionId: payload.run_id }
+      _meta: {
+        sessionId: payload.run_id,
+        taskRunId: payload.run_id,
+        systemPrompt: { append: this.buildCloudSystemPrompt() }
+      }
     });
     this.session = {
       payload,
@@ -10322,6 +10326,11 @@ var AgentServer = class {
       logWriter
     };
     this.logger.info("Session initialized successfully");
+    this.posthogAPI.updateTaskRun(payload.task_id, payload.run_id, {
+      status: "in_progress"
+    }).catch(
+      (err) => this.logger.warn("Failed to set task run to in_progress", err)
+    );
     await this.sendInitialTaskMessage(payload);
   }
   async sendInitialTaskMessage(payload) {
@@ -10358,6 +10367,19 @@ var AgentServer = class {
       }
     }
   }
+  buildCloudSystemPrompt() {
+    return `
+# Cloud Task Execution
+
+After completing the requested changes:
+1. Create a new branch with a descriptive name based on the work done
+2. Stage and commit all changes with a clear commit message
+3. Push the branch to origin
+4. Create a pull request using \`gh pr create\` with a descriptive title and body
+
+Important: Always create the PR. Do not ask for confirmation.
+`;
+  }
   async signalTaskComplete(payload, stopReason) {
     const status = stopReason === "cancelled" ? "cancelled" : stopReason === "error" ? "failed" : "completed";
     try {
@@ -10418,10 +10440,70 @@ var AgentServer = class {
           if ((toolName === "Write" || toolName === "Edit") && toolResponse?.filePath) {
             await this.captureTreeState();
           }
+          if (toolName && (toolName.includes("Bash") || toolName.includes("bash"))) {
+            this.detectAndAttachPrUrl(payload, params.update);
+          }
         }
       }
     };
   }
+  detectAndAttachPrUrl(payload, update) {
+    try {
+      const meta = update?._meta?.claudeCode;
+      const toolResponse = meta?.toolResponse;
+      let textToSearch = "";
+      if (toolResponse) {
+        if (typeof toolResponse === "string") {
+          textToSearch = toolResponse;
+        } else if (typeof toolResponse === "object" && toolResponse !== null) {
+          const respObj = toolResponse;
+          textToSearch = String(respObj.stdout || "") + String(respObj.stderr || "");
+          if (!textToSearch && respObj.output) {
+            textToSearch = String(respObj.output);
+          }
+        }
+      }
+      const content = update?.content;
+      if (Array.isArray(content)) {
+        for (const item of content) {
+          if (item.type === "text" && item.text) {
+            textToSearch += ` ${item.text}`;
+          }
+        }
+      }
+      if (!textToSearch) return;
+      const prUrlMatch = textToSearch.match(
+        /https:\/\/github\.com\/[^/]+\/[^/]+\/pull\/\d+/
+      );
+      if (!prUrlMatch) return;
+      const prUrl = prUrlMatch[0];
+      this.logger.info("Detected PR URL in bash output", {
+        runId: payload.run_id,
+        prUrl
+      });
+      this.posthogAPI.updateTaskRun(payload.task_id, payload.run_id, {
+        output: { pr_url: prUrl }
+      }).then(() => {
+        this.logger.info("PR URL attached to task run", {
+          taskId: payload.task_id,
+          runId: payload.run_id,
+          prUrl
+        });
+      }).catch((err) => {
+        this.logger.error("Failed to attach PR URL to task run", {
+          taskId: payload.task_id,
+          runId: payload.run_id,
+          prUrl,
+          error: err
+        });
+      });
+    } catch (err) {
+      this.logger.debug("Error in PR URL detection", {
+        runId: payload.run_id,
+        error: err
+      });
+    }
+  }
   async cleanupSession() {
     if (!this.session) return;
     this.logger.info("Cleaning up session");