Fix vector parsing

This commit is contained in:
Louis Dureuil 2024-11-07 22:35:06 +01:00
parent d97af4d8e6
commit 4706a0eb49
No known key found for this signature in database
2 changed files with 90 additions and 5 deletions

View file

@ -167,7 +167,7 @@ fn entry_from_raw_value(
value: &RawValue,
has_configured_embedder: bool,
) -> std::result::Result<VectorEntry<'_>, serde_json::Error> {
let value: RawVectors = serde_json::from_str(value.get())?;
let value: RawVectors = RawVectors::from_raw_value(value)?;
Ok(match value {
RawVectors::Explicit(raw_explicit_vectors) => VectorEntry {
@ -177,7 +177,7 @@ fn entry_from_raw_value(
},
RawVectors::ImplicitlyUserProvided(value) => VectorEntry {
has_configured_embedder,
embeddings: Some(Embeddings::FromJsonImplicityUserProvided(value)),
embeddings: value.map(Embeddings::FromJsonImplicityUserProvided),
regenerate: false,
},
})