Add strict mode to prevent unrelated code changes

Author: paul-fracture

.github/workflows/claude-code.yml

@@ -32,7 +32,7 @@ jobs:
           echo "feedback=$FEEDBACK" >> $GITHUB_OUTPUT
       
       - name: Process with Claude Code
-        uses: fractureinc/claude-code-github-action@v0.2.0
+        uses: fractureinc/claude-code-github-action@v0.2.2
         with:
           mode: 'review'
           pr-number: ${{ steps.pr.outputs.number }}
@@ -64,7 +64,7 @@ jobs:
           echo "feedback=$FEEDBACK" >> $GITHUB_OUTPUT
       
       - name: Process with Claude Code Suggestions
-        uses: fractureinc/claude-code-github-action@v0.2.0
+        uses: fractureinc/claude-code-github-action@v0.2.2
         with:
           mode: 'suggest'
           pr-number: ${{ steps.pr.outputs.number }}
@@ -104,7 +104,7 @@ jobs:
           echo "line=$LINE" >> $GITHUB_OUTPUT
       
       - name: Process with Claude Code for code review comment
-        uses: fractureinc/claude-code-github-action@v0.2.0
+        uses: fractureinc/claude-code-github-action@v0.2.2
         with:
           mode: 'review'
           pr-number: ${{ steps.details.outputs.number }}
@@ -143,13 +143,14 @@ jobs:
           echo "line=$LINE" >> $GITHUB_OUTPUT
       
       - name: Process with Claude Code Suggestions for code review
-        uses: fractureinc/claude-code-github-action@v0.2.1
+        uses: fractureinc/claude-code-github-action@v0.2.2
         with:
           mode: 'suggest-review'
           pr-number: ${{ steps.details.outputs.number }}
           feedback: ${{ steps.details.outputs.feedback }}
           file-path: ${{ steps.details.outputs.file_path }}
           line-number: ${{ steps.details.outputs.line }}
           comment-id: ${{ steps.details.outputs.comment_id }}
+          strict-mode: 'true'
           anthropic-api-key: ${{ secrets.ANTHROPIC_API_KEY }}
           github-token: ${{ secrets.GITHUB_TOKEN }}

README.md

@@ -49,7 +49,7 @@ jobs:
           echo "feedback=$FEEDBACK" >> $GITHUB_OUTPUT
       
       - name: Process with Claude Code
-        uses: fractureinc/claude-code-github-action@v0.2.1
+        uses: fractureinc/claude-code-github-action@v0.2.2
         with:
           mode: 'review'
           pr-number: ${{ steps.pr.outputs.number }}
@@ -81,7 +81,7 @@ jobs:
           echo "feedback=$FEEDBACK" >> $GITHUB_OUTPUT
       
       - name: Process with Claude Code Suggestions
-        uses: fractureinc/claude-code-github-action@v0.2.1
+        uses: fractureinc/claude-code-github-action@v0.2.2
         with:
           mode: 'suggest'
           pr-number: ${{ steps.pr.outputs.number }}
@@ -94,18 +94,23 @@ jobs:
 
 | Input | Description | Required | Default |
 |-------|-------------|----------|---------|
-| `mode` | Operation mode (review, suggest, direct) | Yes | `review` |
+| `mode` | Operation mode (review, suggest, suggest-review, direct) | Yes | `review` |
 | `pr-number` | Pull request number | Yes* | |
 | `feedback` | User query text | Yes | |
+| `file-path` | Path to the file being reviewed (for suggest-review mode) | No** | |
+| `line-number` | Line number in the file (for suggest-review mode) | No** | |
+| `comment-id` | GitHub comment ID to reply to (for suggest-review mode) | No** | |
+| `strict-mode` | Whether to strictly follow user requests without adding unrelated improvements | No | `true` |
 | `anthropic-api-key` | Anthropic API key | Yes | |
 | `github-token` | GitHub token | Yes | |
 | `output-file` | Output file path (for direct mode) | No | `claude-code-output` |
 
