def should_continue_to_pricing( state: AgentState, ) -> Literal["calculate_price", "end"]: if state["booking_details"].is_complete: return "calculate_price" return "end"