Refactor model
paydii committed
Apr 25, 2026 at 21:51 UTC
cfc52c77cd7984f4dcde408a01db425eb7592b47
2 files changed
+9
-4
src/chat.rs
+6
-1
index 1de1824..fca334d 100644
--- a/src/chat.rs
+++ b/src/chat.rs
@@ -459,7 +459,12 @@ fn render_model_picker(frame: &mut Frame, app: &App, area: ratatui::layout::Rect
ModelCacheHealth::Complete => "",
ModelCacheHealth::Incomplete => " (unselectable)",
};
- let source = format!(" [{} {}]", item.brand_mark, item.source_label);
+ let brand_mark = match item.source {
+ ModelSource::Onde => "◉",
+ ModelSource::HuggingFace => "○",
+ ModelSource::Fallback => "◎",
+ };
+ let source = format!(" [{} {}]", brand_mark, item.source_label);
let base_style = if selected {
Style::default().fg(Color::Black).bg(Color::Green)
src/models.rs
+3
-3
index 3e18909..c04b53c 100644
--- a/src/models.rs
+++ b/src/models.rs
@@ -27,7 +27,7 @@ pub(crate) struct ModelPickerItem {
pub(crate) max_tokens: u64,
pub(crate) config: GgufModelConfig,
pub(crate) source_label: String,
- pub(crate) brand_mark: &'static str,
+
pub(crate) source: ModelSource,
pub(crate) cache_health: ModelCacheHealth,
}
@@ -66,7 +66,7 @@ pub(crate) fn build_model_picker_items() -> Vec<ModelPickerItem> {
max_tokens,
config,
source_label: "Platform default".to_string(),
- brand_mark: "◎",
+
source: ModelSource::Fallback,
cache_health: ModelCacheHealth::Complete,
});
@@ -111,7 +111,7 @@ fn discovered_model_to_picker_item(model: DiscoveredModel) -> Option<ModelPicker
max_tokens,
config,
source_label,
- brand_mark: if model.from_app_group { "◉" } else { "○" },
+
source: if model.from_app_group {
ModelSource::Onde
} else {