refactor: let Auggie agent create PR instead of action code

- Remove createPRForChanges function
- Update invokeAuggie to accept shouldCreatePR parameter
- For /pr command, instruct Auggie to create branch, commit, and create PR using its tools
- This avoids needing PAT_TOKEN since Auggie has GitHub API access
- Simplifies setup and makes it work out of the box

Author: Tulga Tsogtgerel

assistant/src/index.ts

@@ -250,87 +250,10 @@ Automatically generated by Auggie Assistant`;
   core.info('✅ Changes committed and pushed successfully');
 }
 
-/**
- * Create a new branch, commit changes, and create a PR targeting the original PR
- */
-async function createPRForChanges(
-  commentId: number,
-  headBranch: string,
-  githubToken: string,
-  owner: string,
-  repo: string,
-  prNumber: number,
-  octokit: Octokit
-): Promise<{ newBranch: string; newPrNumber: number }> {
-  core.info('📝 Creating new branch for PR...');
-
-  // Stage all changes
-  await exec.exec('git', ['add', '-A']);
-
-  // Check if there are changes to commit using git status
-  let statusOutput = '';
-  await exec.exec('git', ['status', '--porcelain'], {
-    listeners: {
-      stdout: (data: Buffer) => {
-        statusOutput += data.toString();
-      },
-    },
-  });
-
-  const hasChanges = statusOutput.trim().length > 0;
-
-  if (!hasChanges) {
-    core.info('ℹ️ No changes to commit');
-    throw new Error('No changes to create PR with');
-  }
-
-  core.info(`📝 Changes detected:\n${statusOutput}`);
-
-  // Create a new branch name based on the comment ID
-  const newBranch = `auggie/pr-${prNumber}-comment-${commentId}`;
-  core.info(`🌿 Creating new branch: ${newBranch}`);
-
-  // Create and checkout new branch
-  await exec.exec('git', ['checkout', '-b', newBranch]);
-
-  // Commit changes
-  const commitMessage = `feat: implement changes requested in comment #${commentId}
-
-Automatically generated by Auggie Assistant`;
-
-  await exec.exec('git', ['commit', '-m', commitMessage]);
-
-  // Push new branch
-  core.info(`🚀 Pushing to ${newBranch}...`);
-  const remoteUrl = `https://x-access-token:${githubToken}@github.com/${owner}/${repo}.git`;
-  await exec.exec('git', ['push', remoteUrl, `HEAD:${newBranch}`]);
-
-  core.info('✅ Changes committed and pushed to new branch');
-
-  // Create PR targeting the original PR's branch
-  core.info(`📬 Creating PR targeting ${headBranch}...`);
-  const { data: newPr } = await octokit.rest.pulls.create({
-    owner,
-    repo,
-    title: `Changes from comment #${commentId} on PR #${prNumber}`,
-    body: `This PR contains changes requested in [comment #${commentId}](https://github.com/${owner}/${repo}/pull/${prNumber}#issuecomment-${commentId}) on PR #${prNumber}.
-
-Automatically generated by Auggie Assistant.
-
-**Target Branch:** \`${headBranch}\` (from PR #${prNumber})`,
-    head: newBranch,
-    base: headBranch,
-  });
-
-  core.info(`✅ Created PR #${newPr.number}`);
-
-  return { newBranch, newPrNumber: newPr.number };
-}
-
 /**
  * Invoke Auggie to implement changes
  */
