[NOID] Fixes #4233: Improve Weaviate error handling (#4239)

* Fixes #4233: Improve Weaviate error handling

* fix tests

Author: vga91

full-it/src/test/java/apoc/full/it/vectordb/WeaviateTest.java

@@ -10,13 +10,15 @@
 import static apoc.vectordb.VectorDbTestUtil.EntityType.*;
 import static apoc.vectordb.VectorDbUtil.ERROR_READONLY_MAPPING;
 import static apoc.vectordb.VectorEmbeddingConfig.ALL_RESULTS_KEY;
+import static apoc.vectordb.VectorEmbeddingConfig.FIELDS_KEY;
 import static apoc.vectordb.VectorEmbeddingConfig.MAPPING_KEY;
 import static apoc.vectordb.VectorMappingConfig.*;
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
 import static org.neo4j.configuration.GraphDatabaseSettings.DEFAULT_DATABASE_NAME;
 import static org.neo4j.configuration.GraphDatabaseSettings.SYSTEM_DATABASE_NAME;
@@ -167,6 +169,24 @@ public void writeOperationWithReadOnlyUser() {
             assertThat(e.getMessage()).contains("HTTP response code: 403");
         }
     }
+    
+    @Test
+    public void queryWithWrongEmbeddingSize() {
+        Map<String, Object> conf = map(ALL_RESULTS_KEY, true,
+                FIELDS_KEY, FIELDS,
+                HEADERS_KEY, READONLY_AUTHORIZATION);
+        
+
+        try {
+            testCall(db, "CALL apoc.vectordb.weaviate.query($host, 'TestCollection', [0.2, 0.1, 0.9], null, 5, $conf)",
+                    map("host", HOST, "conf", conf),
+                    r -> fail());
+        } catch (Exception e) {
+            String message = e.getMessage();
+            String expectedErrMsg = "distance between entrypoint and query node: vector lengths don't match: 4 vs 3";
+            assertTrue(message.contains(expectedErrMsg));
+        }
+    }
 
     @Test
     public void getVectorsWithoutVectorResult() {

full/src/main/java/apoc/vectordb/Weaviate.java

@@ -18,6 +18,8 @@
 import java.util.Map;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
+import org.apache.commons.collections4.CollectionUtils;
+import org.apache.commons.lang3.StringUtils;
 import org.neo4j.graphdb.GraphDatabaseService;
 import org.neo4j.graphdb.Transaction;
 import org.neo4j.internal.kernel.api.procs.ProcedureCallContext;
@@ -248,7 +250,13 @@ private Stream<EmbeddingResult> queryCommon(
         VectorEmbeddingConfig conf =
                 DB_HANDLER.getEmbedding().fromQuery(config, procedureCallContext, vector, filter, limit, collection);
         return getEmbeddingResultStream(conf, procedureCallContext, tx, v -> {
-            Object getValue = ((Map<String, Map>) v).get("data").get("Get");
+            Map<String, Map> mapResult = (Map<String, Map>) v;
+            List<Map> errors = (List<Map>) mapResult.get("errors");
+            if (CollectionUtils.isNotEmpty(errors)) {
+                String message = "An error occurred during Weaviate API response: \n" + StringUtils.join(errors, "\n");
+                throw new RuntimeException(message);
+            }
+            Object getValue = mapResult.get("data").get("Get");
             Object collectionValue = ((Map) getValue).get(collection);
             return ((List<Map>) collectionValue).stream().map(i -> {
                 Map additional = (Map) i.remove("_additional");