diff options
| -rw-r--r-- | genai/Makefile | 3 | ||||
| -rw-r--r-- | genai/aistudio.go | 12 |
2 files changed, 6 insertions, 9 deletions
diff --git a/genai/Makefile b/genai/Makefile index 4ec151e..1b7e731 100644 --- a/genai/Makefile +++ b/genai/Makefile @@ -2,6 +2,9 @@ testgo: goimports # GO111MODULE=off go run aistudio.go go run -v aistudio.go +real: + GO111MODULE=off go run aistudio.go + vet: GO111MODULE=off go vet diff --git a/genai/aistudio.go b/genai/aistudio.go index ab6ccf2..3fda5ec 100644 --- a/genai/aistudio.go +++ b/genai/aistudio.go @@ -6,7 +6,8 @@ import ( "log" "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" ) @@ -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 - iter := model.GenerateContentStream(ctx, []*genai.Content{userContent}...) + iter := model.GenerateContentStream(ctx, genai.Text("what is the current version of forge on the webpage?")) for { resp, err := iter.Next() if err != nil { |