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 )zadd document classification fields to knowledge_documents

Revision ID: f1a2b3c4d5e6
Revises: e1f2a3b4c5d6
Create Date: 2026-03-31 11:00:00+08:00

    )SequenceUnion)opNf1a2b3c4d5e6revisione1f2a3b4c5d6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
ddg t dddg d S )Nknowledge_documentsdoc_type2   T)nullable
doc_numberd   issuing_authority   ix_knowledge_documents_doc_type!ix_knowledge_documents_doc_number)r   
add_columnsaColumnStringcreate_index r   r   i/lsinfo/ai/hellotax_ai/base_platform/alembic/versions/20260331_1100_add_document_classification_fields.pyupgrade   s
   r   c                   C   s@   t dd t dd t dd t dd t dd d S )Nr   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    

