inbox.py

"""Inbox message models."""

from datetime import datetime
from enum import Enum
from typing import Literal

from pydantic import BaseModel, Field

OrchestrationType = Literal["send_message", "handoff", "assign"]


class MessageStatus(str, Enum):
    """Message status enumeration."""

    PENDING = "pending"
    DELIVERED = "delivered"
    FAILED = "failed"


class InboxMessage(BaseModel):
    """Inbox message model."""

    id: int = Field(..., description="Message ID")
    sender_id: str = Field(..., description="Sender terminal ID")
    receiver_id: str = Field(..., description="Receiver terminal ID")
    message: str = Field(..., description="Message content")
    orchestration_type: OrchestrationType = Field(
        default="send_message",
        description="The orchestration mode that caused the message delivery",
    )
    status: MessageStatus = Field(..., description="Message status")
    created_at: datetime = Field(..., description="Creation timestamp")