ai code builds kindav0.0.53

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 {