diff --git a/crates/g3-cli/src/lib.rs b/crates/g3-cli/src/lib.rs index b3a4e9c..f0e6835 100644 --- a/crates/g3-cli/src/lib.rs +++ b/crates/g3-cli/src/lib.rs @@ -440,6 +440,7 @@ pub async fn run() -> Result<()> { cli.task.clone(), cli.chrome_headless, cli.safari, + cli.auto_memory, ) .await; } @@ -455,6 +456,7 @@ pub async fn run() -> Result<()> { cli.task.clone(), cli.chrome_headless, cli.safari, + cli.auto_memory, ) .await; } @@ -701,6 +703,7 @@ async fn run_agent_mode( task: Option, chrome_headless: bool, safari: bool, + auto_memory: bool, ) -> Result<()> { use g3_core::get_agent_system_prompt; use g3_core::find_incomplete_agent_session; @@ -832,6 +835,11 @@ async fn run_agent_mode( // Set agent mode for session tracking agent.set_agent_mode(agent_name); + // Apply auto-memory flag if enabled + if auto_memory { + agent.set_auto_memory(true); + } + // If resuming a session, restore context and TODO let initial_task = if let Some(ref incomplete_session) = resuming_session { // Restore the session context @@ -886,6 +894,11 @@ async fn run_agent_mode( let _result = agent.execute_task(final_task, None, true).await?; + // Send auto-memory reminder if enabled and tools were called + if let Err(e) = agent.send_auto_memory_reminder().await { + debug!("Auto-memory reminder failed: {}", e); + } + // Save session continuation for resume capability agent.save_session_continuation(None);