From 503c282be5d5b445ccb2f0a68f50995ab27ad932 Mon Sep 17 00:00:00 2001 From: sigoden Date: Thu, 7 Nov 2024 06:15:33 +0800 Subject: refactor: init agent variables (#977) --- src/config/mod.rs | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) (limited to 'src/config/mod.rs') diff --git a/src/config/mod.rs b/src/config/mod.rs index 6736624..e88ac5c 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -1062,8 +1062,8 @@ impl Config { session.exit(&sessions_dir, self.working_mode.is_repl())?; self.last_message = None; } - if self.agent.is_some() { - self.init_agent_shared_variables()?; + if let Some(agent) = self.agent.as_mut() { + agent.set_session_variables(None); } Ok(()) } @@ -1863,9 +1863,8 @@ impl Config { None => return Ok(()), }; let new_variables = - Agent::init_agent_variables(agent.defined_variables(), agent.shared_variables())?; + Agent::init_agent_variables(agent.defined_variables(), agent.config_variables())?; agent.set_shared_variables(new_variables); - agent.set_session_variables(None); Ok(()) } @@ -1875,9 +1874,16 @@ impl Config { _ => return Ok(()), }; let shared_variables = agent.shared_variables(); - let mut all_variables = shared_variables.clone(); + let mut all_variables = if shared_variables.is_empty() { + agent.config_variables().clone() + } else { + shared_variables.clone() + }; all_variables.extend(session.agent_variables().clone()); let new_variables = Agent::init_agent_variables(agent.defined_variables(), &all_variables)?; + if shared_variables.is_empty() { + agent.set_shared_variables(new_variables.clone()); + } agent.set_session_variables(Some(new_variables)); session.sync_agent(agent, false); Ok(()) -- cgit v1.2.3