From ee273ff21562dc70225fff08b2f89493d9b14ded Mon Sep 17 00:00:00 2001
From: bel <bel>
Date: Sun, 22 Mar 2026 20:34:23 -0600
Subject: [PATCH] accept model

---
 main.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index e6e7899..5f30517 100644
--- a/main.go
+++ b/main.go
@@ -27,6 +27,8 @@ func main() {
 func run(ctx context.Context) error {
 	fs := flag.NewFlagSet(os.Args[0], flag.ContinueOnError)
 	n := fs.Int("n", 1, "loops")
+	model := fs.String("m", "gemini-2.5-flash", "model to use")
+
 	if err := fs.Parse(os.Args[1:]); err != nil {
 		return err
 	}
@@ -105,7 +107,7 @@ For the city {{.City}}, answer the following descriptors '1' for true or '0' for
 	for i := 0; i < *n; i++ {
 		result, err := client.Models.GenerateContent(
 			ctx,
-			"gemini-2.5-flash",
+			*model,
 			genai.Text(w.String()),
 			nil,
 		)