Adding a new parameter for model, and updating the default to 2.5 Flash. (#18)

1 files changed, 9 insertions, 0 deletions

diff --git a/packages/cli/src/config/args.ts b/packages/cli/src/config/args.ts
index 7c4ebbc0..f36e7e58 100644
--- a/packages/cli/src/config/args.ts
+++ b/packages/cli/src/config/args.ts
@@ -1,8 +1,11 @@
 import yargs from 'yargs/yargs';
 import { hideBin } from 'yargs/helpers';
 
+const DEFAULT_GEMINI_MODEL = 'gemini-2.5-flash-preview-04-17';
+
 export interface CliArgs {
   target_dir: string | undefined;
+  model: string | undefined;
   _: (string | number)[]; // Captures positional arguments
   // Add other expected args here if needed
   // e.g., verbose?: boolean;
@@ -16,6 +19,12 @@ export async function parseArguments(): Promise<CliArgs> {
       description:
         'The target directory for Gemini operations. Defaults to the current working directory.',
     })
+    .option('model', {
+      alias: 'm',
+      type: 'string',
+      description: `The Gemini model to use. Defaults to ${DEFAULT_GEMINI_MODEL}.`,
+      default: DEFAULT_GEMINI_MODEL,
+    })
     .help()
     .alias('h', 'help')
     .strict() // Keep strict mode to error on unknown options