diff options
Diffstat (limited to 'genai/aistudio.go')
| -rw-r--r-- | genai/aistudio.go | 12 |
1 files changed, 3 insertions, 9 deletions
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 { |