kent/consciousness
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

cli: route agent run through daemon RPC when available

Browse source 
Previously 'poc-memory agent run <agent> --count N' always ran locally,
loading the full store and executing synchronously. This was slow and
bypassed the daemon's concurrency control and persistent task queue.

Now the CLI checks for a running daemon first and queues via RPC
(returning instantly) unless --local, --debug, or --dry-run is set.
Falls back to local execution if the daemon isn't running.

This also avoids the expensive Store::load() on the fast path.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Kent Overstreet 2026-03-21 15:04:47 -04:00
parent a321f87db6
commit 3fd485a2e9
1 changed files with 18 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

26
poc-memory/src/cli/agent.rs
View file

@ -8,6 +8,19 @@ pub fn cmd_run_agent(agent: &str, count: usize, target: &[String], query: Option
// SAFETY: single-threaded at this point (CLI startup, before any agent work) // SAFETY: single-threaded at this point (CLI startup, before any agent work)
unsafe { std::env::set_var("POC_MEMORY_DRY_RUN", "1"); } unsafe { std::env::set_var("POC_MEMORY_DRY_RUN", "1"); }
} }
let needs_local = local || debug || dry_run;
let has_targets = !target.is_empty() || query.is_some();
// Fast path: no explicit targets, daemon available — just queue via RPC
if !needs_local && !has_targets {
if crate::agents::daemon::send_rpc_pub("ping").is_some() {
return crate::agents::daemon::rpc_run_agent(agent, count);
}
eprintln!("Daemon not running — falling back to local execution");
}
// Slow path: need the store for local execution or target resolution
let mut store = store::Store::load()?; let mut store = store::Store::load()?;
let log = |msg: &str| eprintln!("[{}] {}", agent, msg); let log = |msg: &str| eprintln!("[{}] {}", agent, msg);
@ -30,8 +43,8 @@ pub fn cmd_run_agent(agent: &str, count: usize, target: &[String], query: Option
if !resolved_targets.is_empty() { if !resolved_targets.is_empty() {
// --local or daemon unavailable: run directly // --local or daemon unavailable: run directly
if local || crate::agents::daemon::send_rpc_pub("ping").is_none() { if needs_local || crate::agents::daemon::send_rpc_pub("ping").is_none() {
if !local { if !needs_local {
eprintln!("Daemon not running — falling back to local execution"); eprintln!("Daemon not running — falling back to local execution");
} }
for (i, key) in resolved_targets.iter().enumerate() { for (i, key) in resolved_targets.iter().enumerate() {
@ -55,13 +68,10 @@ pub fn cmd_run_agent(agent: &str, count: usize, target: &[String], query: Option
} }
} }
eprintln!("[{}] queued {} tasks to daemon", agent, queued); eprintln!("[{}] queued {} tasks to daemon", agent, queued);
} else if debug {
crate::agents::knowledge::run_one_agent(
&mut store, agent, count, "test", &log, true,
)?;
} else { } else {
crate::agents::knowledge::run_and_apply_with_log( // Local execution (--local, --debug, dry-run, or daemon unavailable)
&mut store, agent, count, "test", &log, crate::agents::knowledge::run_one_agent(
&mut store, agent, count, "test", &log, debug,
)?; )?;
} }
Ok(()) Ok(())