@posthog/agent 2.1.156 → 2.1.157

package/package.json

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

package/src/server/agent-server.test.ts

@@ -384,5 +384,33 @@ describe("AgentServer HTTP Mode", () => {
       expect(prompt).toContain("Create a draft pull request");
       expect(prompt).toContain("gh pr create --draft");
     });
+
+    it("includes --base flag when baseBranch is configured", () => {
+      server = new AgentServer({
+        port,
+        jwtPublicKey: TEST_PUBLIC_KEY,
+        repositoryPath: repo.path,
+        apiUrl: "http://localhost:8000",
+        apiKey: "test-api-key",
+        projectId: 1,
+        mode: "interactive",
+        taskId: "test-task-id",
+        runId: "test-run-id",
+        baseBranch: "add-yolo-to-readme",
+      });
+      const prompt = (
+        server as unknown as TestableServer
+      ).buildCloudSystemPrompt();
+      expect(prompt).toContain(
+        "gh pr create --draft --base add-yolo-to-readme",
+      );
+    });
+
+    it("omits --base flag when baseBranch is not configured", () => {
+      const s = createServer();
+      const prompt = (s as unknown as TestableServer).buildCloudSystemPrompt();
+      expect(prompt).toContain("gh pr create --draft`");
+      expect(prompt).not.toContain("--base");
+    });
   });
 });

package/src/server/agent-server.ts

@@ -747,7 +747,7 @@ 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 draft pull request using \`gh pr create --draft\` with a descriptive title and body
+4. Create a draft pull request using \`gh pr create --draft${this.config.baseBranch ? ` --base ${this.config.baseBranch}` : ""}\` with a descriptive title and body
 
 Important:
 - Always create the PR as a draft. Do not ask for confirmation.

package/src/server/bin.ts

@@ -50,6 +50,7 @@ program
     "--mcpServers <json>",
     "MCP servers config as JSON array (ACP McpServer[] format)",
   )
+  .option("--baseBranch <branch>", "Base branch for PR creation")
   .action(async (options) => {
     const envResult = envSchema.safeParse(process.env);
 
@@ -99,6 +100,7 @@ program
       taskId: options.taskId,
       runId: options.runId,
       mcpServers,
+      baseBranch: options.baseBranch,
     });
 
     process.on("SIGINT", async () => {

package/src/server/types.ts

@@ -13,4 +13,5 @@ export interface AgentServerConfig {
   runId: string;
   version?: string;
   mcpServers?: RemoteMcpServer[];
+  baseBranch?: string;
 }