Skip to content

Commit 22695f4

Browse files
authored
Merge pull request #916 from skoved/gemini-embedding-fix
fix: list all available gemini embedding models
2 parents b0d97c4 + 75ef2e0 commit 22695f4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lib/models/providers/gemini.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ class GeminiProvider extends BaseModelProvider<GeminiConfig> {
4848
let defaultChatModels: Model[] = [];
4949

5050
data.models.forEach((m: any) => {
51-
if (m.supportedGenerationMethods.includes('embedText')) {
51+
if (m.supportedGenerationMethods.some((genMethod: string) => genMethod === 'embedText' || genMethod === 'embedContent')) {
5252
defaultEmbeddingModels.push({
5353
key: m.name,
5454
name: m.displayName,

0 commit comments

Comments
 (0)