Skip to content

Commit 3ddcced

Browse files
committed
feat(gemini-provider): update embedding models
1 parent e226645 commit 3ddcced

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/lib/providers/gemini.ts

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,12 @@ const geminiChatModels: Record<string, string>[] = [
4040

4141
const geminiEmbeddingModels: Record<string, string>[] = [
4242
{
43-
displayName: 'Gemini Embedding',
44-
key: 'gemini-embedding-exp',
43+
displayName: 'Text Embedding 004',
44+
key: 'models/text-embedding-004',
45+
},
46+
{
47+
displayName: 'Embedding 001',
48+
key: 'models/embedding-001',
4549
},
4650
];
4751

0 commit comments

Comments
 (0)