# -*- coding: utf-8 -*-
from datetime import datetime
from typing import Optional

import sqlalchemy as sa
from sqlmodel import Field, SQLModel


class Prompt(SQLModel, table=True):
    __tablename__ = "prompt"
    id: Optional[int] = Field(default=None, primary_key=True)
    labeler_id: Optional[int] = Field(default=None, foreign_key="labeler.id")
    prompt: str
    response: Optional[str]
    lang: Optional[str]
    created_date: Optional[datetime] = Field(
        sa_column=sa.Column(sa.DateTime(), nullable=False, server_default=sa.func.current_timestamp()),
        nullable=False,
    )