pub(all) enum StageFailureKind {
  SnapshotInvalid
  GatewayUnavailable
  ContextBudgetExceeded
  Stage1Invalid
  Stage2Invalid
  OperatorCancelled
} derive(Eq, ToJson, 
@moonbitlang/core/debug.Debug
Debug, 
@moonbitlang/core/json.FromJson
FromJson)

pub(all) struct StageGate {
  phase : 
@vectie/moonfish/routine.RoutinePhase
RoutinePhase
  validation : 
@vectie/moonfish/domain.ValidationReport
ValidationReport
  retry_decision : 
@vectie/moonfish/retry.RetryDecision
RetryDecision
} derive(Eq, ToJson, 
@moonbitlang/core/debug.Debug
Debug, 
@moonbitlang/core/json.FromJson
FromJson)

pub(all) enum TwoStageOutcome {
  Completed
  AwaitingStage1Retry
  AwaitingStage2Retry
  Cancelled
  Failed
} derive(Eq, ToJson, 
@moonbitlang/core/debug.Debug
Debug, 
@moonbitlang/core/json.FromJson
FromJson)

pub(all) struct TwoStageRoutineInput {
  request : 
@vectie/moonfish/routine.RoutineRequest
RoutineRequest
  gateway : 
@vectie/moonfish/model.ModelGateway
ModelGateway
  diagnosis : 
@vectie/moonfish/domain.Stage1Diagnosis
Stage1Diagnosis
  decision : 
@vectie/moonfish/domain.Stage2Decision
Stage2Decision
  retry_policy : 
@vectie/moonfish/job.RetryPolicy
RetryPolicy
  stage1_attempt : Int
  stage2_attempt : Int
  cancel_after_phase : 
@vectie/moonfish/routine.RoutinePhase
RoutinePhase?
} derive(Eq, ToJson, 
@moonbitlang/core/debug.Debug
Debug, 
@moonbitlang/core/json.FromJson
FromJson)

pub(all) struct TwoStageRoutineResult {
  request : 
@vectie/moonfish/routine.RoutineRequest
RoutineRequest
  outcome : TwoStageOutcome
  failure_kind : StageFailureKind?
  record : 
@vectie/moonfish/domain.AnalysisRecord
AnalysisRecord
  events : Array[
@vectie/moonfish/routine.RoutineEvent
RoutineEvent]
  stage_gates : Array[StageGate]
  stage1_prompt : 
@vectie/moonfish/prompt.PromptPacket
PromptPacket?
  stage2_prompt : 
@vectie/moonfish/prompt.PromptPacket
PromptPacket?
  context_budget : 
@vectie/moonfish/model.ContextBudgetReport
ContextBudgetReport?
  stage1_model_request : 
@vectie/moonfish/model.ModelCallRequest
ModelCallRequest?
  stage2_model_request : 
@vectie/moonfish/model.ModelCallRequest
ModelCallRequest?
  artifact_paths : Array[String]
  partial_record_available : Bool
  summary : String
} derive(Eq, ToJson, 
@moonbitlang/core/debug.Debug
Debug, 
@moonbitlang/core/json.FromJson
FromJson)

fn run_two_stage_routine(input : TwoStageRoutineInput) -> TwoStageRoutineResult

fn two_stage_routine_input(request : 
@vectie/moonfish/routine.RoutineRequest
RoutineRequest, gateway : 
@vectie/moonfish/model.ModelGateway
ModelGateway, diagnosis : 
@vectie/moonfish/domain.Stage1Diagnosis
Stage1Diagnosis, decision : 
@vectie/moonfish/domain.Stage2Decision
Stage2Decision, retry_policy? : 
@vectie/moonfish/job.RetryPolicy
RetryPolicy, stage1_attempt? : Int, stage2_attempt? : Int, cancel_after_phase? : 
@vectie/moonfish/routine.RoutinePhase
RoutinePhase) -> TwoStageRoutineInput