summaryrefslogtreecommitdiffstats
path: root/src/config/mod.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-11-13 09:16:33 +0800
committerGitHub <noreply@github.com>2024-11-13 09:16:33 +0800
commit0aa93fa86da428b08bdc5dba77ca76d213322524 (patch)
tree54769a4da0789d6bedb2aa37229960beac8c4547 /src/config/mod.rs
parent777fc02e0f4e534705c3a0decffd3eebc26ff597 (diff)
downloadaichat-0aa93fa86da428b08bdc5dba77ca76d213322524.tar.gz
refactor: don't init agent RAG/variables when using `--info` (#989)
Diffstat (limited to 'src/config/mod.rs')
-rw-r--r--src/config/mod.rs16
1 files changed, 13 insertions, 3 deletions
diff --git a/src/config/mod.rs b/src/config/mod.rs
index aa0fe00..7c523c5 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -153,6 +153,8 @@ pub struct Config {
#[serde(skip)]
pub working_mode: WorkingMode,
#[serde(skip)]
+ pub print_info_only: bool,
+ #[serde(skip)]
pub last_message: Option<(Input, String)>,
}
@@ -212,6 +214,7 @@ impl Default for Config {
model: Default::default(),
functions: Default::default(),
working_mode: WorkingMode::Cmd,
+ print_info_only: false,
last_message: None,
}
}
@@ -1860,8 +1863,11 @@ impl Config {
Some(v) => v,
None => return Ok(()),
};
- let new_variables =
- Agent::init_agent_variables(agent.defined_variables(), agent.config_variables())?;
+ let new_variables = Agent::init_agent_variables(
+ agent.defined_variables(),
+ agent.config_variables(),
+ self.print_info_only,
+ )?;
agent.set_shared_variables(new_variables);
Ok(())
}
@@ -1878,7 +1884,11 @@ impl Config {
shared_variables.clone()
};
all_variables.extend(session.agent_variables().clone());
- let new_variables = Agent::init_agent_variables(agent.defined_variables(), &all_variables)?;
+ let new_variables = Agent::init_agent_variables(
+ agent.defined_variables(),
+ &all_variables,
+ self.print_info_only,
+ )?;
if shared_variables.is_empty() {
agent.set_shared_variables(new_variables.clone());
}