Vertex AI

The libaray supports the Gecko Embedding model on VertexAI with 768 dimensions.

To use this model, you need to authenticate to Vertex AI on GCP. Refer here on how to do this. Once done, follow these steps -

Install VertexAi addon

npm install @llm-tools/embedjs-vertexai

Usage

import { RAGApplicationBuilder } from '@llm-tools/embedjs';
import { GeckoEmbeddings } from '@llm-tools/embedjs-vertexai';
import { HNSWDb } from '@llm-tools/embedjs-hnswlib';

const app = await new RAGApplicationBuilder()
.setEmbeddingModel(new GeckoEmbeddings())