From a3e2c2a14718b4f23831cea9bce14ee4430dc90b Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 9 Oct 2024 08:28:34 +0800 Subject: refactor: throw error if using non-exist RAG in CMD mode (#917) --- src/config/agent.rs | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/config/agent.rs') 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 { 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")?; -- cgit v1.2.3