Skip to content

Commit 66e2b9e

Browse files
authored
fix(oss): text-embedding-004 -> gemini-embedding-001 (#489)
text-embedding-004 has been discontinued
1 parent 8463342 commit 66e2b9e

File tree

6 files changed

+6
-6
lines changed

6 files changed

+6
-6
lines changed

src/oss/javascript/integrations/text_embedding/google_generativeai.mdx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ import { GoogleGenerativeAIEmbeddings } from "@langchain/google-genai";
5555
import { TaskType } from "@google/generative-ai";
5656

5757
const embeddings = new GoogleGenerativeAIEmbeddings({
58-
model: "text-embedding-004", // 768 dimensions
58+
model: "gemini-embedding-001", // 768 dimensions
5959
taskType: TaskType.RETRIEVAL_DOCUMENT,
6060
title: "Document title",
6161
});

src/oss/javascript/integrations/text_embedding/google_vertex_ai.mdx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ import { VertexAIEmbeddings } from "@langchain/google-vertexai";
6565
// import { VertexAIEmbeddings } from "@langchain/google-vertexai-web"
6666

6767
const embeddings = new VertexAIEmbeddings({
68-
model: "text-embedding-004",
68+
model: "gemini-embedding-001",
6969
// ...
7070
});
7171
```

src/oss/javascript/integrations/text_embedding/index.mdx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ GOOGLE_APPLICATION_CREDENTIALS=credentials.json
184184
import { VertexAIEmbeddings } from "@langchain/google-vertexai";
185185

186186
const embeddings = new VertexAIEmbeddings({
187-
model: "text-embedding-004"
187+
model: "gemini-embedding-001"
188188
});
189189
```
190190

src/oss/javascript/integrations/vectorstores/index.mdx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ GOOGLE_APPLICATION_CREDENTIALS=credentials.json
182182
import { VertexAIEmbeddings } from "@langchain/google-vertexai";
183183

184184
const embeddings = new VertexAIEmbeddings({
185-
model: "text-embedding-004"
185+
model: "gemini-embedding-001"
186186
});
187187
```
188188

src/oss/python/integrations/text_embedding/google_vertex_ai_palm.mdx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ Now we can instantiate our model object and generate embeddings:
8181
from langchain_google_vertexai import VertexAIEmbeddings
8282

8383
# Initialize the a specific Embeddings Model version
84-
embeddings = VertexAIEmbeddings(model_name="text-embedding-004")
84+
embeddings = VertexAIEmbeddings(model_name="gemini-embedding-001")
8585
```
8686

8787
## Indexing and Retrieval

src/snippets/embeddings-tabs-js.mdx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@
8989
import { VertexAIEmbeddings } from "@langchain/google-vertexai";
9090

9191
const embeddings = new VertexAIEmbeddings({
92-
model: "text-embedding-004"
92+
model: "gemini-embedding-001"
9393
});
9494
```
9595
</Tab>

0 commit comments

Comments
 (0)