Support Model

Author: nathro

README.md

@@ -80,6 +80,7 @@ Each example includes a complete workflow file that you can copy to your `.githu
 | `pull_number`          | The number of the pull request being reviewed                                              | Yes      | `${{ github.event.pull_request.number }}`           |
 | `repo_name`            | The full name (owner/repo) of the repository                                               | Yes      | `${{ github.repository }}`                          |
 | `custom_guidelines`    | Custom guidelines to include in PR reviews                                                 | No       | See [Custom Guidelines](#custom-guidelines) section |
+| `model`                | Optional model name to use; passed directly to augment agent                               | No       | e.g., openai/gpt-4o-mini                            |
 
 ## How It Works
 

action.yml

@@ -18,6 +18,9 @@ inputs:
   custom_guidelines:
     description: "Optional custom guidelines to include in the PR review process. These will be added to the default guidelines."
     required: false
+  model:
+    description: "Optional model name to use; passed directly to augmentcode/augment-agent."
+    required: false
 
 runs:
   using: "composite"
@@ -44,11 +47,12 @@ runs:
         CUSTOM_GUIDELINES: ${{ inputs.custom_guidelines }}
 
     - name: Run Augment Agent
-      uses: augmentcode/augment-agent@6f08f56902d6728a0ffbbb2004c136c06fec8104 # v0.1.0
+      uses: augmentcode/augment-agent@v0
       with:
         augment_session_auth: ${{ inputs.augment_session_auth }}
         github_token: ${{ inputs.github_token }}
         template_directory: "${{ github.action_path }}/templates"
         pull_number: ${{ inputs.pull_number }}
         repo_name: ${{ inputs.repo_name }}
         custom_context: ${{ steps.custom_context.outputs.context }}
+        model: ${{ inputs.model }}