diff --git a/src/lib/models/providers/gemini.ts b/src/lib/models/providers/gemini.ts index 6cf3584..976832b 100644 --- a/src/lib/models/providers/gemini.ts +++ b/src/lib/models/providers/gemini.ts @@ -48,7 +48,7 @@ class GeminiProvider extends BaseModelProvider { let defaultChatModels: Model[] = []; data.models.forEach((m: any) => { - if (m.supportedGenerationMethods.includes('embedText')) { + if (m.supportedGenerationMethods.some((genMethod: string) => genMethod === 'embedText' || genMethod === 'embedContent')) { defaultEmbeddingModels.push({ key: m.name, name: m.displayName,