from pydantic import BaseModel

class Probe(BaseModel):
    command: str
    probe_id: str

class ProbeRegister(Probe):
    capabilities: str
    probe_type: str

class ProbeHeartBeat(Probe):
    status: str

class ProbeUnregister(Probe):
    pass

class ProbeBroadcast(BaseModel):
    command: str
    probe_type: str