mirror of https://github.com/velour/catbase.git
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/google/generative-ai-go/genai"
|
||
|
|
"google.golang.org/api/option"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
ctx := context.Background()
|
||
|
|
client, err := genai.NewClient(ctx, option.WithAPIKey(os.Getenv("GEMINI_API_KEY")))
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
defer client.Close()
|
||
|
|
|
||
|
|
model := client.GenerativeModel("gemini-1.5-flash")
|
||
|
|
model.SafetySettings = []*genai.SafetySetting{
|
||
|
|
{
|
||
|
|
Category: genai.HarmCategoryHarassment,
|
||
|
|
Threshold: genai.HarmBlockOnlyHigh,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
resp, err := model.GenerateContent(ctx, genai.Text("I support Martians Soccer Club and I think Jupiterians Football Club sucks! Write a ironic phrase about them."))
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
printResponse(resp)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func printResponse(resp *genai.GenerateContentResponse) {
|
||
|
|
for _, cand := range resp.Candidates {
|
||
|
|
if cand.Content != nil {
|
||
|
|
for _, part := range cand.Content.Parts {
|
||
|
|
fmt.Println(part)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
fmt.Println("---")
|
||
|
|
}
|
||
|
|
|
||
|
|
// curl \
|
||
|
|
// -H 'Content-Type: application/json' \
|
||
|
|
// -d '{"contents":[{"parts":[{"text":"Explain how AI works"}]}]}' \
|
||
|
|
// -X POST 'https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent?key=YOUR_API_KEY'
|