diff options
Diffstat (limited to 'src/config')
| -rw-r--r-- | src/config/agent.rs | 3 | ||||
| -rw-r--r-- | src/config/mod.rs | 12 |
2 files changed, 12 insertions, 3 deletions
diff --git a/src/config/agent.rs b/src/config/agent.rs index 813df3a..2f44477 100644 --- a/src/config/agent.rs +++ b/src/config/agent.rs @@ -32,6 +32,9 @@ impl Agent { ) -> Result<Self> { let functions_dir = Config::agent_functions_dir(name)?; let definition_file_path = functions_dir.join("index.yaml"); + if !definition_file_path.exists() { + bail!("Unknown agent `{name}`"); + } let functions_file_path = functions_dir.join("functions.json"); let variables_path = Config::agent_variables_file(name)?; let rag_path = Config::agent_rag_file(name, "rag")?; diff --git a/src/config/mod.rs b/src/config/mod.rs index 8e3e2c3..33bde56 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -205,7 +205,7 @@ impl Default for Config { agent: None, model: Default::default(), functions: Default::default(), - working_mode: WorkingMode::Command, + working_mode: WorkingMode::Cmd, last_message: None, } } @@ -1199,6 +1199,9 @@ impl Config { Some(name) => { let rag_path = config.read().rag_file(name)?; if !rag_path.exists() { + if config.read().working_mode.is_cmd() { + bail!("Unknown RAG '{name}'") + } Rag::init(config, name, &rag_path, &[], abort_signal).await? } else { Rag::load(config, name, &rag_path)? @@ -1376,7 +1379,7 @@ impl Config { pub fn apply_prelude(&mut self) -> Result<()> { let prelude = match self.working_mode { - WorkingMode::Command => self.prelude.as_ref(), + WorkingMode::Cmd => self.prelude.as_ref(), WorkingMode::Repl => self.repl_prelude.as_ref().or(self.prelude.as_ref()), WorkingMode::Serve => return Ok(()), }; @@ -2043,12 +2046,15 @@ pub fn load_env_file() -> Result<()> { #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)] pub enum WorkingMode { - Command, + Cmd, Repl, Serve, } impl WorkingMode { + pub fn is_cmd(&self) -> bool { + *self == WorkingMode::Cmd + } pub fn is_repl(&self) -> bool { *self == WorkingMode::Repl } |