from pydantic import BaseModel
from typing import Dict, Any

class GeneratePlanRequest(BaseModel):
    user_prompt: str

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

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