aboutsummaryrefslogtreecommitdiff
path: root/cmd/_old/geminitest
diff options
context:
space:
mode:
authorXe Iaso <me@xeiaso.net>2024-10-06 08:53:06 -0400
committerXe Iaso <me@xeiaso.net>2024-10-06 08:56:03 -0400
commit57c4d9757ae09700e6bd65de6e55f5cd17e6e6e2 (patch)
tree0b4b8365328c7e66ab1fb6d4b4d4bb94ac796f78 /cmd/_old/geminitest
parentfda09b55e316b7f162371cf52bc401fa742913e8 (diff)
downloadx-57c4d9757ae09700e6bd65de6e55f5cd17e6e6e2.tar.xz
x-57c4d9757ae09700e6bd65de6e55f5cd17e6e6e2.zip
move some old code to the _old folder to save CI time
Signed-off-by: Xe Iaso <me@xeiaso.net>
Diffstat (limited to 'cmd/_old/geminitest')
-rw-r--r--cmd/_old/geminitest/main.go40
1 files changed, 40 insertions, 0 deletions
diff --git a/cmd/_old/geminitest/main.go b/cmd/_old/geminitest/main.go
new file mode 100644
index 0000000..e06c872
--- /dev/null
+++ b/cmd/_old/geminitest/main.go
@@ -0,0 +1,40 @@
+package main
+
+import (
+ "context"
+ "flag"
+ "fmt"
+ "log"
+
+ "github.com/google/generative-ai-go/genai"
+ "google.golang.org/api/iterator"
+ "google.golang.org/api/option"
+ "within.website/x/internal"
+)
+
+var (
+ geminiApiKey = flag.String("gemini-api-key", "", "The Gemini API key")
+ geminiModel = flag.String("gemini-model", "gemini-1.5-flash", "The model to use for generating text")
+)
+
+func main() {
+ internal.HandleStartup()
+ ctx := context.Background()
+ client, err := genai.NewClient(ctx, option.WithAPIKey(*geminiApiKey))
+ if err != nil {
+ log.Fatal(err)
+ }
+ defer client.Close()
+
+ iter := client.ListModels(ctx)
+ for {
+ m, err := iter.Next()
+ if err == iterator.Done {
+ break
+ }
+ if err != nil {
+ panic(err)
+ }
+ fmt.Println(m.Name, m.Description)
+ }
+}