42 lines
1015 B
Python
42 lines
1015 B
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Literal, Self
|
|
from uuid import UUID
|
|
|
|
from pydantic import model_validator
|
|
from sqlmodel import SQLModel
|
|
|
|
ApprovalStatus = Literal["pending", "approved", "rejected"]
|
|
|
|
|
|
class ApprovalBase(SQLModel):
|
|
action_type: str
|
|
task_id: UUID | None = None
|
|
payload: dict[str, object] | None = None
|
|
confidence: int
|
|
rubric_scores: dict[str, int] | None = None
|
|
status: ApprovalStatus = "pending"
|
|
|
|
|
|
class ApprovalCreate(ApprovalBase):
|
|
agent_id: UUID | None = None
|
|
|
|
|
|
class ApprovalUpdate(SQLModel):
|
|
status: ApprovalStatus | None = None
|
|
|
|
@model_validator(mode="after")
|
|
def validate_status(self) -> Self:
|
|
if "status" in self.model_fields_set and self.status is None:
|
|
raise ValueError("status is required")
|
|
return self
|
|
|
|
|
|
class ApprovalRead(ApprovalBase):
|
|
id: UUID
|
|
board_id: UUID
|
|
agent_id: UUID | None = None
|
|
created_at: datetime
|
|
resolved_at: datetime | None = None
|