ai code builds kinda

This commit is contained in:
Jeff Carr 2025-04-12 11:27:04 -05:00
parent 1b0344899f
commit 605ebd6c8b
2 changed files with 6 additions and 9 deletions

View File

@ -2,6 +2,9 @@ testgo: goimports
# GO111MODULE=off go run aistudio.go # GO111MODULE=off go run aistudio.go
go run -v aistudio.go go run -v aistudio.go
real:
GO111MODULE=off go run aistudio.go
vet: vet:
GO111MODULE=off go vet GO111MODULE=off go vet

View File

@ -6,7 +6,8 @@ import (
"log" "log"
"os" "os"
"github.com/google/generative-ai-go/genai" // "github.com/google/generative-ai-go/genai"
"cloud.google.com/go/ai/vertexai/genai"
"google.golang.org/api/option" "google.golang.org/api/option"
) )
@ -33,15 +34,8 @@ func main() {
}, },
} }
// Create user content
userContent := &genai.Content{
Parts: []genai.Part{
genai.Text("what is the current version of forge on the webpage?"),
},
}
// Stream content generation // Stream content generation
iter := model.GenerateContentStream(ctx, []*genai.Content{userContent}...) iter := model.GenerateContentStream(ctx, genai.Text("what is the current version of forge on the webpage?"))
for { for {
resp, err := iter.Next() resp, err := iter.Next()
if err != nil { if err != nil {