magi/deputy.rs
1//! Deputies: the seat that waits on a conductor's question, so the conductor
2//! never has to - and on land's merge approval, whose free-text reply used to
3//! reach nobody either.
4//!
5//! [`crate::conduct`] files a question for the operator and moves on; it is
6//! non-blocking by construction, because one task's question must not park the
7//! whole polling loop. The price used to be that nothing at all was waiting on
8//! it: the owner's free-text reply landed in a thread no agent read, while the
9//! phone said "waiting for the agent". Question f1dc on task 684d sat open
10//! forever on a "setup done".
11//!
12//! A deputy is the answer. For each open conductor question, `magi serve`
13//! runs one short-lived agent seat that is handed what the conductor knew
14//! ([`brief`]: the task, why it asked, what each option leads to), blocks on
15//! the question with `magi ask --wait`, answers back in the thread with
16//! `magi ask --thread`, and - when the owner's own words clearly pick an
17//! option - records that with `magi ask --settle`. Applying the outcome is not
18//! the deputy's job: an answered conductor question goes through the daemon's
19//! existing `resolve_blockers` / action path, exactly as one tapped on the
20//! phone does.
21//!
22//! # Invariants
23//!
24//! - **One seat per question, keyed by seat name.** `deputy-<question id>`,
25//! never the conductor's shared seat (`conduct/seat.json`, overwritten by
26//! every cycle) and never an agent id.
27//! - **Persisted on the question** ([`crate::ask::Deputy`]), always through
28//! [`Questions::update`]: the owner's say and answer write the same file. A
29//! restarted daemon resumes the seat only when [`agent::has_session`] says
30//! it can; otherwise it starts a fresh seat and re-sends the whole context,
31//! never assuming memory.
32//! - **Bounded.** At most `daemon.max_deputies` at once; at most
33//! [`MAX_STARTS`] starts per question, and a restart does not reset that. A
34//! deputy never outlives the question's `answer_timeout`: the waiter retires
35//! the question at its deadline ([`crate::waiter`]) and the task goes to a
36//! machine hold (`daemon::resolve_blockers`).
37//! - **Never a second agent on one question.** A fresh lease or the claim file
38//! means somebody is already on it.
39//! - **A merge approval is served too, but stays land's.** Its deputy has no
40//! `cwd` (so the waiter never touches it), its deadline is `asked_at +
41//! answer_timeout` and never moves on a reply ([`deadline`]), and the only
42//! thing that retires it is `daemon::land_resume_state`. A say alone never
43//! merges: `--settle` accepts `merge` only for the owner's own word `merge`.
44//! - **A release-watch question is served too, and stays the watcher's.** The
45//! escalation, local-mode approval and failed-release questions
46//! ([`crate::release_watch`], node `release-bump`, seat `release-watch`) have
47//! no asker, so a say reached nobody. Its deputy has no `cwd` either, a
48//! fixed `asked_at + answer_timeout` deadline ([`deadline`]), and applies
49//! nothing: `--settle` records a choice and the watcher applies it on its next
50//! lap. A local approval's `merge` / `hold` go through the merge-approval
51//! rules ([`merge_gated`]). A deputy never closes or merges the pull request.
52//! - **Every open question the owner can say something to has a listener.**
53//! [`Kind::Triage`] serves the triage questions; [`Kind::Generic`] is the
54//! fallback for any other question with choices and no `cwd` (the divergence
55//! question today, a node nobody has written yet tomorrow). It is decided by
56//! the missing `cwd`, never by node name. Only a conductor question is ever
57//! given a `cwd` ([`Deputies::attach`]); the other kinds would otherwise be
58//! resumed and expired by the waiter beside the deputy. A test enumerates
59//! every node filed without an asker, so a new kind of question cannot ship
60//! without one.
61//! - **No new authority.** A deputy does not edit, merge or touch the queue,
62//! and `--settle` accepts only an offered label backed by a verbatim quote
63//! of the owner, never on a task the operator holds.
64
65use std::collections::{BTreeMap, HashMap, HashSet};
66use std::path::{Path, PathBuf};
67use std::sync::Arc;
68use std::time::{Duration, Instant};
69
70use anyhow::{Context as _, Result};
71use jiff::Timestamp;
72use tokio::task::JoinSet;
73
74use crate::agent::{self, Invocation, SeatState};
75use crate::ask::{ChoiceAction, Deputy, Question, Questions, Waiter as Note, WaiterKind, Who};
76use crate::config::Config;
77use crate::notices::{self, Notice};
78use crate::prompt;
79
80/// Node name a deputy runs under (`MAGI_NODE`); `magi ask --settle` accepts
81/// it and nothing else.
82pub const NODE: &str = "deputy";
83
84/// How often the runner looks at the store.
85pub const TICK: Duration = Duration::from_secs(5);
86
87/// Most turns ever started for one question. A deputy that keeps ending
88/// without an answer is a problem for the owner, not something to retry until
89/// the quota is gone.
90pub const MAX_STARTS: u32 = 3;
91
92/// Least gap between two starts for one question.
93const RESTART_AFTER: Duration = Duration::from_secs(30);
94
95/// Added to the question's remaining time for the invocation's wall clock, so
96/// the deputy's own `magi ask --wait` reaches the deadline first.
97const SLACK_SECS: u64 = 120;
98
99/// A claim file older than this was left by a process that died.
100const CLAIM_STALE: Duration = Duration::from_secs(60);
101
102/// How long the context-handover turn may take.
103const HANDOVER_TIMEOUT: Duration = Duration::from_secs(10 * 60);
104
105/// Stops the deputy when true: the daemon is parking.
106pub type Halt = Arc<dyn Fn() -> bool + Send + Sync>;
107
108/// What the conductor knew about one question, for its deputy.
109///
110/// `reason` is the conductor's own one-line reasoning (may be empty). Each
111/// choice is listed with what picking it does: an attached action, or - for
112/// the usual conductor question - that the answer is recorded on the task and
113/// the task is unblocked with it in its instructions.
114pub fn brief(
115 task_id: &str,
116 reason: &str,
117 choices: &[String],
118 actions: &BTreeMap<String, ChoiceAction>,
119) -> String {
120 let reason = reason.trim();
121 let mut s = format!(
122 "Task {task_id} (`magi task show {task_id}`) is blocked on this question. \
123 The conductor asked because: {}\n\n\
124 When the owner answers, magi records the question and the answer on the \
125 task and unblocks it, and the answer becomes part of the instructions of \
126 the task's next attempt.",
127 if reason.is_empty() {
128 "(it recorded no reasoning)"
129 } else {
130 reason
131 }
132 );
133 if !choices.is_empty() {
134 s.push_str("\n\nWhat each option does:");
135 for c in choices {
136 match actions.get(c) {
137 Some(a) => s.push_str(&format!("\n- `{c}`: also {}", a.describe())),
138 None => s.push_str(&format!(
139 "\n- `{c}`: recorded on the task as the answer and the task is unblocked"
140 )),
141 }
142 }
143 }
144 s
145}
146
147/// What a deputy serves: the question kinds it is attached to.
148#[derive(Debug, Clone, Copy, PartialEq, Eq)]
149pub enum Kind {
150 /// A conductor's question ([`crate::conduct::NODE`]).
151 Conduct,
152 /// The merge approval ([`crate::land::APPROVAL_NODE`]).
153 Land,
154 /// A question the release watcher filed ([`crate::release_watch`]).
155 Release,
156 /// A triage question about a held task or a stuck dependency root
157 /// ([`crate::triage::NODE`], [`crate::triage::DEPS_NODE`]).
158 Triage,
159 /// Any other question nobody asked from inside a run (no `cwd`) that offers
160 /// choices: the fallback, so a node nobody has met yet still has a listener.
161 Generic,
162}
163
164/// The kind of question a deputy serves for `q`, `None` for every other.
165///
166/// The fallback is decided by the absence of a `cwd`, never by the node name: a
167/// question filed with `magi ask` inside a run records its `cwd` and has an
168/// asker (and the waiter), and the node a seat asks from is the seat's own name
169/// (a reviewer's `review` is also the divergence question's node). A question
170/// with no choices cannot be settled and is a notice, not something to answer.
171pub fn kind_of(q: &Question) -> Option<Kind> {
172 match q.node.as_str() {
173 crate::conduct::NODE => Some(Kind::Conduct),
174 crate::land::APPROVAL_NODE => Some(Kind::Land),
175 // The seat matters: `bump` files choice-less notices on the same node,
176 // which nobody can answer and which must not cost a deputy.
177 crate::bump::NOTICE_NODE if q.seat == "release-watch" => Some(Kind::Release),
178 crate::bump::NOTICE_NODE => None,
179 crate::triage::NODE | crate::triage::DEPS_NODE => Some(Kind::Triage),
180 _ if q.cwd.is_none() && !q.choices.is_empty() => Some(Kind::Generic),
181 _ => None,
182 }
183}
184
185/// What a deputy is told about a question of no known kind: only what the
186/// question itself stored.
187pub fn generic_brief(
188 q: &Question,
189 actions: &BTreeMap<String, ChoiceAction>,
190 home: &Path,
191) -> String {
192 // `Question::run` may also hold a task id; only a run id can be shown.
193 let run = crate::run::is_run_id(&q.run).then_some(q.run.as_str());
194 let knows = if run.is_some() {
195 "You know what the question itself says and what you can learn from \
196 its run by the read-only investigation described below."
197 } else {
198 "You know only what the question itself says."
199 };
200 let mut s = format!(
201 "This question was filed by magi (node `{}`, seat `{}`) with no agent \
202 waiting on it. Magi records the owner's answer and the component that \
203 asked applies it; you apply nothing. {knows} A choice whose effect you \
204 cannot read is not yours to guess: do not settle it, ask the owner \
205 what they mean with `--thread` instead.",
206 q.node, q.seat
207 );
208 if let Some(run) = run {
209 let artifacts = home.join("runs").join(run).join("artifacts");
210 s.push_str(&format!(
211 "\n\nThis question concerns run `{run}`. You may investigate it on \
212 your own, read-only: run `magi show {run}` and read files under \
213 `{}`. When the owner asks about details (for example which part \
214 was wrong), look it up yourself and answer; do not reply that you \
215 cannot see the details, and do not tell the owner to run `magi show` \
216 themselves. Investigating never changes anything: edit, commit, push \
217 and apply nothing. If the run cannot be read, say exactly that and do \
218 not guess at the cause or at what an option does.",
219 artifacts.display()
220 ));
221 }
222 if q.node == crate::github_text::ASK_NODE {
223 s.push_str("\n\n");
224 s.push_str(&crate::github_text::deputy_brief());
225 }
226 if !q.choices.is_empty() {
227 s.push_str("\n\nWhat each option does:");
228 for c in &q.choices {
229 match actions.get(c) {
230 Some(a) => s.push_str(&format!("\n- `{c}`: also {}", a.describe())),
231 None => s.push_str(&format!(
232 "\n- `{c}`: recorded as the answer (nothing more is known)"
233 )),
234 }
235 }
236 }
237 s
238}
239
240/// Is picking `label` on `q` something that cannot be taken back, so that
241/// `--settle` must hold the owner's words to the same mechanical standard as a
242/// merge (`land::unhedged`)? Discarding a task deletes it; a divergence answer
243/// drops commits.
244pub fn destructive(q: &Question, label: &str) -> bool {
245 let at = q.choices.iter().position(|c| c == label);
246 // By label, not position: the retry question orders its choices differently.
247 if q.node == crate::github_text::ASK_NODE {
248 return matches!(
249 label,
250 crate::github_text::POST_ORIGINAL | crate::github_text::USE_REWRITE
251 );
252 }
253 match q.node.as_str() {
254 crate::triage::NODE => at == Some(2),
255 crate::triage::DEPS_NODE => at == Some(1),
256 n => n == crate::reconcile::NODE && q.seat == crate::reconcile::SEAT,
257 }
258}
259
260/// Does `--settle` hold `q` to the merge-approval rules (the verbatim quote of
261/// the latest message for `merge`, the whole message for `hold`)? A merge
262/// approval, and any other served question that offers `merge` (the release
263/// watcher's local-mode approval merges just as irreversibly).
264pub fn merge_gated(q: &Question) -> bool {
265 q.node == crate::land::APPROVAL_NODE
266 || (kind_of(q).is_some_and(|k| k != Kind::Conduct)
267 && q.choices.iter().any(|c| c == crate::land::APPROVE))
268}
269
270/// Does `q`'s clock run from `asked_at` and never move on a reply? True for the
271/// questions that something other than the waiter retires: a merge approval
272/// (land) and a release-watch question (the watcher, by silence being a hold).
273pub fn fixed_clock(q: &Question) -> bool {
274 matches!(kind_of(q), Some(Kind::Land | Kind::Release)) || q.node == crate::github_text::ASK_NODE
275}
276
277/// Second after which nobody is to be started or kept on `q`.
278///
279/// A conductor question runs from its last activity (`magi ask --thread`
280/// re-arms it, and the waiter retires it). A merge approval never moves: it
281/// runs from `asked_at`, exactly where `daemon::land_resume_state` abandons it,
282/// so a conversation cannot stretch the hold and land stays the only place that
283/// retires one.
284pub fn deadline(q: &Question, default_timeout: u64) -> i64 {
285 let secs = if q.answer_timeout > 0 {
286 q.answer_timeout
287 } else {
288 default_timeout
289 };
290 let from = if fixed_clock(q) {
291 q.asked_at.as_second()
292 } else {
293 q.last_activity()
294 };
295 from.saturating_add(secs as i64)
296}
297
298/// Dispatch and text admission share the fixed-clock and unread-word rules.
299pub fn restart_in_time(q: &Question, default_timeout: u64, now: Timestamp) -> bool {
300 now.as_second() <= deadline(q, default_timeout) || q.unread_from_owner().is_some()
301}
302
303/// Can `magi serve` start a deputy at all under `cfg`? Not when deputies are
304/// switched off (`daemon.max_deputies = 0`) or the config could not be read.
305///
306/// Also false when the agent the deputy would run as cannot be resolved
307/// (`agent` is the deputy's recorded agent, empty when it has none): `turn`
308/// fails on that before it counts a start, so it would never reach
309/// [`MAX_STARTS`].
310pub fn can_start(cfg: Option<&Config>, agent: &str) -> bool {
311 cfg.is_some_and(|c| {
312 c.daemon.max_deputies > 0
313 && ((!agent.is_empty() && c.agent(agent).is_ok()) || c.resolve_roles().is_ok())
314 })
315}
316
317/// The agent a question's deputy records, empty when it has none yet.
318pub fn agent_of(q: &Question) -> &str {
319 q.deputy.as_ref().map_or("", |d| d.agent.as_str())
320}
321
322/// Has this question's deputy run out of starts, or can none ever start, with
323/// the deadline gone?
324///
325/// Then nothing will ever read an unread say, and the waiter must retire the
326/// question anyway instead of deferring to a deputy that no longer starts.
327/// `startable` is [`can_start`]; a fresh lease is the caller's to check.
328pub fn exhausted_past_deadline(
329 q: &Question,
330 startable: bool,
331 default_timeout: u64,
332 now: Timestamp,
333) -> bool {
334 q.status.open()
335 && q.deputy
336 .as_ref()
337 .is_some_and(|d| d.starts >= MAX_STARTS || !startable)
338 && now.as_second() > deadline(q, default_timeout)
339}
340
341/// The deputy runner: its own task inside `magi serve`, beside the waiter.
342pub struct Deputies {
343 store: Questions,
344 home: PathBuf,
345 cfg: Option<Config>,
346 /// Working directory for a question that recorded none.
347 fallback_repo: PathBuf,
348 max: usize,
349 halt: Halt,
350 tasks: JoinSet<String>,
351 inflight: HashSet<String>,
352 /// Earliest next start per question.
353 memo: HashMap<String, Instant>,
354}
355
356impl Deputies {
357 /// A runner over `store`, with at most `max` deputies at once.
358 pub fn new(
359 store: Questions,
360 home: PathBuf,
361 cfg: Option<Config>,
362 fallback_repo: PathBuf,
363 max: usize,
364 halt: Halt,
365 ) -> Self {
366 Self {
367 store,
368 home,
369 cfg,
370 fallback_repo,
371 max,
372 halt,
373 tasks: JoinSet::new(),
374 inflight: HashSet::new(),
375 memo: HashMap::new(),
376 }
377 }
378
379 fn default_timeout(&self) -> u64 {
380 self.cfg
381 .as_ref()
382 .map_or(Config::default().graph.answer_timeout, |c| {
383 c.graph.answer_timeout
384 })
385 }
386
387 fn reap(&mut self) {
388 while let Some(done) = self.tasks.try_join_next() {
389 if let Ok(id) = done {
390 self.inflight.remove(&id);
391 }
392 }
393 if self.tasks.is_empty() {
394 self.inflight.clear();
395 }
396 }
397
398 /// Give a question the record a deputy needs: the deputy itself (with a
399 /// brief rebuilt from what the question stored, when it was filed before
400 /// deputies existed - a lost reason is not invented) and the deadline.
401 ///
402 /// A conductor question also gets the working directory the waiter and
403 /// `magi ask --wait` use. Every other kind never does: `cwd` is what makes a
404 /// question the waiter's (it would resume and expire it beside the deputy),
405 /// and a release-watch question, a merge approval, a triage question and a
406 /// fallback one have no asker to resume.
407 fn attach(&self, q: &Question, kind: Kind) -> Option<Question> {
408 let default_timeout = self.default_timeout();
409 let repo = self.fallback_repo.to_string_lossy().into_owned();
410 let state = match kind {
411 Kind::Land => crate::run::RunState::load(&q.run).ok(),
412 Kind::Conduct | Kind::Release | Kind::Triage | Kind::Generic => None,
413 };
414 // The deadline land itself enforces for this run.
415 let timeout = state
416 .as_ref()
417 .map_or(default_timeout, |s| s.config.graph.answer_timeout);
418 self.store
419 .update(&q.id, |r| {
420 if r.deputy.is_none() {
421 r.deputy = Some(Deputy::new(match kind {
422 Kind::Conduct => brief(&r.run, &r.detail, &r.choices, &r.actions),
423 Kind::Land => crate::land::deputy_brief(r, state.as_ref()),
424 Kind::Release => crate::release_watch::deputy_brief(r, &self.home),
425 Kind::Triage => crate::triage::deputy_brief(
426 r,
427 &crate::queue::Queue::at(self.home.join("queue")),
428 ),
429 Kind::Generic => generic_brief(r, &r.actions, &self.home),
430 }));
431 }
432 if kind == Kind::Conduct && r.cwd.is_none() {
433 r.cwd = Some(repo.clone());
434 }
435 if r.answer_timeout == 0 {
436 r.answer_timeout = match kind {
437 Kind::Conduct => default_timeout,
438 Kind::Land => timeout,
439 Kind::Release | Kind::Triage | Kind::Generic => default_timeout,
440 };
441 }
442 Ok(())
443 })
444 .map(|(r, ())| r)
445 .map_err(|e| tracing::warn!("question {}: cannot attach a deputy: {e:#}", q.short()))
446 .ok()
447 }
448
449 /// Look at every open conductor question once and start a deputy on each
450 /// that has none, within the limits. Returns without waiting for them.
451 pub fn tick(&mut self, now: Timestamp) {
452 self.reap();
453 for q in self.store.list() {
454 if (self.halt)() {
455 return;
456 }
457 let Some(kind) = kind_of(&q) else {
458 continue;
459 };
460 if !q.status.open() {
461 continue;
462 }
463 let needs_cwd = kind == Kind::Conduct && q.cwd.is_none();
464 let q = if q.deputy.is_none() || needs_cwd || q.answer_timeout == 0 {
465 match self.attach(&q, kind) {
466 Some(q) => q,
467 None => continue,
468 }
469 } else {
470 q
471 };
472 let Some(dep) = q.deputy.as_ref() else {
473 continue;
474 };
475 if self.inflight.contains(&q.id)
476 || self.store.read_lease(&q.id).is_some_and(|l| l.fresh(now))
477 {
478 continue;
479 }
480 if dep.starts >= MAX_STARTS {
481 continue;
482 }
483 // Past the deadline the waiter retires the question; a say that
484 // arrived in time is still read first.
485 if !restart_in_time(&q, self.default_timeout(), now) {
486 continue;
487 }
488 if self.inflight.len() >= self.max || !can_start(self.cfg.as_ref(), dep.agent.as_str())
489 {
490 continue;
491 }
492 if matches!(self.memo.get(&q.id), Some(until) if Instant::now() < *until) {
493 continue;
494 }
495 self.memo
496 .insert(q.id.clone(), Instant::now() + RESTART_AFTER);
497 self.inflight.insert(q.id.clone());
498 let job = Job {
499 store: self.store.clone(),
500 home: self.home.clone(),
501 cfg: self.cfg.clone(),
502 fallback_repo: self.fallback_repo.clone(),
503 halt: Arc::clone(&self.halt),
504 };
505 let id = q.id.clone();
506 self.tasks.spawn(async move {
507 if let Err(e) = job.turn(&id).await {
508 tracing::warn!("deputy for question {}: {e:#}", crate::ask::short_id(&id));
509 }
510 id
511 });
512 }
513 }
514
515 /// Wait for every deputy turn in flight. The daemon never calls this; the
516 /// tests do, to look at the record once a turn is over.
517 pub async fn drain(&mut self) {
518 while let Some(done) = self.tasks.join_next().await {
519 if let Ok(id) = done {
520 self.inflight.remove(&id);
521 }
522 }
523 self.inflight.clear();
524 }
525}
526
527/// Everything one deputy turn needs, owned so it can run detached.
528struct Job {
529 store: Questions,
530 home: PathBuf,
531 cfg: Option<Config>,
532 fallback_repo: PathBuf,
533 halt: Halt,
534}
535
536impl Job {
537 /// Run one invocation, beating the lease; `None` when the daemon is
538 /// parking and the turn was dropped.
539 async fn drive(
540 &self,
541 spec: &crate::config::AgentSpec,
542 seat: &mut SeatState,
543 inv: &Invocation<'_>,
544 id: &str,
545 ) -> Option<Result<agent::AgentOutput>> {
546 let fut = agent::invoke(spec, seat, inv);
547 tokio::pin!(fut);
548 let mut beat = tokio::time::interval(Duration::from_secs(1));
549 let mut beats = 0u32;
550 loop {
551 tokio::select! {
552 r = &mut fut => break Some(r),
553 _ = beat.tick() => {
554 if (self.halt)() {
555 break None;
556 }
557 beats += 1;
558 if beats % 20 == 0 {
559 self.store.beat(id, WaiterKind::Deputy);
560 }
561 }
562 }
563 }
564 }
565
566 /// The question settled or expired during the handover: release it without
567 /// starting the long turn. The start stays counted - the handover ran.
568 fn park_quietly(&self, id: &str) {
569 let _ = self.store.update(id, |r| {
570 r.waiter = None;
571 Ok(())
572 });
573 self.store.drop_lease(id);
574 }
575
576 /// Parking: the turn is dropped and the start refunded. The seat stays as
577 /// last persisted - after the handover turn, resumable.
578 fn park(&self, id: &str) {
579 let _ = self.store.update(id, |r| {
580 if let Some(d) = r.deputy.as_mut() {
581 d.starts = d.starts.saturating_sub(1);
582 }
583 r.waiter = None;
584 Ok(())
585 });
586 self.store.drop_lease(id);
587 }
588
589 async fn turn(&self, id: &str) -> Result<()> {
590 let claim = self.store.root().join(format!("{id}.deputy-claim"));
591 // The claim only covers the decision to start and the write that records
592 // it, so a daemon that dies holding it blocks a restart for a minute,
593 // not for a turn's length. The lease guards the turn itself.
594 if !crate::waiter::take_claim(&claim, CLAIM_STALE) {
595 return Ok(());
596 }
597 let release = crate::waiter::Release(claim);
598
599 // Decided again under the claim, on the record as it is now.
600 let q = self.store.get(id)?;
601 let now = Timestamp::now();
602 let Some(dep) = q.deputy.clone() else {
603 return Ok(());
604 };
605 if !q.status.open()
606 || dep.starts >= MAX_STARTS
607 || self.store.read_lease(&q.id).is_some_and(|l| l.fresh(now))
608 {
609 return Ok(());
610 }
611 let cfg = self
612 .cfg
613 .as_ref()
614 .context("the deputy's configuration is not available")?;
615 let spec = match cfg.agent(&dep.agent) {
616 Ok(s) if !dep.agent.is_empty() => s.clone(),
617 _ => {
618 cfg.resolve_roles()
619 .context("resolving the deputy's agent")?
620 .conductor
621 }
622 };
623 // The same CLI conversation when it can be resumed; otherwise a fresh
624 // seat - and either way the prompt carries the whole context.
625 let key = crate::ask::deputy_seat_key(&q.id);
626 let (mut seat, resumed) = match dep.seat.clone() {
627 Some(s)
628 if s.agent == spec.id && agent::has_session(spec.kind, &s, cfg.graph.sessions) =>
629 {
630 (s, true)
631 }
632 _ => (SeatState::new(&key, &spec.id, crate::rng::entropy()), false),
633 };
634 // A release-watch question has no `cwd`; its watch record names the
635 // checkout the pull request belongs to.
636 let recorded = q.cwd.clone().or_else(|| match kind_of(&q) {
637 Some(Kind::Release) => {
638 crate::release_watch::state_for_question(&self.home, &q.id).map(|st| st.repo)
639 }
640 // These record a task id in `run` (or a run id for a divergence's
641 // sibling kinds): the task's repository, when it can be found.
642 Some(Kind::Triage | Kind::Generic) => crate::queue::Queue::at(self.home.join("queue"))
643 .get(&q.run)
644 .ok()
645 .map(|t| t.repo.to_string_lossy().into_owned())
646 .filter(|r| !r.is_empty()),
647 _ => None,
648 });
649 let cwd = recorded
650 .as_deref()
651 .map(PathBuf::from)
652 .filter(|p| p.is_dir())
653 .unwrap_or_else(|| self.fallback_repo.clone());
654
655 // A settle's note is part of what the seat said, so a resumed seat
656 // reads its own report back.
657 let bodies: Vec<String> = q
658 .thread
659 .iter()
660 .map(|t| match &t.note {
661 Some(n) => format!("{}\n(note: {n})", t.body),
662 None => t.body.clone(),
663 })
664 .collect();
665 let thread: Vec<(&str, &str)> = q
666 .thread
667 .iter()
668 .zip(&bodies)
669 .map(|(t, body)| {
670 (
671 if t.who == Who::Operator {
672 "operator"
673 } else {
674 "agent"
675 },
676 body.as_str(),
677 )
678 })
679 .collect();
680 let read = &thread[..q.delivered_turns.min(thread.len())];
681 let unread = q.unread_from_owner();
682 let snapshot = q.thread.len();
683 let current_brief = format!(
684 "{}\n\n{}",
685 dep.brief,
686 crate::queue::dependency_facts(
687 &crate::queue::Queue::at(self.home.join("queue")),
688 &self.store
689 )
690 );
691 let body = prompt::deputy(&prompt::DeputyPrompt {
692 id: &q.id,
693 summary: &q.summary,
694 detail: &q.detail,
695 brief: ¤t_brief,
696 choices: &q.choices,
697 thread: read,
698 unread: unread.as_deref(),
699 resumed,
700 handover: false,
701 kind: kind_of(&q).unwrap_or(Kind::Conduct),
702 language: &cfg.graph.language,
703 });
704
705 // Recorded before the turn starts, so a daemon that dies mid-turn
706 // leaves a start counted and the seat on the record.
707 self.store.beat(&q.id, WaiterKind::Deputy);
708 let starts = dep.starts + 1;
709 let first = seat.clone();
710 self.store.update(&q.id, |r| {
711 if let Some(d) = r.deputy.as_mut() {
712 d.agent = spec.id.clone();
713 d.seat = Some(first);
714 d.starts = starts;
715 }
716 r.waiter = Some(Note {
717 kind: WaiterKind::Deputy,
718 since: now,
719 });
720 Ok(())
721 })?;
722 drop(release);
723 tracing::info!(
724 "question {}: deputy seat {} {} (start {starts}/{MAX_STARTS})",
725 q.short(),
726 seat.key,
727 if resumed { "resuming" } else { "starting" }
728 );
729
730 let left = (deadline(&q, cfg.graph.answer_timeout) - now.as_second()).max(0) as u64;
731 let artifacts = self.store.root().join(format!("{}.deputy", q.id));
732 let cache_dir = cfg.cache_dir();
733 // `magi ask` writes the question record and its lock, which a read-only
734 // sandbox refuses, so the seat cannot be read-only. It is told never to
735 // edit anything; the daemon, not the deputy, applies outcomes.
736 let allow_write = true;
737 // The question store is outside the repository, and `magi ask` writes it.
738 // A merge approval's deputy also files follow-up tasks with `magi task
739 // add`, which writes the queue; that is the one other place it writes.
740 let writable = [self.store.root().to_path_buf(), self.home.join("queue")];
741 macro_rules! invocation {
742 ($prompt:expr, $stem:expr, $timeout:expr) => {
743 Invocation {
744 cwd: &cwd,
745 prompt: $prompt,
746 timeout: $timeout,
747 allow_write,
748 unsandboxed: false,
749 sessions: cfg.graph.sessions,
750 artifacts: &artifacts,
751 stem: $stem,
752 // The question's own run key (the task id) is what lets this
753 // seat's `magi ask` pass the ownership check on a conductor
754 // question.
755 run: &q.run,
756 node: NODE,
757 cache_dir: cache_dir.as_deref(),
758 attachments: &[],
759 writable: &writable,
760 }
761 };
762 }
763
764 // A fresh seat first takes a short turn that ends: `agent::invoke` only
765 // learns the CLI's session id when it returns, and the real turn blocks
766 // in `magi ask` for hours, so a daemon stopped mid-wait would otherwise
767 // leave nothing to resume. This turn persists the seat before the wait.
768 let mut early = None;
769 if !resumed && cfg.graph.sessions {
770 let hbody = prompt::deputy(&prompt::DeputyPrompt {
771 id: &q.id,
772 summary: &q.summary,
773 detail: &q.detail,
774 brief: ¤t_brief,
775 choices: &q.choices,
776 thread: read,
777 unread: None,
778 resumed: false,
779 handover: true,
780 kind: kind_of(&q).unwrap_or(Kind::Conduct),
781 language: &cfg.graph.language,
782 });
783 let hstem = format!("handover-{starts}");
784 // Bounded by the question's own deadline, not only by its own cap: the
785 // lease it beats would otherwise keep an expired question alive.
786 let hlimit = HANDOVER_TIMEOUT.min(Duration::from_secs(left.max(1)));
787 let hinv = invocation!(&hbody, &hstem, hlimit);
788 let Some(done) = self.drive(&spec, &mut seat, &hinv, &q.id).await else {
789 self.park(&q.id);
790 return Ok(());
791 };
792 let kept = seat.clone();
793 self.store.update(&q.id, |r| {
794 if let Some(d) = r.deputy.as_mut() {
795 d.seat = Some(kept);
796 }
797 Ok(())
798 })?;
799 if !matches!(&done, Ok(o) if o.usable()) {
800 early = Some(done);
801 }
802 }
803
804 // The handover may have been slow: look at the question again, and
805 // measure the long turn from what is left now.
806 let left = if early.is_none() && !resumed && cfg.graph.sessions {
807 let now = Timestamp::now();
808 let again = self.store.get(&q.id)?;
809 let left = (deadline(&again, cfg.graph.answer_timeout) - now.as_second()).max(0) as u64;
810 if !again.status.open() || (left == 0 && again.unread_from_owner().is_none()) {
811 self.park_quietly(&q.id);
812 return Ok(());
813 }
814 left
815 } else {
816 left
817 };
818 let out = match early {
819 Some(done) => done,
820 None => {
821 let stem = format!("turn-{starts}");
822 let timeout = Duration::from_secs(left.max(60) + SLACK_SECS);
823 let inv = invocation!(&body, &stem, timeout);
824 match self.drive(&spec, &mut seat, &inv, &q.id).await {
825 Some(out) => out,
826 None => {
827 self.park(&q.id);
828 return Ok(());
829 }
830 }
831 }
832 };
833
834 let (text, why) = match out {
835 Ok(o) if o.usable() => (Some(o.text.trim().to_owned()), None),
836 Ok(o) if o.timed_out => (None, Some("its turn timed out".to_owned())),
837 Ok(o) if o.quota_exhausted() => (None, Some("the agent is out of quota".to_owned())),
838 Ok(o) => (
839 None,
840 Some(format!("its turn failed (exit {:?})", o.exit_code)),
841 ),
842 Err(e) => (None, Some(format!("the agent could not be started: {e:#}"))),
843 };
844 let kept = seat.clone();
845 self.store.update(&q.id, |r| {
846 if let Some(d) = r.deputy.as_mut() {
847 d.seat = Some(kept);
848 }
849 r.waiter = None;
850 // Retire ownership under the same lock used by text admission.
851 self.store.drop_lease(&q.id);
852 // The owner's words were in the prompt. An agent that read them but
853 // answered in prose instead of `magi ask --thread` still spoke;
854 // keep it on the record so the owner reads it.
855 if let Some(text) = text
856 && unread.is_some()
857 && !text.is_empty()
858 && r.status.open()
859 && r.thread.len() == snapshot
860 {
861 r.delivered_turns = r.delivered_turns.max(snapshot);
862 let choices = r.choices.clone();
863 r.reply(text, choices)?;
864 }
865 Ok(())
866 })?;
867 if let Some(why) = why {
868 tracing::warn!("question {}: the deputy ended: {why}", q.short());
869 notices::raise_in(
870 &self.home,
871 Notice::warn(
872 &format!("deputy-turn:{}", q.id),
873 format!(
874 "Question {} \"{}\": its follow-up agent stopped ({why}).",
875 q.short(),
876 q.summary
877 ),
878 ),
879 );
880 }
881 Ok(())
882 }
883}
884
885/// The runner's loop, until `stop` is asked for.
886pub async fn run(mut deputies: Deputies, stop: crate::daemon::Stop) {
887 while !stop.stopped() {
888 deputies.tick(Timestamp::now());
889 tokio::time::sleep(TICK).await;
890 }
891}