daniel543x/artcraft
1
0 Fork 0
mirror of https://github.com/storytold/artcraft.git synced 2026-11-03 17:08:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

query exact indices in elasticsearch

Browse Source
This commit is contained in:
Brandon Thomas committed 2024-01-11 04:57:08 -05:00
1 parent 95987860f4
commit 792c109bb6
2 files changed
+10 -5
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

crates/lib/elasticsearch_schema/src/searches/search_model_weights.rs
+8 -3
View File
@@ -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::<ModelWeightDocument>(value)?;
let document = serde_json::from_value::<ModelWeightDocument>(value)
.map_err(|err| {
error!("Error: {:?}", &err);
err
})?;
documents.push(document);
}
}
crates/lib/elasticsearch_schema/src/searches/search_tts_models.rs
+2 -2
View File
@@ -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()