o
    BÌi<	  ã                   @   sœ   U d Z ddlmZmZ ddlmZ ddlZdZe	e
d< dZee	ee	 df e
d< dZee	ee	 df e
d	< dZee	ee	 df e
d
< ddd„Zddd„ZdS )zˆadd hybrid chunking fields to document_vectors

Revision ID: e1f2a3b4c5d6
Revises: d4e5f6a7b8c9
Create Date: 2026-03-31 10:00:00+08:00

é    )ÚSequenceÚUnion)ÚopNÚe1f2a3b4c5d6ÚrevisionÚd4e5f6a7b8c9Údown_revisionÚbranch_labelsÚ
depends_onÚreturnc                	   C   s$  t  dtjdt d¡dd¡ t  dtjdt ¡ ddd	¡ t  dtjd
t d¡dd¡ t  dtjdt d¡dd¡ t  dtjdt d¡dd¡ t  dtjdt d¡dd¡ t  dtjdt d¡dd¡ t  dtjdt d¡dd¡ t  dddg¡ t  dddg¡ t  dddg¡ d S )NÚdocument_vectorsÚparent_chunk_idéd   T)ÚnullableÚ	is_parentÚfalseF)Úserver_defaultr   Úchunk_levelé   Úprev_chunk_idÚnext_chunk_idÚdoc_typeé2   Ú
doc_numberÚissuing_authorityÚ#ix_document_vectors_parent_chunk_idÚix_document_vectors_is_parentÚix_document_vectors_doc_type)r   Ú
add_columnÚsaÚColumnÚStringÚBooleanÚcreate_index© r$   r$   úa/lsinfo/ai/hellotax_ai/base_platform/alembic/versions/20260331_1000_add_hybrid_chunking_fields.pyÚupgrade   s   r&   c                   C   sˆ   t  dd¡ t  dd¡ t  dd¡ t  dd¡ t  dd¡ t  dd¡ t  dd¡ t  dd	¡ t  dd
¡ t  dd¡ t  dd¡ d S )Nr   r   r   r   r   r   r   r   r   r   r   r   )r   Ú
drop_indexÚdrop_columnr$   r$   r$   r%   Ú	downgrade#   s   r)   )r   N)Ú__doc__Útypingr   r   Úalembicr   Ú
sqlalchemyr   r   ÚstrÚ__annotations__r   r	   r
   r&   r)   r$   r$   r$   r%   Ú<module>   s    
