summaryrefslogtreecommitdiffstats
path: root/src/rag
diff options
context:
space:
mode:
Diffstat (limited to 'src/rag')
-rw-r--r--src/rag/mod.rs18
1 files changed, 18 insertions, 0 deletions
diff --git a/src/rag/mod.rs b/src/rag/mod.rs
index 8855f4f..6165969 100644
--- a/src/rag/mod.rs
+++ b/src/rag/mod.rs
@@ -842,6 +842,24 @@ fn select_embedding_model(models: &[&Model]) -> Result<String> {
Ok(result.value)
}
+#[derive(Debug)]
+struct SelectOption {
+ pub value: String,
+ pub description: String,
+}
+
+impl SelectOption {
+ pub fn new(value: String, description: String) -> Self {
+ Self { value, description }
+ }
+}
+
+impl std::fmt::Display for SelectOption {
+ fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+ write!(f, "{} ({})", self.value, self.description)
+ }
+}
+
fn set_chunk_size(model: &Model) -> Result<usize> {
let default_value = model.default_chunk_size().to_string();
let help_message = model