-async function invokeAuggie(context: PRContext): Promise<void> {
+async function invokeAuggie(context: PRContext, shouldCreatePR: boolean): Promise<void> {
   core.info('🤖 Invoking Auggie to implement changes...');
 
   const augmentApiToken = process.env.AUGMENT_API_TOKEN;
@@ -343,7 +266,7 @@ async function invokeAuggie(context: PRContext): Promise<void> {
   }
 
   // Build context for Auggie
-  const instruction = `You are an AI assistant helping to implement code changes in a Pull Request.
+  const baseInstruction = `You are an AI assistant helping to implement code changes in a Pull Request.
 
 ## User Request
 ${context.commentBody}
@@ -367,7 +290,23 @@ ${context.diff}
 
 ## Instructions
 You MUST actually implement the requested changes by editing the files. Do NOT just describe what you would do.
-Use the str-replace-editor, save-file, or other file editing tools to make the actual changes to the code.
+Use the str-replace-editor, save-file, or other file editing tools to make the actual changes to the code.`;
+
+  const instruction = shouldCreatePR
+    ? `${baseInstruction}
+
+After making the changes, you MUST:
+1. Create a new branch named "auggie/pr-${context.prNumber}-comment-{commentId}" (use the actual comment ID from the context)
+2. Commit all your changes to this new branch with message: "feat: implement changes requested in comment #{commentId}"
+3. Push the new branch to origin
+4. Create a PR using the github-api tool targeting the "${context.headBranch}" branch with:
+   - Title: "Changes from comment on PR #${context.prNumber}"
+   - Body: A description of what you implemented
+5. Provide the PR number in your response
+
+Use git commands and the github-api tool to accomplish this.`
+    : `${baseInstruction}
+
 After making the changes, provide a brief summary of what you implemented.`;
 
   core.info('📤 Sending request to Auggie...');
@@ -521,68 +460,35 @@ async function main(): Promise<void> {
 
     // Step 5: Invoke Auggie to implement changes
     try {
-      await invokeAuggie(context);
-
-      // Step 6: Commit and push changes (different behavior for /pr vs /commit)
-      if (isPrCommand) {
-        // For /pr: create a new branch and PR targeting the original PR
-        const { newPrNumber } = await createPRForChanges(
-          commentId,
-          context.headBranch,
-          githubToken,
-          owner,
-          repo,
-          prNumber,
-          octokit
-        );
-
-        // Step 7: Add success comment to original PR
-        const quotedComment = context.commentBody
-          .split('\n')
-          .map(line => `> ${line}`)
-          .join('\n');
-
-        await addPRComment(
-          octokit,
-          owner,
-          repo,
-          prNumber,
-          `${quotedComment}
+      await invokeAuggie(context, isPrCommand);
 
-✅ Successfully implemented the requested changes!
-
-Created PR #${newPrNumber} with the changes targeting \`${context.headBranch}\`.
+      // Step 6: Commit and push changes (only for /commit, /pr is handled by Auggie)
+      if (!isPrCommand) {
+        await commitAndPush(commentId, context.headBranch, githubToken, owner, repo);
+      }
 
-👉 [View the new PR](https://github.com/${owner}/${repo}/pull/${newPrNumber})`
-        );
+      // Step 7: Add success comment to PR
+      const quotedComment = context.commentBody
+        .split('\n')
+        .map(line => `> ${line}`)
+        .join('\n');
 
-        core.info(`✨ PR Assistant completed successfully - created PR #${newPrNumber}`);
-      } else {
-        // For /commit: commit directly to the PR branch (existing behavior)
-        await commitAndPush(commentId, context.headBranch, githubToken, owner, repo);
+      const successMessage = isPrCommand
+        ? `${quotedComment}
 
-        // Step 7: Add success comment to PR
-        const quotedComment = context.commentBody
-          .split('\n')
-          .map(line => `> ${line}`)
-          .join('\n');
+✅ Successfully implemented the requested changes!
 
-        await addPRComment(
-          octokit,
-          owner,
-          repo,
-          prNumber,
-          `${quotedComment}
+The agent has created a new PR with the changes targeting \`${context.headBranch}\`. Check the agent's response above for the PR link.`
+        : `${quotedComment}
 
 ✅ Successfully implemented the requested changes!
 
-The changes have been committed to the \`${context.headBranch}\` branch.`
-        );
+The changes have been committed to the \`${context.headBranch}\` branch.`;
 
-        core.info('✨ PR Assistant completed successfully');
-      }
+      await addPRComment(octokit, owner, repo, prNumber, successMessage);
 
       core.setOutput('success', 'true');
+      core.info('✨ PR Assistant completed successfully');
     } catch (error) {
       // Add failure comment to PR
       const quotedComment = context.commentBody