o
    "Æiã  ã                   @   sh   d 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
Training Dataset schemas
é    )ÚOptionalÚDictÚAny)Ú	BaseModelc                   @   sF   e Zd ZU eed< dZee ed< eed< dZeeee	f  ed< dS )ÚDatasetCreateÚnameNÚdescriptionÚdataset_typeÚmetadata)
Ú__name__Ú
__module__Ú__qualname__ÚstrÚ__annotations__r   r   r
   r   r   © r   r   úE/lsinfo/ai/hellotax_ai/training_center/backend/app/schemas/dataset.pyr      s
   
 r   c                   @   s>   e Zd ZU dZee ed< dZee ed< dZee ed< dS )ÚDatasetUpdateNr   r   Ústatus)	r   r   r   r   r   r   r   r   r   r   r   r   r   r      s   
 r   c                   @   sN   e Zd ZU 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 )ÚSampleCreateÚcontentÚlabelÚsplitNÚsource_task_idr
   )r   r   r   r   r   r   r   Úintr
   r   r   r   r   r   r   r      s   
 r   c                   @   s¨   e Zd ZU eed< eed< ee ed< eed< eed< ee ed< eed< eed< eed	< eed
< ee ed< ee ed< ee ed< eeef ed< G dd„ dƒZ	dS )ÚDatasetResponseÚidr   r   r	   ÚversionÚparent_version_idÚtotal_samplesÚtrain_samplesÚvalidation_samplesÚtest_samplesÚ
created_byÚ
created_atÚ
updated_atÚ	meta_datac                   @   s   e Zd ZdZdS )zDatasetResponse.ConfigTN)r   r   r   Úfrom_attributesr   r   r   r   ÚConfig-   s    r'   N)
r   r   r   r   r   r   r   r   r   r'   r   r   r   r   r      s    
 r   N)Ú__doc__Útypingr   r   r   Úpydanticr   r   r   r   r   r   r   r   r   Ú<module>   s    