Format

langchain-ai · Dec 19, 2024 · 604219b · 604219b
1 parent 7526de8
commit 604219b
Showing 1 changed file with 23 additions and 20 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
@@ -14,24 +14,27 @@ const testModelsLocations = [
   ["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,
-    location,
-  });
-  const res = await embeddings.embedDocuments([
-    "Hello world",
-    "Bye bye",
-    "we need",
-    "at least",
-    "six documents",
-    "to test pagination",
-  ]);
-  // console.log(res);
-  expect(res).toHaveLength(6);
-  res.forEach((r) => {
-    expect(typeof r[0]).toBe("number");
-  });
-});
+test.each(testModelsLocations)(
+  "VertexAIEmbeddings.embedDocuments %s %s",
+  async (model, location) => {
+    const embeddings = new VertexAIEmbeddings({
+      model,
+      location,
+    });
+    const res = await embeddings.embedDocuments([
+      "Hello world",
+      "Bye bye",
+      "we need",
+      "at least",
+      "six documents",
+      "to test pagination",
+    ]);
+    // console.log(res);
+    expect(res).toHaveLength(6);
+    res.forEach((r) => {
+      expect(typeof r[0]).toBe("number");
+    });
+  }
+);