From cb303514035440b1631964cad5093a4c80cd3e43 Mon Sep 17 00:00:00 2001
From: Evan Senter <evansenter@gmail.com>
Date: Fri, 18 Apr 2025 17:06:16 +0100
Subject: Adding a new parameter for model, and updating the default to 2.5
 Flash. (#18)

---
 packages/cli/src/config/args.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'packages/cli/src/config/args.ts')

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
-- 
cgit v1.2.3