21 lines
480 B
Python
21 lines
480 B
Python
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
|