0aac93e9c4
<!-- .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. --------- Co-authored-by: vasilije <vas.markovic@gmail.com> Co-authored-by: Igor Ilic <30923996+dexters1@users.noreply.github.com> Co-authored-by: Vasilije <8619304+Vasilije1990@users.noreply.github.com> Co-authored-by: Igor Ilic <igorilic03@gmail.com> Co-authored-by: Hande <159312713+hande-k@users.noreply.github.com> Co-authored-by: Matea Pesic <80577904+matea16@users.noreply.github.com> Co-authored-by: hajdul88 <52442977+hajdul88@users.noreply.github.com> Co-authored-by: Daniel Molnar <soobrosa@gmail.com> Co-authored-by: Diego Baptista Theuerkauf <34717973+diegoabt@users.noreply.github.com>
27 lines
949 B
Python
27 lines
949 B
Python
import enum
|
|
from uuid import uuid4
|
|
from datetime import datetime, timezone
|
|
from sqlalchemy import Column, DateTime, JSON, Enum, UUID, String
|
|
from cognee.infrastructure.databases.relational import Base
|
|
|
|
|
|
class PipelineRunStatus(enum.Enum):
|
|
DATASET_PROCESSING_INITIATED = "DATASET_PROCESSING_INITIATED"
|
|
DATASET_PROCESSING_STARTED = "DATASET_PROCESSING_STARTED"
|
|
DATASET_PROCESSING_COMPLETED = "DATASET_PROCESSING_COMPLETED"
|
|
DATASET_PROCESSING_ERRORED = "DATASET_PROCESSING_ERRORED"
|
|
|
|
|
|
class PipelineRun(Base):
|
|
__tablename__ = "pipeline_runs"
|
|
|
|
id = Column(UUID, primary_key=True, default=uuid4)
|
|
|
|
created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
|
|
|
|
status = Column(Enum(PipelineRunStatus))
|
|
pipeline_run_id = Column(UUID, index=True)
|
|
pipeline_name = Column(String)
|
|
pipeline_id = Column(UUID, index=True)
|
|
dataset_id = Column(UUID, index=True)
|
|
run_info = Column(JSON)
|