searchengines-web/MeiliSearch
searchengines-web/MeiliSearch
1
0
Fork 0
mirror of https://github.com/meilisearch/MeiliSearch synced 2025-07-04 20:37:15 +02:00
Code Issues Releases Wiki Activity

Display the _semanticSimilarity even if the _vectors field is not displayed

Browse source
This commit is contained in:
Kerollmops 2023-06-20 15:54:28 +02:00 • committed by Clément Renault
parent 737aec1705
commit 7aa1275337
No known key found for this signature in database
GPG key ID: 92ADA4E935E71FA4
3 changed files with 53 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

11
milli/src/update/index_documents/extract/extract_vector_points.rs
View file

@ -3,11 +3,10 @@ use std::fs::File;
use std::io;
use bytemuck::cast_slice;
use either::Either;
use serde_json::from_slice;
use super::helpers::{create_writer, writer_into_reader, GrenadParameters};
use crate::{FieldId, InternalError, Result};
use crate::{FieldId, InternalError, Result, VectorOrArrayOfVectors};
/// Extracts the embedding vector contained in each document under the `_vectors` field.
///
@ -31,9 +30,11 @@ pub fn extract_vector_points<R: io::Read + io::Seek>(
// first we retrieve the _vectors field
if let Some(vectors) = obkv.get(vectors_fid) {
// extract the vectors
let vectors: Either<Vec<Vec<f32>>, Vec<f32>> =
from_slice(vectors).map_err(InternalError::SerdeJson).unwrap();
let vectors = vectors.map_right(|v| vec![v]).into_inner();
// TODO return a user error before unwrapping
let vectors = from_slice(vectors)
.map_err(InternalError::SerdeJson)
.map(VectorOrArrayOfVectors::into_array_of_vectors)
.unwrap();
for (i, vector) in vectors.into_iter().enumerate() {
match u16::try_from(i) {