Vector index test

Author: apetenchea

src/test/11-managing-indexes.ts

@@ -4,6 +4,7 @@ import { Database } from "../databases.js";
 import { config } from "./_config.js";
 
 const it312 = config.arangoVersion! >= 31200 ? it : it.skip;
+const it31205 = config.arangoVersion! >= 31205 ? it : it.skip;
 
 describe("Managing indexes", function () {
   let system: Database, db: Database;
@@ -80,6 +81,35 @@ describe("Managing indexes", function () {
       expect(info).to.have.property("isNewlyCreated", true);
     });
   });
+  describe("collection.ensureIndex#vector", () => {
+    it31205("should create a vector index", async () => {
+      await collection.save({
+        _key: "sample1",
+        embedding: Array(128).fill(0.1),
+      });
+      await collection.save({
+        _key: "sample2",
+        embedding: Array(128).fill(0.1),
+      });
+      const info = await collection.ensureIndex({
+        type: "vector",
+        fields: ["embedding"],
+        params: {
+          metric: "cosine",
+          dimension: 128,
+          nLists: 2,
+        },
+      });
+      expect(info).to.have.property("id");
+      expect(info).to.have.property("type", "vector");
+      expect(info).to.have.property("fields");
+      expect(info.fields).to.eql(["embedding"]);
+      expect(info).to.have.property("isNewlyCreated", true);
+      expect(info).to.have.nested.property("params.metric", "cosine");
+      expect(info).to.have.nested.property("params.dimension", 128);
+      expect(info).to.have.nested.property("params.nLists", 2);
+    });
+  });
   describe("collection.index", () => {
     it("should return information about a index", async () => {
       const info = await collection.ensureIndex({