o
    "id                     @   s@   d Z ddlmZ ddlZdZdZdZdZdddZ	dd	d
Z
dS )zAdd training platform fields and template model

Revision ID: 008_add_training_platform_fields
Revises: 007_training_global_schema
Create Date: 2026-03-17 15:57:00

    )opN 008_add_training_platform_fields007_training_global_schemareturnc                   C   s  t jdtjdtddddd t jdtjdtd	d
ddd t jdtjdt d
ddd t jdtjdt dddd t jdtjdt d
ddd t jdtjdt d
ddd t jdtjdt d
ddd t jdtjdt d
ddd t j	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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tjdt ddtddd t jd ddgdd d S )!N	dpo_tasksplatform2   mock)server_default
hub_globalschemaplatform_job_id   T)nullabletraining_configprogressz0.0logserror_message
started_atcompleted_attraining_templatesidFnamedescription	task_typeconfig	is_publicfalse	tenant_id
created_by
created_at
updated_atix_training_templates_tenant_id)r   
add_columnsaColumnStringJSONFloatTextDateTimecreate_tableIntegerBooleanPrimaryKeyConstraintcreate_index r1   r1   c/lsinfo/ai/hellotax_ai/training_center/backend/alembic/versions/008_add_training_platform_fields.pyupgrade   s2   ""      r3   c                   C   s"  t jdddd t jddd t jdddd t jdddd t jdddd t jdd	dd t jdd
dd t jdddd t jdddd t jdddd t jdddd t jdddd t jdddd t jdd	dd t jdd
dd t jdddd t jdddd t jdddd d S )Nr#   r   r   r   r   r   r   r   r   r   r   r   r   	sft_tasks)r   
drop_index
drop_tabledrop_columnr1   r1   r1   r2   	downgrade8   s$   r8   )r   N)__doc__alembicr   
sqlalchemyr%   revisiondown_revisionbranch_labels
depends_onr3   r8   r1   r1   r1   r2   <module>   s    
(