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"
    )