From e5facdba197c76e176eefb8ced7089f63a604ea0 Mon Sep 17 00:00:00 2001 From: Andrey Vasnetsov Date: Wed, 19 Jun 2024 15:18:48 +0200 Subject: [PATCH] fix: replace vector_count with dim (#4505) --- lib/api/src/grpc/conversions.rs | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/lib/api/src/grpc/conversions.rs b/lib/api/src/grpc/conversions.rs index a4bb602446..963d5dffdd 100644 --- a/lib/api/src/grpc/conversions.rs +++ b/lib/api/src/grpc/conversions.rs @@ -700,10 +700,12 @@ impl TryFrom for segment_vectors::VectorStructInternal { )); } if let Some(vectors_count) = vectors_count { + let dim = data.len() / vectors_count as usize; + segment_vectors::VectorStructInternal::MultiDense( segment::data_types::vectors::MultiDenseVectorInternal::try_from_flatten( data, - vectors_count as usize, + dim, ).map_err(|err| { Status::invalid_argument(format!("Unable to convert to multi-dense vector: {err}")) })?,