from sqlalchemy import JSON, Boolean, Column, Integer, String, Text

from app.models.base import BaseModel


class TrainingTemplate(BaseModel):
    __tablename__ = 'training_templates'
    __table_args__ = {'schema': 'hub_global'}
    name = Column(String(255), nullable=False)
    description = Column(Text, nullable=True)
    task_type = Column(String(50), nullable=False)
    platform = Column(String(50), nullable=False)
    config = Column(JSON, nullable=False)
    is_public = Column(Boolean, default=False)
    tenant_id = Column(Integer, nullable=False, index=True)
    created_by = Column(Integer, nullable=True)