-\* Required when mode is 'review' or 'suggest'
+\* Required when mode is 'review' or 'suggest'  
+\** Required when mode is 'suggest-review'
 
 ## Enhanced Context for Claude
 
-With version 0.2.1, Claude now receives complete context for your PRs, including:
+With version 0.2.2, Claude now receives complete context for your PRs, including:
 
 - PR metadata (title, description, branch info)
 - List of all files changed
@@ -127,6 +132,8 @@ Creates suggested changes in a PR comment that outline potential code improvemen
 
 Creates true GitHub-compatible suggestions that can be applied with one click directly from the code review interface. These are attached to specific lines of code.
 
+By default, this mode uses "strict mode" which ensures Claude only makes changes specifically related to what was requested, without adding unrelated improvements. You can disable strict mode by setting `strict-mode: 'false'` to allow Claude to suggest additional improvements.
+
 ### Direct Mode (`mode: 'direct'`)
 
 Sends a query directly to Claude and saves the response to a file without PR context.

action.yml

@@ -24,6 +24,10 @@ inputs:
   comment-id:
     description: 'GitHub comment ID to reply to (for suggest-review mode)'
     required: false
+  strict-mode:
+    description: 'Whether to strictly follow user requests without adding unrelated improvements'
+    required: false
+    default: 'true'
   anthropic-api-key:
     description: 'Anthropic API key for Claude access'
     required: true
@@ -79,4 +83,4 @@ runs:
       shell: bash
       run: |
         chmod +x ${{ github.action_path }}/scripts/suggest-review-mode.sh
-        ${{ github.action_path }}/scripts/suggest-review-mode.sh "${{ inputs.pr-number }}" "${{ inputs.feedback }}" "${{ inputs.file-path }}" "${{ inputs.line-number }}" "${{ inputs.comment-id }}" "${{ inputs.anthropic-api-key }}" "${{ inputs.github-token }}"
+        ${{ github.action_path }}/scripts/suggest-review-mode.sh "${{ inputs.pr-number }}" "${{ inputs.feedback }}" "${{ inputs.file-path }}" "${{ inputs.line-number }}" "${{ inputs.comment-id }}" "${{ inputs.anthropic-api-key }}" "${{ inputs.github-token }}" "${{ inputs.strict-mode }}"

package.json

