Skip to main content

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 label == crate::github_text::POST_ORIGINAL;
249    }
250    match q.node.as_str() {
251        crate::triage::NODE => at == Some(2),
252        crate::triage::DEPS_NODE => at == Some(1),
253        n => n == crate::reconcile::NODE && q.seat == crate::reconcile::SEAT,
254    }
255}
256
257/// Does `--settle` hold `q` to the merge-approval rules (the verbatim quote of
258/// the latest message for `merge`, the whole message for `hold`)? A merge
259/// approval, and any other served question that offers `merge` (the release
260/// watcher's local-mode approval merges just as irreversibly).
261pub fn merge_gated(q: &Question) -> bool {
262    q.node == crate::land::APPROVAL_NODE
263        || (kind_of(q).is_some_and(|k| k != Kind::Conduct)
264            && q.choices.iter().any(|c| c == crate::land::APPROVE))
265}
266
267/// Does `q`'s clock run from `asked_at` and never move on a reply? True for the
268/// questions that something other than the waiter retires: a merge approval
269/// (land) and a release-watch question (the watcher, by silence being a hold).
270pub fn fixed_clock(q: &Question) -> bool {
271    matches!(kind_of(q), Some(Kind::Land | Kind::Release)) || q.node == crate::github_text::ASK_NODE
272}
273
274/// Second after which nobody is to be started or kept on `q`.
275///
276/// A conductor question runs from its last activity (`magi ask --thread`
277/// re-arms it, and the waiter retires it). A merge approval never moves: it
278/// runs from `asked_at`, exactly where `daemon::land_resume_state` abandons it,
279/// so a conversation cannot stretch the hold and land stays the only place that
280/// retires one.
281pub fn deadline(q: &Question, default_timeout: u64) -> i64 {
282    let secs = if q.answer_timeout > 0 {
283        q.answer_timeout
284    } else {
285        default_timeout
286    };
287    let from = if fixed_clock(q) {
288        q.asked_at.as_second()
289    } else {
290        q.last_activity()
291    };
292    from.saturating_add(secs as i64)
293}
294
295/// Can `magi serve` start a deputy at all under `cfg`? Not when deputies are
296/// switched off (`daemon.max_deputies = 0`) or the config could not be read.
297///
298/// Also false when the agent the deputy would run as cannot be resolved
299/// (`agent` is the deputy's recorded agent, empty when it has none): `turn`
300/// fails on that before it counts a start, so it would never reach
301/// [`MAX_STARTS`].
302pub fn can_start(cfg: Option<&Config>, agent: &str) -> bool {
303    cfg.is_some_and(|c| {
304        c.daemon.max_deputies > 0
305            && ((!agent.is_empty() && c.agent(agent).is_ok()) || c.resolve_roles().is_ok())
306    })
307}
308
309/// The agent a question's deputy records, empty when it has none yet.
310pub fn agent_of(q: &Question) -> &str {
311    q.deputy.as_ref().map_or("", |d| d.agent.as_str())
312}
313
314/// Has this question's deputy run out of starts, or can none ever start, with
315/// the deadline gone?
316///
317/// Then nothing will ever read an unread say, and the waiter must retire the
318/// question anyway instead of deferring to a deputy that no longer starts.
319/// `startable` is [`can_start`]; a fresh lease is the caller's to check.
320pub fn exhausted_past_deadline(
321    q: &Question,
322    startable: bool,
323    default_timeout: u64,
324    now: Timestamp,
325) -> bool {
326    q.status.open()
327        && q.deputy
328            .as_ref()
329            .is_some_and(|d| d.starts >= MAX_STARTS || !startable)
330        && now.as_second() > deadline(q, default_timeout)
331}
332
333/// The deputy runner: its own task inside `magi serve`, beside the waiter.
334pub struct Deputies {
335    store: Questions,
336    home: PathBuf,
337    cfg: Option<Config>,
338    /// Working directory for a question that recorded none.
339    fallback_repo: PathBuf,
340    max: usize,
341    halt: Halt,
342    tasks: JoinSet<String>,
343    inflight: HashSet<String>,
344    /// Earliest next start per question.
345    memo: HashMap<String, Instant>,
346}
347
348impl Deputies {
349    /// A runner over `store`, with at most `max` deputies at once.
350    pub fn new(
351        store: Questions,
352        home: PathBuf,
353        cfg: Option<Config>,
354        fallback_repo: PathBuf,
355        max: usize,
356        halt: Halt,
357    ) -> Self {
358        Self {
359            store,
360            home,
361            cfg,
362            fallback_repo,
363            max,
364            halt,
365            tasks: JoinSet::new(),
366            inflight: HashSet::new(),
367            memo: HashMap::new(),
368        }
369    }
370
371    fn default_timeout(&self) -> u64 {
372        self.cfg
373            .as_ref()
374            .map_or(Config::default().graph.answer_timeout, |c| {
375                c.graph.answer_timeout
376            })
377    }
378
379    fn reap(&mut self) {
380        while let Some(done) = self.tasks.try_join_next() {
381            if let Ok(id) = done {
382                self.inflight.remove(&id);
383            }
384        }
385        if self.tasks.is_empty() {
386            self.inflight.clear();
387        }
388    }
389
390    /// Give a question the record a deputy needs: the deputy itself (with a
391    /// brief rebuilt from what the question stored, when it was filed before
392    /// deputies existed - a lost reason is not invented) and the deadline.
393    ///
394    /// A conductor question also gets the working directory the waiter and
395    /// `magi ask --wait` use. Every other kind never does: `cwd` is what makes a
396    /// question the waiter's (it would resume and expire it beside the deputy),
397    /// and a release-watch question, a merge approval, a triage question and a
398    /// fallback one have no asker to resume.
399    fn attach(&self, q: &Question, kind: Kind) -> Option<Question> {
400        let default_timeout = self.default_timeout();
401        let repo = self.fallback_repo.to_string_lossy().into_owned();
402        let state = match kind {
403            Kind::Land => crate::run::RunState::load(&q.run).ok(),
404            Kind::Conduct | Kind::Release | Kind::Triage | Kind::Generic => None,
405        };
406        // The deadline land itself enforces for this run.
407        let timeout = state
408            .as_ref()
409            .map_or(default_timeout, |s| s.config.graph.answer_timeout);
410        self.store
411            .update(&q.id, |r| {
412                if r.deputy.is_none() {
413                    r.deputy = Some(Deputy::new(match kind {
414                        Kind::Conduct => brief(&r.run, &r.detail, &r.choices, &r.actions),
415                        Kind::Land => crate::land::deputy_brief(r, state.as_ref()),
416                        Kind::Release => crate::release_watch::deputy_brief(r, &self.home),
417                        Kind::Triage => crate::triage::deputy_brief(
418                            r,
419                            &crate::queue::Queue::at(self.home.join("queue")),
420                        ),
421                        Kind::Generic => generic_brief(r, &r.actions, &self.home),
422                    }));
423                }
424                if kind == Kind::Conduct && r.cwd.is_none() {
425                    r.cwd = Some(repo.clone());
426                }
427                if r.answer_timeout == 0 {
428                    r.answer_timeout = match kind {
429                        Kind::Conduct => default_timeout,
430                        Kind::Land => timeout,
431                        Kind::Release | Kind::Triage | Kind::Generic => default_timeout,
432                    };
433                }
434                Ok(())
435            })
436            .map(|(r, ())| r)
437            .map_err(|e| tracing::warn!("question {}: cannot attach a deputy: {e:#}", q.short()))
438            .ok()
439    }
440
441    /// Look at every open conductor question once and start a deputy on each
442    /// that has none, within the limits. Returns without waiting for them.
443    pub fn tick(&mut self, now: Timestamp) {
444        self.reap();
445        for q in self.store.list() {
446            if (self.halt)() {
447                return;
448            }
449            let Some(kind) = kind_of(&q) else {
450                continue;
451            };
452            if !q.status.open() {
453                continue;
454            }
455            let needs_cwd = kind == Kind::Conduct && q.cwd.is_none();
456            let q = if q.deputy.is_none() || needs_cwd || q.answer_timeout == 0 {
457                match self.attach(&q, kind) {
458                    Some(q) => q,
459                    None => continue,
460                }
461            } else {
462                q
463            };
464            let Some(dep) = q.deputy.as_ref() else {
465                continue;
466            };
467            if self.inflight.contains(&q.id)
468                || self.store.read_lease(&q.id).is_some_and(|l| l.fresh(now))
469            {
470                continue;
471            }
472            if dep.starts >= MAX_STARTS {
473                self.give_up(&q);
474                continue;
475            }
476            // Past the deadline the waiter retires the question; a say that
477            // arrived in time is still read first.
478            if now.as_second() > deadline(&q, self.default_timeout())
479                && q.unread_from_owner().is_none()
480            {
481                continue;
482            }
483            if self.inflight.len() >= self.max || !can_start(self.cfg.as_ref(), dep.agent.as_str())
484            {
485                continue;
486            }
487            if matches!(self.memo.get(&q.id), Some(until) if Instant::now() < *until) {
488                continue;
489            }
490            self.memo
491                .insert(q.id.clone(), Instant::now() + RESTART_AFTER);
492            self.inflight.insert(q.id.clone());
493            let job = Job {
494                store: self.store.clone(),
495                home: self.home.clone(),
496                cfg: self.cfg.clone(),
497                fallback_repo: self.fallback_repo.clone(),
498                halt: Arc::clone(&self.halt),
499            };
500            let id = q.id.clone();
501            self.tasks.spawn(async move {
502                if let Err(e) = job.turn(&id).await {
503                    tracing::warn!("deputy for question {}: {e:#}", crate::ask::short_id(&id));
504                }
505                id
506            });
507        }
508    }
509
510    /// Wait for every deputy turn in flight. The daemon never calls this; the
511    /// tests do, to look at the record once a turn is over.
512    pub async fn drain(&mut self) {
513        while let Some(done) = self.tasks.join_next().await {
514            if let Ok(id) = done {
515                self.inflight.remove(&id);
516            }
517        }
518        self.inflight.clear();
519    }
520
521    /// Say once that nobody is listening any more.
522    fn give_up(&self, q: &Question) {
523        notices::raise_in(
524            &self.home,
525            Notice::warn(
526                &format!("deputy:{}", q.id),
527                format!(
528                    "Question {} \"{}\": its follow-up agent ended {MAX_STARTS} times \
529                     without an answer and is not restarted. What you say is recorded \
530                     but nothing will read it; answer with one of the choices instead.",
531                    q.short(),
532                    q.summary
533                ),
534            ),
535        );
536    }
537}
538
539/// Everything one deputy turn needs, owned so it can run detached.
540struct Job {
541    store: Questions,
542    home: PathBuf,
543    cfg: Option<Config>,
544    fallback_repo: PathBuf,
545    halt: Halt,
546}
547
548impl Job {
549    /// Run one invocation, beating the lease; `None` when the daemon is
550    /// parking and the turn was dropped.
551    async fn drive(
552        &self,
553        spec: &crate::config::AgentSpec,
554        seat: &mut SeatState,
555        inv: &Invocation<'_>,
556        id: &str,
557    ) -> Option<Result<agent::AgentOutput>> {
558        let fut = agent::invoke(spec, seat, inv);
559        tokio::pin!(fut);
560        let mut beat = tokio::time::interval(Duration::from_secs(1));
561        let mut beats = 0u32;
562        loop {
563            tokio::select! {
564                r = &mut fut => break Some(r),
565                _ = beat.tick() => {
566                    if (self.halt)() {
567                        break None;
568                    }
569                    beats += 1;
570                    if beats % 20 == 0 {
571                        self.store.beat(id, WaiterKind::Deputy);
572                    }
573                }
574            }
575        }
576    }
577
578    /// The question settled or expired during the handover: release it without
579    /// starting the long turn. The start stays counted - the handover ran.
580    fn park_quietly(&self, id: &str) {
581        let _ = self.store.update(id, |r| {
582            r.waiter = None;
583            Ok(())
584        });
585        self.store.drop_lease(id);
586    }
587
588    /// Parking: the turn is dropped and the start refunded. The seat stays as
589    /// last persisted - after the handover turn, resumable.
590    fn park(&self, id: &str) {
591        let _ = self.store.update(id, |r| {
592            if let Some(d) = r.deputy.as_mut() {
593                d.starts = d.starts.saturating_sub(1);
594            }
595            r.waiter = None;
596            Ok(())
597        });
598        self.store.drop_lease(id);
599    }
600
601    async fn turn(&self, id: &str) -> Result<()> {
602        let claim = self.store.root().join(format!("{id}.deputy-claim"));
603        // The claim only covers the decision to start and the write that records
604        // it, so a daemon that dies holding it blocks a restart for a minute,
605        // not for a turn's length. The lease guards the turn itself.
606        if !crate::waiter::take_claim(&claim, CLAIM_STALE) {
607            return Ok(());
608        }
609        let release = crate::waiter::Release(claim);
610
611        // Decided again under the claim, on the record as it is now.
612        let q = self.store.get(id)?;
613        let now = Timestamp::now();
614        let Some(dep) = q.deputy.clone() else {
615            return Ok(());
616        };
617        if !q.status.open()
618            || dep.starts >= MAX_STARTS
619            || self.store.read_lease(&q.id).is_some_and(|l| l.fresh(now))
620        {
621            return Ok(());
622        }
623        let cfg = self
624            .cfg
625            .as_ref()
626            .context("the deputy's configuration is not available")?;
627        let spec = match cfg.agent(&dep.agent) {
628            Ok(s) if !dep.agent.is_empty() => s.clone(),
629            _ => {
630                cfg.resolve_roles()
631                    .context("resolving the deputy's agent")?
632                    .conductor
633            }
634        };
635        // The same CLI conversation when it can be resumed; otherwise a fresh
636        // seat - and either way the prompt carries the whole context.
637        let key = crate::ask::deputy_seat_key(&q.id);
638        let (mut seat, resumed) = match dep.seat.clone() {
639            Some(s)
640                if s.agent == spec.id && agent::has_session(spec.kind, &s, cfg.graph.sessions) =>
641            {
642                (s, true)
643            }
644            _ => (SeatState::new(&key, &spec.id, crate::rng::entropy()), false),
645        };
646        // A release-watch question has no `cwd`; its watch record names the
647        // checkout the pull request belongs to.
648        let recorded = q.cwd.clone().or_else(|| match kind_of(&q) {
649            Some(Kind::Release) => {
650                crate::release_watch::state_for_question(&self.home, &q.id).map(|st| st.repo)
651            }
652            // These record a task id in `run` (or a run id for a divergence's
653            // sibling kinds): the task's repository, when it can be found.
654            Some(Kind::Triage | Kind::Generic) => crate::queue::Queue::at(self.home.join("queue"))
655                .get(&q.run)
656                .ok()
657                .map(|t| t.repo.to_string_lossy().into_owned())
658                .filter(|r| !r.is_empty()),
659            _ => None,
660        });
661        let cwd = recorded
662            .as_deref()
663            .map(PathBuf::from)
664            .filter(|p| p.is_dir())
665            .unwrap_or_else(|| self.fallback_repo.clone());
666
667        // A settle's note is part of what the seat said, so a resumed seat
668        // reads its own report back.
669        let bodies: Vec<String> = q
670            .thread
671            .iter()
672            .map(|t| match &t.note {
673                Some(n) => format!("{}\n(note: {n})", t.body),
674                None => t.body.clone(),
675            })
676            .collect();
677        let thread: Vec<(&str, &str)> = q
678            .thread
679            .iter()
680            .zip(&bodies)
681            .map(|(t, body)| {
682                (
683                    if t.who == Who::Operator {
684                        "operator"
685                    } else {
686                        "agent"
687                    },
688                    body.as_str(),
689                )
690            })
691            .collect();
692        let read = &thread[..q.delivered_turns.min(thread.len())];
693        let unread = q.unread_from_owner();
694        let snapshot = q.thread.len();
695        let body = prompt::deputy(&prompt::DeputyPrompt {
696            id: &q.id,
697            summary: &q.summary,
698            detail: &q.detail,
699            brief: &dep.brief,
700            choices: &q.choices,
701            thread: read,
702            unread: unread.as_deref(),
703            resumed,
704            handover: false,
705            kind: kind_of(&q).unwrap_or(Kind::Conduct),
706            language: &cfg.graph.language,
707        });
708
709        // Recorded before the turn starts, so a daemon that dies mid-turn
710        // leaves a start counted and the seat on the record.
711        self.store.beat(&q.id, WaiterKind::Deputy);
712        let starts = dep.starts + 1;
713        let first = seat.clone();
714        self.store.update(&q.id, |r| {
715            if let Some(d) = r.deputy.as_mut() {
716                d.agent = spec.id.clone();
717                d.seat = Some(first);
718                d.starts = starts;
719            }
720            r.waiter = Some(Note {
721                kind: WaiterKind::Deputy,
722                since: now,
723            });
724            Ok(())
725        })?;
726        drop(release);
727        tracing::info!(
728            "question {}: deputy seat {} {} (start {starts}/{MAX_STARTS})",
729            q.short(),
730            seat.key,
731            if resumed { "resuming" } else { "starting" }
732        );
733
734        let left = (deadline(&q, cfg.graph.answer_timeout) - now.as_second()).max(0) as u64;
735        let artifacts = self.store.root().join(format!("{}.deputy", q.id));
736        let cache_dir = cfg.cache_dir();
737        // `magi ask` writes the question record and its lock, which a read-only
738        // sandbox refuses, so the seat cannot be read-only. It is told never to
739        // edit anything; the daemon, not the deputy, applies outcomes.
740        let allow_write = true;
741        // The question store is outside the repository, and `magi ask` writes it.
742        // A merge approval's deputy also files follow-up tasks with `magi task
743        // add`, which writes the queue; that is the one other place it writes.
744        let writable = [
745            self.store.root().to_path_buf(),
746            crate::queue::Queue::open().root().to_path_buf(),
747        ];
748        macro_rules! invocation {
749            ($prompt:expr, $stem:expr, $timeout:expr) => {
750                Invocation {
751                    cwd: &cwd,
752                    prompt: $prompt,
753                    timeout: $timeout,
754                    allow_write,
755                    unsandboxed: false,
756                    sessions: cfg.graph.sessions,
757                    artifacts: &artifacts,
758                    stem: $stem,
759                    // The question's own run key (the task id) is what lets this
760                    // seat's `magi ask` pass the ownership check on a conductor
761                    // question.
762                    run: &q.run,
763                    node: NODE,
764                    cache_dir: cache_dir.as_deref(),
765                    attachments: &[],
766                    writable: &writable,
767                }
768            };
769        }
770
771        // A fresh seat first takes a short turn that ends: `agent::invoke` only
772        // learns the CLI's session id when it returns, and the real turn blocks
773        // in `magi ask` for hours, so a daemon stopped mid-wait would otherwise
774        // leave nothing to resume. This turn persists the seat before the wait.
775        let mut early = None;
776        if !resumed && cfg.graph.sessions {
777            let hbody = prompt::deputy(&prompt::DeputyPrompt {
778                id: &q.id,
779                summary: &q.summary,
780                detail: &q.detail,
781                brief: &dep.brief,
782                choices: &q.choices,
783                thread: read,
784                unread: None,
785                resumed: false,
786                handover: true,
787                kind: kind_of(&q).unwrap_or(Kind::Conduct),
788                language: &cfg.graph.language,
789            });
790            let hstem = format!("handover-{starts}");
791            // Bounded by the question's own deadline, not only by its own cap: the
792            // lease it beats would otherwise keep an expired question alive.
793            let hlimit = HANDOVER_TIMEOUT.min(Duration::from_secs(left.max(1)));
794            let hinv = invocation!(&hbody, &hstem, hlimit);
795            let Some(done) = self.drive(&spec, &mut seat, &hinv, &q.id).await else {
796                self.park(&q.id);
797                return Ok(());
798            };
799            let kept = seat.clone();
800            self.store.update(&q.id, |r| {
801                if let Some(d) = r.deputy.as_mut() {
802                    d.seat = Some(kept);
803                }
804                Ok(())
805            })?;
806            if !matches!(&done, Ok(o) if o.usable()) {
807                early = Some(done);
808            }
809        }
810
811        // The handover may have been slow: look at the question again, and
812        // measure the long turn from what is left now.
813        let left = if early.is_none() && !resumed && cfg.graph.sessions {
814            let now = Timestamp::now();
815            let again = self.store.get(&q.id)?;
816            let left = (deadline(&again, cfg.graph.answer_timeout) - now.as_second()).max(0) as u64;
817            if !again.status.open() || (left == 0 && again.unread_from_owner().is_none()) {
818                self.park_quietly(&q.id);
819                return Ok(());
820            }
821            left
822        } else {
823            left
824        };
825        let out = match early {
826            Some(done) => done,
827            None => {
828                let stem = format!("turn-{starts}");
829                let timeout = Duration::from_secs(left.max(60) + SLACK_SECS);
830                let inv = invocation!(&body, &stem, timeout);
831                match self.drive(&spec, &mut seat, &inv, &q.id).await {
832                    Some(out) => out,
833                    None => {
834                        self.park(&q.id);
835                        return Ok(());
836                    }
837                }
838            }
839        };
840
841        let (text, why) = match out {
842            Ok(o) if o.usable() => (Some(o.text.trim().to_owned()), None),
843            Ok(o) if o.timed_out => (None, Some("its turn timed out".to_owned())),
844            Ok(o) if o.quota_exhausted() => (None, Some("the agent is out of quota".to_owned())),
845            Ok(o) => (
846                None,
847                Some(format!("its turn failed (exit {:?})", o.exit_code)),
848            ),
849            Err(e) => (None, Some(format!("the agent could not be started: {e:#}"))),
850        };
851        let kept = seat.clone();
852        self.store.update(&q.id, |r| {
853            if let Some(d) = r.deputy.as_mut() {
854                d.seat = Some(kept);
855            }
856            r.waiter = None;
857            // The owner's words were in the prompt. An agent that read them but
858            // answered in prose instead of `magi ask --thread` still spoke;
859            // keep it on the record so the owner reads it.
860            if let Some(text) = text
861                && unread.is_some()
862                && !text.is_empty()
863                && r.status.open()
864                && r.thread.len() == snapshot
865            {
866                r.delivered_turns = r.delivered_turns.max(snapshot);
867                let choices = r.choices.clone();
868                r.reply(text, choices)?;
869            }
870            Ok(())
871        })?;
872        self.store.drop_lease(&q.id);
873        if let Some(why) = why {
874            tracing::warn!("question {}: the deputy ended: {why}", q.short());
875            notices::raise_in(
876                &self.home,
877                Notice::warn(
878                    &format!("deputy-turn:{}", q.id),
879                    format!(
880                        "Question {} \"{}\": its follow-up agent stopped ({why}).",
881                        q.short(),
882                        q.summary
883                    ),
884                ),
885            );
886        }
887        Ok(())
888    }
889}
890
891/// The runner's loop, until `stop` is asked for.
892pub async fn run(mut deputies: Deputies, stop: crate::daemon::Stop) {
893    while !stop.stopped() {
894        deputies.tick(Timestamp::now());
895        tokio::time::sleep(TICK).await;
896    }
897}