From f9e7efd41c475519d8f7a01c539d9ee4a3d56208 Mon Sep 17 00:00:00 2001 From: sigoden Date: Sun, 23 Jun 2024 22:09:47 +0800 Subject: refactor: reorder config items (#639) --- src/config/mod.rs | 69 +++++++++++++++++++++++++++++++++---------------------- 1 file changed, 42 insertions(+), 27 deletions(-) (limited to 'src/config') diff --git a/src/config/mod.rs b/src/config/mod.rs index 4c0e1e7..0c1c6ac 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -86,36 +86,44 @@ pub struct Config { pub model_id: String, pub temperature: Option, pub top_p: Option, + pub dry_run: bool, pub save: bool, - pub save_session: Option, - pub highlight: bool, - pub light_theme: bool, + pub keybindings: Keybindings, + pub buffer_editor: Option, pub wrap: Option, pub wrap_code: bool, - pub keybindings: Keybindings, + pub prelude: Option, pub repl_prelude: Option, - pub buffer_editor: Option, + pub agent_prelude: Option, + + pub save_session: Option, + pub compress_threshold: usize, + pub summarize_prompt: Option, + pub summary_prompt: Option, + pub function_calling: bool, pub dangerously_functions_filter: Option, - pub agent_prelude: Option, pub agents: Vec, + pub rag_embedding_model: Option, pub rag_rerank_model: Option, + pub rag_top_k: usize, pub rag_chunk_size: Option, pub rag_chunk_overlap: Option, - pub rag_top_k: usize, pub rag_min_score_vector_search: f32, pub rag_min_score_keyword_search: f32, pub rag_min_score_rerank: f32, pub rag_template: Option, - pub compress_threshold: usize, - pub summarize_prompt: Option, - pub summary_prompt: Option, + + pub highlight: bool, + pub light_theme: bool, pub left_prompt: Option, pub right_prompt: Option, + pub clients: Vec, + #[serde(skip)] pub roles: Vec, #[serde(skip)] @@ -142,36 +150,44 @@ impl Default for Config { model_id: Default::default(), temperature: None, top_p: None, - save: false, - save_session: None, - highlight: true, + dry_run: false, - light_theme: false, + save: false, + keybindings: Default::default(), + buffer_editor: None, wrap: None, wrap_code: false, - keybindings: Default::default(), + prelude: None, repl_prelude: None, - buffer_editor: None, + agent_prelude: None, + function_calling: false, dangerously_functions_filter: None, - agent_prelude: None, agents: vec![], + rag_embedding_model: None, rag_rerank_model: None, + rag_top_k: 4, rag_chunk_size: None, rag_chunk_overlap: None, - rag_top_k: 4, rag_min_score_vector_search: 0.0, rag_min_score_keyword_search: 0.0, rag_min_score_rerank: 0.0, rag_template: None, + + save_session: None, compress_threshold: 4000, summarize_prompt: None, summary_prompt: None, + + highlight: true, + light_theme: false, left_prompt: None, right_prompt: None, + clients: vec![], + roles: vec![], role: None, session: None, @@ -453,22 +469,21 @@ impl Config { ), ("temperature", format_option_value(&role.temperature())), ("top_p", format_option_value(&role.top_p())), + ("dry_run", self.dry_run.to_string()), + ("save", self.save.to_string()), + ("keybindings", self.keybindings.stringify().into()), + ("wrap", wrap), + ("wrap_code", self.wrap_code.to_string()), + ("save_session", format_option_value(&self.save_session)), + ("compress_threshold", self.compress_threshold.to_string()), + ("function_calling", self.function_calling.to_string()), ( "rag_rerank_model", format_option_value(&self.rag_rerank_model), ), ("rag_top_k", self.rag_top_k.to_string()), - ("function_calling", self.function_calling.to_string()), - ("compress_threshold", self.compress_threshold.to_string()), - ("dry_run", self.dry_run.to_string()), - ("save", self.save.to_string()), - ("save_session", format_option_value(&self.save_session)), ("highlight", self.highlight.to_string()), ("light_theme", self.light_theme.to_string()), - ("wrap", wrap), - ("wrap_code", self.wrap_code.to_string()), - ("keybindings", self.keybindings.stringify().into()), - ("prelude", format_option_value(&self.prelude)), ("config_file", display_path(&Self::config_file()?)), ("roles_file", display_path(&Self::roles_file()?)), ("functions_dir", display_path(&Self::functions_dir()?)), -- cgit v1.2.3