backend-api/pipeline/models/ingestion.py

26 lines
676 B
Python

from typing import Any
from pydantic import BaseModel, Field
class AdapterRecord(BaseModel):
"""
Record output from each adapter.
"""
source: str = Field(..., description="Source type")
data: dict[str, Any] = Field(..., description="Data output from the adapter")
class OutputData(BaseModel):
"""
Output data from a pipeline run.
"""
records: list[AdapterRecord] = Field(..., description="List of adapter records")
unified: bool | None = Field(
default=False, description="Whether the records are unified"
)
metadata: dict[str, Any] | None = Field(
default=None, description="Metadata about the run"
)