diff --git a/crates/lib/elasticsearch_schema/src/searches/search_model_weights.rs b/crates/lib/elasticsearch_schema/src/searches/search_model_weights.rs index d92b7e825e..d6caad142f 100644 --- a/crates/lib/elasticsearch_schema/src/searches/search_model_weights.rs +++ b/crates/lib/elasticsearch_schema/src/searches/search_model_weights.rs @@ -1,9 +1,10 @@ use elasticsearch::{Elasticsearch, SearchParts}; +use log::error; use serde_json::{json, Value}; use errors::AnyhowResult; -use crate::documents::model_weight_document::ModelWeightDocument; +use crate::documents::model_weight_document::{MODEL_WEIGHT_INDEX, ModelWeightDocument}; pub async fn search_model_weights( client: &Elasticsearch, @@ -17,7 +18,7 @@ pub async fn search_model_weights( }; let search_response = client - .search(SearchParts::None) + .search(SearchParts::Index(&[MODEL_WEIGHT_INDEX])) .body(search_json) .allow_no_indices(true) .send() @@ -43,7 +44,11 @@ pub async fn search_model_weights( let maybe_object = hit.get_mut("_source") .map(|source| source.take()); if let Some(value) = maybe_object { - let document = serde_json::from_value::(value)?; + let document = serde_json::from_value::(value) + .map_err(|err| { + error!("Error: {:?}", &err); + err + })?; documents.push(document); } } diff --git a/crates/lib/elasticsearch_schema/src/searches/search_tts_models.rs b/crates/lib/elasticsearch_schema/src/searches/search_tts_models.rs index 30055f1774..381208694b 100644 --- a/crates/lib/elasticsearch_schema/src/searches/search_tts_models.rs +++ b/crates/lib/elasticsearch_schema/src/searches/search_tts_models.rs @@ -3,7 +3,7 @@ use serde_json::{json, Value}; use errors::AnyhowResult; -use crate::documents::tts_model_document::TtsModelDocument; +use crate::documents::tts_model_document::{TTS_MODEL_INDEX, TtsModelDocument}; pub async fn search_tts_models( client: &Elasticsearch, @@ -17,7 +17,7 @@ pub async fn search_tts_models( }; let search_response = client - .search(SearchParts::None) + .search(SearchParts::Index(&[TTS_MODEL_INDEX])) .body(search_json) .allow_no_indices(true) .send()