14be2a5f5d
<!-- .github/pull_request_template.md --> ## Description <!-- Provide a clear description of the changes in this PR --> ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin.
35 lines
682 B
Python
35 lines
682 B
Python
from typing import Any, Optional
|
|
from uuid import UUID
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class PipelineRunInfo(BaseModel):
|
|
status: str
|
|
pipeline_run_id: UUID
|
|
dataset_id: UUID
|
|
dataset_name: str
|
|
payload: Optional[Any] = None
|
|
|
|
model_config = {
|
|
"arbitrary_types_allowed": True,
|
|
}
|
|
|
|
|
|
class PipelineRunStarted(PipelineRunInfo):
|
|
status: str = "PipelineRunStarted"
|
|
pass
|
|
|
|
|
|
class PipelineRunYield(PipelineRunInfo):
|
|
status: str = "PipelineRunYield"
|
|
pass
|
|
|
|
|
|
class PipelineRunCompleted(PipelineRunInfo):
|
|
status: str = "PipelineRunCompleted"
|
|
pass
|
|
|
|
|
|
class PipelineRunErrored(PipelineRunInfo):
|
|
status: str = "PipelineRunErrored"
|
|
pass
|