diff options
| author | sigoden <sigoden@gmail.com> | 2024-11-13 09:16:33 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-11-13 09:16:33 +0800 |
| commit | 0aa93fa86da428b08bdc5dba77ca76d213322524 (patch) | |
| tree | 54769a4da0789d6bedb2aa37229960beac8c4547 /src/config/mod.rs | |
| parent | 777fc02e0f4e534705c3a0decffd3eebc26ff597 (diff) | |
| download | aichat-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.rs | 16 |
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()); } |
