fix: add --auto-memory support to agent mode
The --auto-memory flag was not being passed to run_agent_mode() and send_auto_memory_reminder() was not being called after agent task execution. Changes: - Pass auto_memory parameter to run_agent_mode() - Add auto_memory parameter to run_agent_mode() function signature - Call agent.set_auto_memory(true) when flag is enabled - Call send_auto_memory_reminder() after execute_task() in agent mode
This commit is contained in:
@@ -440,6 +440,7 @@ pub async fn run() -> Result<()> {
|
|||||||
cli.task.clone(),
|
cli.task.clone(),
|
||||||
cli.chrome_headless,
|
cli.chrome_headless,
|
||||||
cli.safari,
|
cli.safari,
|
||||||
|
cli.auto_memory,
|
||||||
)
|
)
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
@@ -455,6 +456,7 @@ pub async fn run() -> Result<()> {
|
|||||||
cli.task.clone(),
|
cli.task.clone(),
|
||||||
cli.chrome_headless,
|
cli.chrome_headless,
|
||||||
cli.safari,
|
cli.safari,
|
||||||
|
cli.auto_memory,
|
||||||
)
|
)
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
@@ -701,6 +703,7 @@ async fn run_agent_mode(
|
|||||||
task: Option<String>,
|
task: Option<String>,
|
||||||
chrome_headless: bool,
|
chrome_headless: bool,
|
||||||
safari: bool,
|
safari: bool,
|
||||||
|
auto_memory: bool,
|
||||||
) -> Result<()> {
|
) -> Result<()> {
|
||||||
use g3_core::get_agent_system_prompt;
|
use g3_core::get_agent_system_prompt;
|
||||||
use g3_core::find_incomplete_agent_session;
|
use g3_core::find_incomplete_agent_session;
|
||||||
@@ -832,6 +835,11 @@ async fn run_agent_mode(
|
|||||||
// Set agent mode for session tracking
|
// Set agent mode for session tracking
|
||||||
agent.set_agent_mode(agent_name);
|
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
|
// If resuming a session, restore context and TODO
|
||||||
let initial_task = if let Some(ref incomplete_session) = resuming_session {
|
let initial_task = if let Some(ref incomplete_session) = resuming_session {
|
||||||
// Restore the session context
|
// Restore the session context
|
||||||
@@ -886,6 +894,11 @@ async fn run_agent_mode(
|
|||||||
|
|
||||||
let _result = agent.execute_task(final_task, None, true).await?;
|
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
|
// Save session continuation for resume capability
|
||||||
agent.save_session_continuation(None);
|
agent.save_session_continuation(None);
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user