diff options
| -rw-r--r-- | src/config/mod.rs | 10 | ||||
| -rw-r--r-- | src/main.rs | 1 |
2 files changed, 8 insertions, 3 deletions
diff --git a/src/config/mod.rs b/src/config/mod.rs index 644d162..ebbd909 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -82,13 +82,17 @@ impl Config { let mut config: Config = serde_yaml::from_str(&content) .with_context(|| format!("Invalid config at {}", config_path.display()))?; config.load_roles()?; - if config.conversation_first { - config.start_conversation()?; - } Ok(config) } + pub fn on_repl(&mut self) -> Result<()> { + if self.conversation_first { + self.start_conversation()?; + } + Ok(()) + } + pub fn find_role(&self, name: &str) -> Option<Role> { self.roles.iter().find(|v| v.name == name).cloned() } diff --git a/src/main.rs b/src/main.rs index b940053..b198d14 100644 --- a/src/main.rs +++ b/src/main.rs @@ -98,6 +98,7 @@ fn start_directive( fn start_interactive(client: ChatGptClient, config: SharedConfig) -> Result<()> { cl100k_base_singleton(); + config.lock().on_repl()?; let mut repl = Repl::init(config.clone())?; repl.run(client, config) } |