| 1 | from typing import List |
| 2 | |
| 3 | from pydantic import BaseModel |
| 4 | from pydantic import Field |
| 5 | |
| 6 | |
| 7 | class DecommissionedData(BaseModel): |
| 8 | agents_deleted: List[str] = Field( |
| 9 | ..., |
| 10 | examples=[["agent1", "agent2"]], |
| 11 | description="List of agents deleted", |
| 12 | ) |
| 13 | groups_deleted: List[str] = Field( |
| 14 | ..., |
| 15 | examples=[["group1", "group2"]], |
| 16 | description="List of groups deleted", |
| 17 | ) |
| 18 | stream_deleted: str = Field(..., examples=["stream1"], description="Stream deleted") |
| 19 | index_deleted: str = Field(..., examples=["index1"], description="Index deleted") |
| 20 | |
| 21 | |
| 22 | class DecommissionCustomerResponse(BaseModel): |
| 23 | message: str = Field( |
| 24 | ..., |
| 25 | examples=["Customer decommissioned successfully"], |
| 26 | description="Message indicating the customer was decommissioned successfully", |
| 27 | ) |
| 28 | success: bool = Field( |
| 29 | ..., |
| 30 | examples=[True], |
| 31 | description="Whether the customer was decommissioned successfully or not", |
| 32 | ) |
| 33 | decomissioned_data: DecommissionedData = Field( |
| 34 | ..., |
| 35 | description="Data from the decomissioning process", |
| 36 | ) |