@@ -1,6 +1,6 @@
 {
   "name": "claude-code-github-action",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "description": "GitHub action for Claude Code Integration in PR comments, reviews and inline code suggestions",
   "main": "index.js",
   "scripts": {

scripts/suggest-review-mode.sh

@@ -9,13 +9,20 @@ LINE_NUMBER=$4
 COMMENT_ID=$5
 ANTHROPIC_API_KEY=$6
 GITHUB_TOKEN=$7
+STRICT_MODE=$8
+
+# Default to strict mode if not provided
+if [ -z "$STRICT_MODE" ]; then
+    STRICT_MODE="true"
+fi
 
 # Set up authentication
 echo "$GITHUB_TOKEN" | gh auth login --with-token
 export ANTHROPIC_API_KEY="$ANTHROPIC_API_KEY"
 
-# Create a temp file for Claude's response
+# Create a temp files for Claude's responses
 RESPONSE_FILE=$(mktemp)
+ADDITIONAL_SUGGESTIONS_FILE=$(mktemp)
 
 # Get PR details using GitHub CLI
 echo "Fetching PR details for PR #$PR_NUMBER"
@@ -50,6 +57,21 @@ REPO_DESC=$(echo "$REPO_INFO" | jq -r '.description')
 REPO_DEFAULT_BRANCH=$(echo "$REPO_INFO" | jq -r '.defaultBranchRef.name')
 REPO_LANGUAGES=$(echo "$REPO_INFO" | jq -r '.languages[].name' | tr '\n' ', ' | sed 's/,$//')
 
+# Add strict mode instructions if enabled
+STRICT_INSTRUCTIONS=""
+if [ "$STRICT_MODE" = "true" ]; then
+    STRICT_INSTRUCTIONS=$(cat <<EOF
+
+IMPORTANT - STRICT MODE IS ENABLED:
+1. You MUST address ONLY what the user explicitly asked for in their query
+2. Do NOT make any unrelated improvements to the code, even if they would be beneficial
+3. If the user asks to "add X", focus exclusively on adding X, not refactoring existing code
+4. If you identify other issues in the code, DO NOT address them in your suggestion
+5. Stay hyper-focused on the specific request, even if other improvements seem obvious
+EOF
+)
+fi
+
 # Build the prompt for Claude to generate a suggested change
 PROMPT=$(cat <<EOF
 This is a GitHub code review. Create a suggestion for the specific code at line $LINE_NUMBER in file '$FILE_PATH'.
@@ -60,6 +82,7 @@ You MUST format your response as a SINGLE suggestion using the exact GitHub sugg
 \`\`\`suggestion
 [Your improved code here]
 \`\`\`
+$STRICT_INSTRUCTIONS
 
 Guidelines:
 1. Focus ONLY on line $LINE_NUMBER and immediately surrounding lines
@@ -100,7 +123,49 @@ echo "Posting Claude's suggested change as a reply to the comment"
 gh api --method POST "/repos/:owner/:repo/pulls/$PR_NUMBER/comments/$COMMENT_ID/replies" \
    -F body="@$RESPONSE_FILE"
 
+# If in non-strict mode, check for additional improvement suggestions
+if [ "$STRICT_MODE" = "false" ]; then
+    ADDITIONAL_PROMPT=$(cat <<EOF
+You've already provided a suggestion that directly addresses the user's query for line $LINE_NUMBER in file '$FILE_PATH'.
+
+Now, identify any additional code improvements that would be beneficial beyond what was specifically requested. 
+These should be improvements that weren't part of the original request but would enhance code quality, 
+readability, performance, or maintainability.
+
+Format your response as:
+1. A brief explanation of why these additional improvements would be valuable
+2. Clearly labeled additional suggestions (not using the GitHub suggestion format)
+
+File being reviewed: $FILE_PATH
+Line number context: $LINE_NUMBER
+
+Context (code around line $LINE_NUMBER):
+\`\`\`
+$CONTEXT_CONTENT
+\`\`\`
+
+Complete file content:
+\`\`\`
+$FILE_CONTENT
+\`\`\`
+
+If you don't have any additional suggestions beyond what was directly requested, respond with "No additional improvements suggested."
+EOF
+)
+
+    echo "Checking for additional improvement suggestions..."
+    echo "$ADDITIONAL_PROMPT" | claude -p - > "$ADDITIONAL_SUGGESTIONS_FILE"
+    
+    # Only post additional suggestions if they exist
+    if ! grep -q "No additional improvements suggested" "$ADDITIONAL_SUGGESTIONS_FILE"; then
+        echo "Posting additional improvement suggestions..."
+        ADDITIONAL_CONTENT="## Additional Suggestions\n\nWhile addressing your specific request, I noticed some other potential improvements:\n\n$(cat "$ADDITIONAL_SUGGESTIONS_FILE")\n\n*These are optional suggestions beyond what you specifically requested.*"
+        gh api --method POST "/repos/:owner/:repo/pulls/$PR_NUMBER/comments/$COMMENT_ID/replies" \
+           -F body="$ADDITIONAL_CONTENT"
+    fi
+fi
+
 # Clean up
-rm -f "$RESPONSE_FILE"
+rm -f "$RESPONSE_FILE" "$ADDITIONAL_SUGGESTIONS_FILE"
 
-echo "Claude's in-line suggestion posted successfully!"
+echo "Claude's suggestions posted successfully!"