feat: `.agent` accept variable key-value pairs (#1090)

1 files changed, 4 insertions, 2 deletions

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<String>) -> 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<String>) -> 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)?;