"""Phase III: Cognitive Descent."""

from __future__ import annotations

import time

from ..core import DescentVector
from ..types import PhaseOutput, RunConfig, SolutionState


def run(state: SolutionState, config: RunConfig) -> tuple[SolutionState, PhaseOutput]:
    t0 = time.perf_counter()
    # cache_mode "offline" means: no network. Use deterministic heuristic
    # refinement so the descent stays reproducible (and tests stay hermetic).
    descent = DescentVector(
        offline=(config.cache_mode == "offline"),
        model=config.model_name,
    )
    initial_context = {
        "problem": state.problem,
        "block_solutions": state.block_solutions,
        "building_height": state.building_height,
        "blocks": state.blocks,
    }
    descent_result = descent.cognitive_descent(state.building_height, initial_context)
    dt = time.perf_counter() - t0
    state.extra["descent_result"] = descent_result
    return (
        state,
        PhaseOutput(
            name="phase3_cognitive_descent",
            input={"building_height": state.building_height},
            output={
                "final_answer": descent_result.get("final_answer", ""),
                "counterfactual_summary": descent_result.get("counterfactual_summary", ""),
                "attribution_explanation": descent_result.get("attribution_explanation", ""),
            },
            duration_sec=dt,
        ),
    )