o
    "ÆiÙ  ã                   @   st   d Z ddlmZ ddlmZmZmZ ddlmZ G dd„ deƒZG dd„ deƒZ	G d	d
„ d
eƒZ
G dd„ deƒZdS )z
DPO Task schemas
é    )Ú	BaseModel)ÚOptionalÚDictÚAny)Údatetimec                   @   sb   e Zd ZU dZeed< dZee ed< eed< eed< dZ	ee ed< dZ
eeeef  ed	< dS )
ÚDPOTaskBasezBase DPO task schemaÚnameNÚdescriptionÚ
dataset_idÚ
model_nameÚ
qwen_localÚplatformÚhyperparameters)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚstrÚ__annotations__r	   r   Úintr   r   r   r   © r   r   úF/lsinfo/ai/hellotax_ai/training_center/backend/app/schemas/dpo_task.pyr   	   s   
 r   c                   @   ó   e Zd ZdZdS )ÚDPOTaskCreatezSchema for creating a DPO taskN)r   r   r   r   r   r   r   r   r      s    r   c                   @   sr   e Zd ZU dZdZee ed< dZee ed< dZ	ee ed< dZ
eeeef  ed< dZeeeef  ed< dS )ÚDPOTaskUpdatezSchema for updating a DPO taskNr   r	   Ústatusr   Úresult)r   r   r   r   r   r   r   r   r	   r   r   r   r   r   r   r   r   r   r      s   
 r   c                   @   sX   e Zd ZU dZeed< eed< dZee	ee
f  ed< eed< eed< G dd	„ d	ƒZdS )
ÚDPOTaskResponsezSchema for DPO task responseÚidr   Nr   Ú
created_atÚ
updated_atc                   @   r   )zDPOTaskResponse.ConfigTN)r   r   r   Úfrom_attributesr   r   r   r   ÚConfig)   s    r"   )r   r   r   r   r   r   r   r   r   r   r   r   r"   r   r   r   r   r   !   s   
 r   N)r   Úpydanticr   Útypingr   r   r   r   r   r   r   r   r   r   r   r   Ú<module>   s    
	