Support Model (#2)

* Support Model

* Update model example

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 described        | Yes      | `${{ github.event.pull_request.number }}`           |
 | `repo_name`            | The full name (owner/repo) of the repository          | Yes      | `${{ github.repository }}`                          |
 | `custom_guidelines`    | Custom guidelines for PR descriptions (optional)      | No       | See [Custom Guidelines](#custom-guidelines) section |
+| `model`                | Optional model to use for generation                  | No       | e.g. `sonnet4`, from `auggie --list-models`         |
 
 ## How It Works
 

action.yml

@@ -18,6 +18,9 @@ inputs:
   custom_guidelines:
     description: "Optional custom guidelines to include in PR description generation. Provide additional instructions or requirements specific to your project."
     required: false
+  model:
+    description: "Optional model to use for generation. Passed directly to 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 }}