29 lines
689 B
Plaintext
29 lines
689 B
Plaintext
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/google/generative-ai-go/genai"
|
|
"google.golang.org/api/option"
|
|
)
|
|
|
|
// doConnect initializes and returns a Gemini client.
|
|
// It reads the API key from the GEMINI_API_KEY environment variable.
|
|
func doConnect() (*genai.GenerativeModel, error) {
|
|
apiKey := os.Getenv("GEMINI_API_KEY")
|
|
if apiKey == "" {
|
|
return nil, fmt.Errorf("GEMINI_API_KEY environment variable not set")
|
|
}
|
|
|
|
ctx := context.Background()
|
|
client, err := genai.NewClient(ctx, option.WithAPIKey(apiKey))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create new genai client: %w", err)
|
|
}
|
|
|
|
model := client.GenerativeModel("gemini-pro")
|
|
return model, nil
|
|
}
|