Fixes for a couple of issues discovered with vector support (#2433)

* Fixed a couple of issues with creating a vector index of embeddings and querying the index.

* Cleaned-up the test case by using the TestHelper

(cherry picked from commit 531fcf6)

Author: odysseaspenta

engine/src/main/java/com/arcadedb/query/sql/function/vector/SQLFunctionVectorNeighbors.java

@@ -64,7 +64,7 @@ public Object execute(final Object self, final Identifiable currentRecord, final
 
     final int limit = params[2] instanceof Number n ? n.intValue() : Integer.parseInt(params[2].toString());
 
-    final List<Pair<Vertex, ? extends Number>> neighbors = vIndex.findNeighborsFromVector(key, limit, null);
+    final List<Pair<Vertex, ? extends Number>> neighbors = vIndex.findNeighborsFromId(key, limit, null);
 
     final ArrayList<Object> result = new ArrayList<>(neighbors.size());
     for (Pair<Vertex, ? extends Number> n : neighbors)

integration/src/main/java/com/arcadedb/integration/importer/vector/TextEmbeddingsImporter.java

@@ -227,10 +227,10 @@ public TextEmbeddingsImporter setContext(final ImporterContext context) {
 
   private List<TextFloatsEmbedding> loadFromFile() throws IOException {
     try (BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream))) {
-      final Stream<String> parser = reader.lines();
+      Stream<String> parser = reader.lines();
 
       if (settings.parsingLimitEntries > 0)
-        parser.limit(settings.parsingLimitEntries);
+        parser = parser.limit(settings.parsingLimitEntries);
 
       final AtomicInteger vectorSize = new AtomicInteger(301);
 

integration/src/test/java/com/arcadedb/integration/importer/FastTextVectorImportTest.java

@@ -0,0 +1,59 @@
+package com.arcadedb.integration.importer;
+
+import com.arcadedb.database.Database;
+import com.arcadedb.database.DatabaseFactory;
+import com.arcadedb.graph.Vertex;
+import com.arcadedb.integration.TestHelper;
+import com.arcadedb.query.sql.executor.Result;
+import com.arcadedb.query.sql.executor.ResultSet;
+import com.arcadedb.utility.FileUtils;
+import org.junit.jupiter.api.Test;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.concurrent.atomic.AtomicInteger;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class FastTextVectorImportTest extends com.arcadedb.TestHelper
+{
+    @Test
+    public void vectorNeighborsQuery() {
+        database.command("sql", "import database file://src/test/resources/cc.en.300.small.vec.gz "  //
+                + "with distanceFunction = cosine, m = 16, ef = 128, efConstruction = 128, " //
+                + "vertexType = Word, edgeType = Proximity, vectorProperty = vector, idProperty = name" //
+        );
+        assertThat(database.countType("Word", true)).isEqualTo(1000);
+
+        final ResultSet rs = database.command("SQL",
+                "select expand(vectorNeighbors('Word[name,vector]','with',10))");
+
+        final AtomicInteger total = new AtomicInteger();
+        while (rs.hasNext()) {
+            final Result record = rs.next();
+            assertThat(record).isNotNull();
+            Vertex vertex = (Vertex) record.getElementProperty("vertex");
+            Float distance = record.getProperty("distance");
+            total.incrementAndGet();
+        }
+
+        assertThat(total.get()).isEqualTo(10);
+    }
+
+    @Test
+    public void parsingLimitEntries() {
+        database.command("sql", "import database file://src/test/resources/cc.en.300.small.vec.gz "  //
+                + "with distanceFunction = cosine, m = 16, ef = 128, efConstruction = 128, " //
+                + "vertexType = Word, edgeType = Proximity, vectorProperty = vector, idProperty = name, "
+                + "parsingLimitEntries = 101"
+        );
+
+        // The header is skipped, so we expect 100 entries
+        assertThat(database.countType("Word", true)).isEqualTo(100);
+    }
+
+    @Override
+    protected String getDatabasePath() {
+        return "target/databases/test-fasttextsmall";
+    }
+}