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 )zbCreate DPO tasks table

Revision ID: 005_dpo_tasks
Revises: 004_sft_tasks
Create Date: 2026-03-13
    )opN)JSON005_dpo_tasks004_sft_tasksc                   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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	dpo_tasksidT)primary_keyindexname   F)nullabledescriptionstatus2   pending)defaultr   
dataset_id
model_namehyperparametersresult	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   DateTimeBoolean r(   r(   W/lsinfo/ai/hellotax_ai/training_center/backend/alembic/versions/005_create_dpo_tasks.pyupgrade   s&   
r*   c                   C   s   t jddd d S )Nr   r   r   )r   
drop_tabler(   r(   r(   r)   	downgrade'   s   r,   )__doc__alembicr   
sqlalchemyr!   sqlalchemy.dialects.postgresqlr   revisiondown_revisionbranch_labels
depends_onr*   r,   r(   r(   r(   r)   <module>   s    