import uuid
from datetime import datetime
from typing import Any

from pydantic import BaseModel

from app.models.prompt_template import TemplateType


class TemplateCreate(BaseModel):
    name: str
    description: str = ""
    type: TemplateType = TemplateType.custom
    system_prompt: str
    user_prompt: str
    output_schema: dict[str, Any] = {}
    is_default: bool = False


class TemplateUpdate(BaseModel):
    name: str | None = None
    description: str | None = None
    system_prompt: str | None = None
    user_prompt: str | None = None
    output_schema: dict[str, Any] | None = None
    is_default: bool | None = None


class TemplateResponse(BaseModel):
    id: uuid.UUID
    name: str
    description: str
    type: TemplateType
    system_prompt: str
    user_prompt: str
    output_schema: dict[str, Any]
    is_default: bool
    created_at: datetime

    model_config = {"from_attributes": True}