Support model

Author: nathro

README.md

@@ -79,6 +79,7 @@ Each example includes a complete workflow file that you can copy to your `.githu
 | `pull_number`          | PR number for template context extraction             | No       | `${{ github.event.pull_request.number }}` |
 | `repo_name`            | Repository name for template context                  | No       | `${{ github.repository }}`                |
 | `custom_context`       | Additional JSON context for templates                 | No       | `'{"priority": "high"}'`                  |
+| `model`                | Model to use; passed through to auggie as --model     | No       | `gpt-4o-mini`                              |
 
 \*Either `instruction`, `instruction_file`, or `template_directory` must be provided.
 

action.yml

@@ -37,6 +37,9 @@ inputs:
   custom_context:
     description: "Additional context data as JSON string to pass to templates."
     required: false
+  model:
+    description: "Model to use; forwarded to auggie as --model"
+    required: false
 
 runs:
   using: "composite"
@@ -67,3 +70,4 @@ runs:
         INPUT_PULL_NUMBER: ${{ inputs.pull_number }}
         INPUT_REPO_NAME: ${{ inputs.repo_name }}
         INPUT_CUSTOM_CONTEXT: ${{ inputs.custom_context }}
+        INPUT_MODEL: ${{ inputs.model }}

src/config/constants.ts

@@ -24,6 +24,7 @@ export const INPUT_FIELD_MAP: Record<string, InputField> = {
   repoName: { envVar: 'INPUT_REPO_NAME', required: false },
   templateDirectory: { envVar: 'INPUT_TEMPLATE_DIRECTORY', required: false },
   templateName: { envVar: 'INPUT_TEMPLATE_NAME', required: false },
+  model: { envVar: 'INPUT_MODEL', required: false },
 };
 
 export const TEMPLATE_CONFIG = {

src/index.ts

@@ -113,17 +113,19 @@ async function runAugmentScript(inputs: ActionInputs): Promise<void> {
       instructionFile: instruction_value,
     });
   }
+  const args = ["--print"];
+  if (inputs.model && inputs.model.trim().length > 0) {
+    args.push('--model', inputs.model.trim());
+  }
   if (is_file) {
     logger.info(`📄 Using instruction file: ${instruction_value}`);
-    await execCommand('auggie', [
-      '--instruction-file',
-      instruction_value,
-      '--print',
-    ]);
+    args.push('--instruction-file');
   } else {
     logger.info('📝 Using direct instruction');
-    await execCommand('auggie', ['--print', instruction_value]);
+    args.push('--instruction');
   }
+  args.push(instruction_value);
+  await execCommand('auggie', args);
   logger.info('✅ Augment Agent completed successfully');
 }
 

src/types/inputs.ts

@@ -16,6 +16,7 @@ export interface ActionInputs {
   githubToken?: string | undefined;
   instruction?: string | undefined;
   instructionFile?: string | undefined;
+  model?: string | undefined;
 
   // Template inputs
   templateDirectory?: string | undefined;

src/utils/validation.ts

@@ -18,6 +18,7 @@ const ActionInputsSchema = z
     githubToken: z.string().optional(),
     instruction: z.string().optional(),
     instructionFile: z.string().optional(),
+    model: z.string().optional(),
     templateDirectory: z.string().optional(),
     templateName: z.string().default('prompt.njk'),
     customContext: z.string().optional(),