diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-06 05:27:21 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-06 05:27:21 +0800 |
| commit | c26594c3c85fb99ad5b0db1c535b337585d3c103 (patch) | |
| tree | 1829b2b1eee599fd835f2eb6f4751097c145b802 /src/rag | |
| parent | 171a46eee2afcb92bcbbd6a11b52ebc82f2476db (diff) | |
| download | aichat-c26594c3c85fb99ad5b0db1c535b337585d3c103.tar.gz | |
refactor: config refinement (#570)
Diffstat (limited to 'src/rag')
| -rw-r--r-- | src/rag/mod.rs | 5 |
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) } |
