Removed unused imports and fixed StartupExtendedTest

neo4j-contrib · Jul 24, 2023 · 3e66a1e · 3e66a1e
1 parent 3ffb31c
commit 3e66a1e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/extended/src/main/java/apoc/ml/Prompt.java b/extended/src/main/java/apoc/ml/Prompt.java
@@ -6,7 +6,6 @@
 import com.fasterxml.jackson.core.JsonProcessingException;
 import org.jetbrains.annotations.NotNull;
 import org.neo4j.graphdb.QueryExecutionException;
-import org.neo4j.graphdb.Result;
 import org.neo4j.graphdb.Transaction;
 import org.neo4j.internal.kernel.api.procs.ProcedureCallContext;
 import org.neo4j.logging.Log;
@@ -20,7 +19,6 @@
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
-import java.util.concurrent.atomic.AtomicReference;
 import java.util.stream.Collectors;
 import java.util.stream.LongStream;
 import java.util.stream.Stream;

diff --git a/extended/src/main/resources/extended.txt b/extended/src/main/resources/extended.txt
@@ -73,6 +73,9 @@ apoc.log.warn
 apoc.metrics.get
 apoc.metrics.list
 apoc.metrics.storage
+apoc.ml.cypher
+apoc.ml.query
+apoc.ml.schema
 apoc.ml.openai.chat
 apoc.ml.openai.completion
 apoc.ml.openai.embedding