from typing import Optional
from datetime import datetime
from pydantic import BaseModel


class LlmPromptSchema(BaseModel):
    is_default: bool
    id: int
    text: str
    name: str
    type: str
    date_created: datetime

    class Config:
        json_schema_extra = {
            'example': {
                'is_default': True,
                'text': 'Я большой и страшный промпт. Я в поросятах знаю толк.',
                'name': 'Большой и страшный промпт',
                'type': 'system'
            }
        }

class LlmPromptCreateSchema(BaseModel):
    is_default: bool
    text: str
    name: str
    type: str
    class Config:
        json_schema_extra = {
            'example': {
                'is_default': True,
                'text': 'Я большой и страшный промпт. Я в поросятах знаю толк.',
                'name': 'Большой и страшный промпт',
                'type': 'system'
            }
        }