o
    "i                     @   sH   d Z ddlmZ ddlZddlmZ dZdZdZ	dZ
dd Zd	d
 ZdS )zoCreate SFT tasks table

Revision ID: 004_sft_tasks
Revises: 001_expert_training_models
Create Date: 2026-03-13
    )opN)JSON004_sft_tasks001_expert_training_modelsc                   C   s  t jdtjdt dddtjdtdddtjd	t ddtjd
tddddtjdt ddtjdtdddtjdtddtjdtddtjdtddddtjdtdddtjdtddtjdt dddtjdt ddtjdt ddtjdt	 ddtjdt	 ddtjdt dddtjdt ddtjdt	 ddtjd t	 ddtjd!t
 dddd"tjd#t	 ddtjd$t ddd%d& d S )'N	sft_tasksidT)primary_keyindexname   F)nullabledescriptionstatus2   pending)defaultr   
dataset_id
model_namehyperparametersresultplatformmockplatform_job_idtraining_configprogressg        logserror_message
started_atcompleted_at	tenant_id)r   r	   
created_by
created_at
updated_at
is_deleted)r   r   r	   
deleted_at
deleted_by
hub_globalschema)r   create_tablesaColumnIntegerStringTextr   FloatDateTimeBoolean r2   r2   W/lsinfo/ai/hellotax_ai/training_center/backend/alembic/versions/004_create_sft_tasks.pyupgrade   s6   
r4   c                   C   s   t jddd d S )Nr   r&   r'   )r   
drop_tabler2   r2   r2   r3   	downgrade/   s   r6   )__doc__alembicr   
sqlalchemyr*   sqlalchemy.dialects.postgresqlr   revisiondown_revisionbranch_labels
depends_onr4   r6   r2   r2   r2   r3   <module>   s    