Skip to content

Commit dd5ff9c

Browse files
bharadwaj-adityaAditya Bharadwaj
andauthored
updated text model to gemini-embedding-001 (#1247)
Co-authored-by: Aditya Bharadwaj <adibh@google.com>
1 parent f4f7ab9 commit dd5ff9c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

assessment/collectors/embeddings/generate_embedding.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ func attachEmbeddings(ctx context.Context, client PredictionClientInterface, pro
133133
func newAIPredictionClient(location string) (context.Context, PredictionClientInterface, string, error) {
134134
ctx := context.Background()
135135
apiEndpoint := fmt.Sprintf("%s-aiplatform.googleapis.com:443", location)
136-
model := "text-embedding-preview-0815"
136+
model := "gemini-embedding-001"
137137

138138
client, err := aiplatform.NewPredictionClient(ctx, option.WithEndpoint(apiEndpoint))
139139
if err != nil {

0 commit comments

Comments
 (0)