summaryrefslogtreecommitdiffstats
path: root/src/rag/mod.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-06 05:27:21 +0800
committerGitHub <noreply@github.com>2024-06-06 05:27:21 +0800
commitc26594c3c85fb99ad5b0db1c535b337585d3c103 (patch)
tree1829b2b1eee599fd835f2eb6f4751097c145b802 /src/rag/mod.rs
parent171a46eee2afcb92bcbbd6a11b52ebc82f2476db (diff)
downloadaichat-c26594c3c85fb99ad5b0db1c535b337585d3c103.tar.gz
refactor: config refinement (#570)
Diffstat (limited to 'src/rag/mod.rs')
-rw-r--r--src/rag/mod.rs5
1 files changed, 2 insertions, 3 deletions
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<Model> {
- 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)
}