File size: 993 Bytes
b404de3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | from typing import Any, Dict, List, Optional
from pydantic import BaseModel, Field
class PipelineNodeRequest(BaseModel):
service: str
policies: List[Dict]
config: Optional[Dict[str, Any]] = Field(default_factory=dict)
class CreatePipelineRequest(BaseModel):
pipeline_id: Optional[str] = Field(None, description="Optional pipeline ID")
broken_pipeline: Optional[Dict[str, List[PipelineNodeRequest]]] = Field(
None, description="Broken pipeline JSON"
)
ideal_pipeline: Optional[Dict[str, List[PipelineNodeRequest]]] = Field(
None, description="Ideal pipeline JSON"
)
class UpdatePipelineRequest(BaseModel):
pipeline_id: str = Field(..., description="Pipeline ID to update")
broken_pipeline: Optional[Dict[str, List[PipelineNodeRequest]]] = Field(
None, description="Broken pipeline JSON"
)
ideal_pipeline: Optional[Dict[str, List[PipelineNodeRequest]]] = Field(
None, description="Ideal pipeline JSON"
)
|