From 8bf426920831f36c565ca695ec7a07bf75a607c9 Mon Sep 17 00:00:00 2001
From: Hristo <53634432+izo0x90@users.noreply.github.com>
Date: Tue, 21 May 2024 16:23:34 -0400
Subject: [PATCH] Add vertexai text embeddings capability

---
 src/lib/providers.ts | 11 +++++++++++
 1 file changed, 11 insertions(+)

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({