From 6d148c9c53cd4ac7849f52b956aba87e84cc3d4d Mon Sep 17 00:00:00 2001 From: sigoden Date: Sun, 23 Jun 2024 07:17:26 +0800 Subject: refactor: embedding model add price and dimension (#636) --- src/rag/mod.rs | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'src/rag/mod.rs') diff --git a/src/rag/mod.rs b/src/rag/mod.rs index 116beea..30ca92f 100644 --- a/src/rag/mod.rs +++ b/src/rag/mod.rs @@ -548,9 +548,12 @@ pub fn split_document_id(value: DocumentId) -> (usize, usize) { } fn select_embedding_model(models: &[&Model]) -> Result { - let model_ids: Vec<_> = models.iter().map(|v| v.id()).collect(); - let model_id = Select::new("Select embedding model:", model_ids).prompt()?; - Ok(model_id) + let models: Vec<_> = models + .iter() + .map(|v| SelectOption::new(v.id(), v.description())) + .collect(); + let result = Select::new("Select embedding model:", models).prompt()?; + Ok(result.value) } fn set_chunk_size(model: &Model) -> Result { -- cgit v1.2.3