chore: upadate namedVectors() to accept Map<String, Vector>

qdrant · Dec 29, 2023 · 5139936 · 5139936
1 parent ad7bf6b
commit 5139936
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main/java/io/qdrant/client/VectorsFactory.java b/src/main/java/io/qdrant/client/VectorsFactory.java
@@ -7,6 +7,7 @@
 
 import static io.qdrant.client.VectorFactory.vector;
 import static io.qdrant.client.grpc.Points.NamedVectors;
+import io.qdrant.client.grpc.Points.Vector;
 import static io.qdrant.client.grpc.Points.Vectors;
 
 /**
@@ -18,13 +19,13 @@ private VectorsFactory() {
 
 	/**
 	 * Creates named vectors
-	 * @param values A map of vector names to values
+	 * @param values A map of vector names to {@link Vector}
 	 * @return a new instance of {@link Vectors}
 	 */
-	public static Vectors namedVectors(Map<String, List<Float>> values) {
+	public static Vectors namedVectors(Map<String, Vector> values) {
 		return Vectors.newBuilder()
 			.setVectors(NamedVectors.newBuilder()
-				.putAllVectors(Maps.transformValues(values, v -> vector(v)))
+				.putAllVectors(Maps.transformValues(values, v -> v))
 			)
 			.build();
 	}