From c26594c3c85fb99ad5b0db1c535b337585d3c103 Mon Sep 17 00:00:00 2001 From: sigoden Date: Thu, 6 Jun 2024 05:27:21 +0800 Subject: refactor: config refinement (#570) --- src/rag/mod.rs | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'src/rag/mod.rs') diff --git a/src/rag/mod.rs b/src/rag/mod.rs index 387d3d9..5ab9de5 100644 --- a/src/rag/mod.rs +++ b/src/rag/mod.rs @@ -372,9 +372,8 @@ pub fn split_vector_id(value: VectorID) -> (usize, usize) { } fn retrieve_embedding_model(config: &Config, model_id: &str) -> Result { - let models = list_embedding_models(config); - let model = - Model::find(&models, model_id).ok_or_else(|| anyhow!("No embedding model '{model_id}'"))?; + let model = Model::find(&list_embedding_models(config), model_id) + .ok_or_else(|| anyhow!("No embedding model '{model_id}'"))?; Ok(model) } -- cgit v1.2.3