o
    "i                     @   sH   d Z ddlmZmZmZmZmZmZmZ ddl	m
Z
 G dd de
ZdS )z
SFT Task model
    )ColumnStringTextIntegerJSONFloatDateTime)	BaseModelc                   @   s   e Zd ZdZdZddiZeedddZee	ddZ
eed	d
dZeeddZeedddZeeddZeeddZeedddZeeddZeed	ddZeedddZeeddZeeddZee	ddZee	ddZeeddZeeddZdS )SFTTaskz1
    SFT (Supervised Fine-Tuning) Task model
    	sft_tasksschema
hub_global   F)nullableT2   pending)default)r   indexmockg        N)__name__
__module____qualname____doc____tablename____table_args__r   r   namer   descriptionstatusr   
dataset_id
model_namer   hyperparametersresult	tenant_id
created_byplatformplatform_job_idtraining_configr   progresslogserror_messager   
started_atcompleted_at r,   r,   E/lsinfo/ai/hellotax_ai/training_center/backend/app/models/sft_task.pyr
      s*    r
   N)r   
sqlalchemyr   r   r   r   r   r   r   app.models.baser	   r
   r,   r,   r,   r-   <module>   s    $