diff options
| author | Jeff Carr <[email protected]> | 2025-08-30 15:15:43 -0500 |
|---|---|---|
| committer | Jeff Carr <[email protected]> | 2025-08-30 15:15:43 -0500 |
| commit | c7896e47f9c91e330b023bdb34a7540f193a5422 (patch) | |
| tree | 76a0801d34c7b53442e2e5d2ecaeba5ef5e961c0 /doConnect.go | |
| parent | 7f8f5e3b9b37fccbb357b70e6490823a8b22ae7e (diff) | |
code cleanup of early drafts of code from Gemini AI
Diffstat (limited to 'doConnect.go')
| -rw-r--r-- | doConnect.go | 31 |
1 files changed, 6 insertions, 25 deletions
diff --git a/doConnect.go b/doConnect.go index 2285f0b..b7abd40 100644 --- a/doConnect.go +++ b/doConnect.go @@ -10,42 +10,23 @@ import ( ) // doConnect initializes the Gemini client and handles the request flow. -func doConnect() error { +func doConnect() (*genai.Client, error) { apiKey := os.Getenv("GEMINI_API_KEY") if apiKey == "" { - return log.Errorf("GEMINI_API_KEY environment variable not set") + return nil, log.Errorf("GEMINI_API_KEY environment variable not set") } ctx := context.Background() client, err := genai.NewClient(ctx, &genai.ClientConfig{APIKey: apiKey}) if err != nil { - return log.Errorf("failed to create new genai client: %w", err) + return nil, log.Errorf("failed to create new genai client: %w", err) } - if argv.JsonFile != "" { - req, err := parseJSON(argv.JsonFile) - if err != nil { - return err - } - log.Info("parseJSON() ok. model =", req.Model) - - genaiContent, err := convertToGenai(req) - if err != nil { - return log.Errorf("failed to convert to genai.Content: %w", err) - } - log.Info("Successfully converted JSON to genai.Content") - // Here you would now use the 'genaiContent' to send to the API - _ = genaiContent // Prevent unused variable error for now - - return nil - } - - log.Info("doing sampleHello()") - return sampleHello(client) + return client, err } // sampleHello sends a hardcoded prompt to the model and prints the response. -func sampleHello(client *genai.Client) error { +func simpleHello(client *genai.Client) error { log.Info("Sending 'hello, how are you' to the Gemini API...") ctx := context.Background() @@ -56,7 +37,7 @@ func sampleHello(client *genai.Client) error { content := []*genai.Content{{Parts: parts}} - resp, err := client.Models.GenerateContent(ctx, "gemini-1.5-flash-latest", content, nil) + resp, err := client.Models.GenerateContent(ctx, "gemini-2.5-flash", content, nil) if err != nil { return log.Errorf("error sending message: %v", err) } |
