33 lines
683 B
Python
33 lines
683 B
Python
from typing import Any, Optional
|
|
from uuid import UUID
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class PipelineRunInfo(BaseModel):
|
|
status: str
|
|
pipeline_run_id: UUID
|
|
payload: Optional[Any] = None
|
|
|
|
model_config = {
|
|
"arbitrary_types_allowed": True,
|
|
}
|
|
|
|
|
|
class PipelineRunStarted(PipelineRunInfo):
|
|
status: str = "PipelineRunStarted"
|
|
|
|
|
|
class PipelineRunYield(PipelineRunInfo):
|
|
status: str = "PipelineRunYield"
|
|
|
|
|
|
class PipelineRunCompleted(PipelineRunInfo):
|
|
status: str = "PipelineRunCompleted"
|
|
|
|
|
|
class PipelineRunErrored(PipelineRunInfo):
|
|
status: str = "PipelineRunErrored"
|
|
|
|
|
|
class PipelineRunActivity(BaseModel):
|
|
status: str = "PipelineRunActivity"
|