from datetime import datetime
from typing import Literal

from pydantic import BaseModel, Field


class Result(BaseModel):
    message: str
    success: bool


class Message(BaseModel):
    content: str = Field(..., description='The content of the message')
    name: str = Field(
        default='', description='The name of the episodic node for the message (message uuid)'
    )
    role_type: Literal['user', 'assistant', 'system'] = Field(
        ..., description='The role type of the message (user, assistant or system)'
    )
    role: str | None = Field(
        description='The custom role of the message to be used alongside role_type (user name, bot name, etc.)',
    )
    timestamp: datetime = Field(
        default_factory=datetime.now, description='The timestamp of the message'
    )
    source_description: str = Field(
        default='', description='The description of the source of the message'
    )