From 0aa93fa86da428b08bdc5dba77ca76d213322524 Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 13 Nov 2024 09:16:33 +0800 Subject: refactor: don't init agent RAG/variables when using `--info` (#989) --- src/config/agent.rs | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'src/config/agent.rs') diff --git a/src/config/agent.rs b/src/config/agent.rs index e16e7a7..49d5835 100644 --- a/src/config/agent.rs +++ b/src/config/agent.rs @@ -66,7 +66,7 @@ impl Agent { let rag = if rag_path.exists() { Some(Arc::new(Rag::load(config, DEFAULT_AGENT_NAME, &rag_path)?)) - } else if !definition.documents.is_empty() { + } else if !definition.documents.is_empty() && !config.read().print_info_only { let mut ans = false; if *IS_STDOUT_TERMINAL { ans = Confirm::new("The agent has the documents, init RAG?") @@ -109,6 +109,7 @@ impl Agent { pub fn init_agent_variables( agent_variables: &[AgentVariable], variables: &IndexMap, + no_interaction: bool, ) -> Result> { let mut output = IndexMap::new(); if agent_variables.is_empty() { @@ -127,6 +128,9 @@ impl Agent { output.insert(key, value); continue; } + if no_interaction { + continue; + } if *IS_STDOUT_TERMINAL { if !printed { println!("⚙ Init agent variables..."); -- cgit v1.2.3