codetether-agent 4.7.0-a-002.4

//! Prompt execution loop for resumable `codetether run`.
//!
//! When `--auto-continue-until N` is set, the loop will attempt up to `N`
//! resume cycles after the step budget is exhausted. `Some(1)` means the
//! initial run plus one resume attempt; `None` (default) means no automatic
//! resume.

use crate::session::Session;
use anyhow::Result;
use std::path::Path;

pub async fn execute_prompt_with_resume(
    session: &mut Session,
    message: &str,
    max_steps: Option<usize>,
    resume_attempts: Option<usize>,
    workspace: &Path,
) -> Result<crate::session::SessionResult> {
    let mut prompt_text = message.to_string();
    let mut resumes_left = resume_attempts.unwrap_or(0);
    loop {
        let before = session.messages.len();
        let result = session.prompt(&prompt_text).await?;
        if !crate::session::step_limit::was_budget_exhausted() {
            session.clear_run_checkpoint().await?;
            return Ok(result);
        }
        let budget = max_steps.unwrap_or(crate::session::DEFAULT_MAX_STEPS);
        if super::run_checkpoint::should_checkpoint(before, session, budget) {
            super::run_checkpoint::persist_exhaustion_checkpoint(
                session, message, budget, workspace,
            )
            .await?;
        }
        if resumes_left == 0 {
            return Ok(result);
        }
        resumes_left -= 1;
        let Some(plan) = super::run_checkpoint::resume_plan(session, resumes_left).await? else {
            return Ok(result);
        };
        session.max_steps = Some(budget);
        prompt_text = plan.prompt;
    }
}