From b6bac09ca77a5920bd1e77dec4c0b9367d32c0ba Mon Sep 17 00:00:00 2001 From: Nathan Rockenbach Date: Mon, 11 Aug 2025 04:52:17 +0000 Subject: [PATCH 1/2] Support Model --- README.md | 1 + action.yml | 6 +++++- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 084d3b8..6883f34 100644 --- a/README.md +++ b/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. `gpt-4o`, `claude-3-5-sonnet`, etc. | ## How It Works diff --git a/action.yml b/action.yml index 93cabcf..d5ad8ad 100644 --- a/action.yml +++ b/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,7 +47,7 @@ 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 }} @@ -52,3 +55,4 @@ runs: pull_number: ${{ inputs.pull_number }} repo_name: ${{ inputs.repo_name }} custom_context: ${{ steps.custom_context.outputs.context }} + model: ${{ inputs.model }} From 87b2c5d932805e1a19e4014070e61a4fced588e6 Mon Sep 17 00:00:00 2001 From: Nathan Rockenbach Date: Mon, 11 Aug 2025 18:09:45 +0000 Subject: [PATCH 2/2] Update model example --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 6883f34..05a202f 100644 --- a/README.md +++ b/README.md @@ -80,7 +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. `gpt-4o`, `claude-3-5-sonnet`, etc. | +| `model` | Optional model to use for generation | No | e.g. `sonnet4`, from `auggie --list-models` | ## How It Works