Test different embedding models over different regions.

langchain-ai · Dec 19, 2024 · 7526de8 · 7526de8
1 parent 886cb68
commit 7526de8
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/libs/langchain-google-vertexai/src/tests/embeddings.int.test.ts b/libs/langchain-google-vertexai/src/tests/embeddings.int.test.ts
@@ -9,9 +9,17 @@ test("Test VertexAIEmbeddings.embedQuery", async () => {
   expect(typeof res[0]).toBe("number");
 });
 
-test("Test VertexAIEmbeddings.embedDocuments", async () => {
+const testModelsLocations = [
+  ["text-embedding-005", "us-central1"],
+  ["text-multilingual-embedding-002", "us-central1"],
+  ["text-embedding-005", "europe-west9"],
+  ["text-multilingual-embedding-002", "europe-west9"],
+]
+
+test.each(testModelsLocations)("VertexAIEmbeddings.embedDocuments %s %s", async (model, location) => {
   const embeddings = new VertexAIEmbeddings({
-    model: "text-embedding-004",
+    model,
+    location,
   });
   const res = await embeddings.embedDocuments([
     "Hello world",