From d3ae5612a51ba6a5d31dfabfe7f638afb8a35f24 Mon Sep 17 00:00:00 2001 From: sigoden Date: Fri, 17 Jan 2025 22:06:57 +0800 Subject: feat: `.agent` accept variable key-value pairs (#1090) --- src/main.rs | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'src/main.rs') diff --git a/src/main.rs b/src/main.rs index 48eb2b9..a7b5198 100644 --- a/src/main.rs +++ b/src/main.rs @@ -101,7 +101,7 @@ async fn run(config: GlobalConfig, cli: Cli, text: Option) -> Result<()> None => TEMP_SESSION_NAME, }); if !cli.agent_variable.is_empty() { - config.write().cli_agent_variables = Some( + config.write().agent_variables = Some( cli.agent_variable .chunks(2) .map(|v| (v[0].to_string(), v[1].to_string())) @@ -109,7 +109,9 @@ async fn run(config: GlobalConfig, cli: Cli, text: Option) -> Result<()> ); } - Config::use_agent(&config, agent, session, abort_signal.clone()).await? + let ret = Config::use_agent(&config, agent, session, abort_signal.clone()).await; + config.write().agent_variables = None; + ret?; } else { if let Some(prompt) = &cli.prompt { config.write().use_prompt(prompt)?; -- cgit v1.2.3