diff options
| author | Xe Iaso <me@xeiaso.net> | 2024-10-06 08:53:06 -0400 |
|---|---|---|
| committer | Xe Iaso <me@xeiaso.net> | 2024-10-06 08:56:03 -0400 |
| commit | 57c4d9757ae09700e6bd65de6e55f5cd17e6e6e2 (patch) | |
| tree | 0b4b8365328c7e66ab1fb6d4b4d4bb94ac796f78 /cmd/_old/geminitest | |
| parent | fda09b55e316b7f162371cf52bc401fa742913e8 (diff) | |
| download | x-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.go | 40 |
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) + } +} |
