From b0ce9a280e8b8b61c63824d43cf7b046e1f64212 Mon Sep 17 00:00:00 2001
From: km-anthropic <kashyap@anthropic.com>
Date: Thu, 21 Aug 2025 11:08:14 -0700
Subject: [PATCH] fix: Add branch environment variable support to agent mode
 for signed commits

- Read CLAUDE_BRANCH and BASE_BRANCH env vars in agent mode
- Pass correct branch info to MCP file ops server
- Enables signed auto-fix workflows to create branches via API
---
 src/modes/agent/index.ts | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/modes/agent/index.ts b/src/modes/agent/index.ts
index 03c3564..ecc33e2 100644
--- a/src/modes/agent/index.ts
+++ b/src/modes/agent/index.ts
@@ -81,8 +81,12 @@ export const agentMode: Mode = {
     const userClaudeArgs = process.env.CLAUDE_ARGS || "";
     const allowedTools = parseAllowedTools(userClaudeArgs);
 
+    // Check for branch info from environment variables (useful for auto-fix workflows)
+    const claudeBranch = process.env.CLAUDE_BRANCH || undefined;
+    const baseBranch = process.env.BASE_BRANCH || context.inputs.baseBranch || "main";
+    
     // Detect current branch from GitHub environment
-    const currentBranch =
+    const currentBranch = claudeBranch || 
       process.env.GITHUB_HEAD_REF || process.env.GITHUB_REF_NAME || "main";
 
     // Get our GitHub MCP servers config
@@ -91,7 +95,7 @@ export const agentMode: Mode = {
       owner: context.repository.owner,
       repo: context.repository.repo,
       branch: currentBranch,
-      baseBranch: context.inputs.baseBranch || "main",
+      baseBranch: baseBranch,
       claudeCommentId: undefined, // No tracking comment in agent mode
       allowedTools,
       context,
@@ -122,9 +126,9 @@ export const agentMode: Mode = {
     return {
       commentId: undefined,
       branchInfo: {
-        baseBranch: context.inputs.baseBranch || "main",
-        currentBranch,
-        claudeBranch: undefined,
+        baseBranch: baseBranch,
+        currentBranch: baseBranch, // Use base branch as current when creating new branch
+        claudeBranch: claudeBranch,
       },
       mcpConfig: ourMcpConfig,
     };