from pydantic import BaseModel
from typing import Dict, Any

class GeneratePlanRequest(BaseModel):
    user_prompt: str
    drone_state: str = "on_ground" # "on_ground" or "in_air"

class ExecuteMissionRequest(BaseModel):
    py_tree: Dict[str, Any]

class StatusUpdate(BaseModel):
    node_id: str
    status: int


class DebugStageRequest(BaseModel):
    """Stage 分阶段调试请求"""
    user_prompt: str
    drone_state: str = "on_ground"
    target_stage: int = 1  # 1-6