diff --git a/src/lib/providers.ts b/src/lib/providers.ts index 74aa198..36c733c 100644 --- a/src/lib/providers.ts +++ b/src/lib/providers.ts @@ -1,6 +1,7 @@ import { ChatOpenAI, OpenAIEmbeddings } from '@langchain/openai'; import { ChatOllama } from '@langchain/community/chat_models/ollama'; import { VertexAI } from "@langchain/google-vertexai"; +import { GoogleVertexAIEmbeddings } from "@langchain/community/embeddings/googlevertexai"; import { OllamaEmbeddings } from '@langchain/community/embeddings/ollama'; import { HuggingFaceTransformersEmbeddings } from './huggingfaceTransformer'; import { hasGCPCredentials } from '../auth'; @@ -186,6 +187,16 @@ export const getAvailableEmbeddingModelProviders = async () => { } } + if (await hasGCPCredentials()) { + try { + models['vertexai'] = { + 'Text Gecko default': new GoogleVertexAIEmbeddings(), + } + } catch (err) { + logger.error(`Error loading VertexAI embeddings: ${err}`); + } + } + try { models['local'] = { 'BGE Small': new HuggingFaceTransformersEmbeddings({