o
    "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	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 )zeInitial migration

Revision ID: fce73aea7a0b
Revises:
Create Date: 2026-03-16 15:58:27.829116+08:00

    )SequenceUnion)opNfce73aea7a0brevisiondown_revisionbranch_labels
depends_onreturnc                  C   s   ddl m}  ddlm}m}m}m}m}m}m	}m
}m}	m}
m}m}m}m}m}m}m}m}m}m} ddlm} zddlm} W n	 tyI   Y nw zddlm}m}m}m } W n	 tya   Y nw | j!j"t#$ d dS )	u!   升级数据库 - 创建所有表r   BaseTenantUserRoleMenuUserRoleRoleMenuModelProviderModelAgentChatMessageKnowledgeBaseKnowledgeCategoryKnowledgeTagKnowledgeDocumentDocumentTagDocumentVectorDocumentVersionKnowledgeQAAuditLogPermissionAuditLog)
CasbinRuleRolePermissionTagCategoryTagAutoRuleKnowledgeMetadataFieldDocumentMetadataValuebindN)%app.db.baser   
app.modelsr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   app.models.casbin_ruler"   app.models.role_permissionr$   ImportErrorapp.models.data_modelr&   r'   r(   r)   metadata
create_allr   get_bind)r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r$   r&   r'   r(   r)    r5   h/lsinfo/ai/hellotax_ai/base_platform/alembic/versions/20260316_1558_fce73aea7a0b_init_database_schema.pyupgrade   s   Xr7   c                  C   s   ddl m}  ddlm}m}m}m}m}m}m	}m
}m}	m}
m}m}m}m}m}m}m}m}m}m} zddlm} W n	 tyC   Y nw zddlm}m}m}m} W n	 ty[   Y nw | jj t!" d dS )u!   降级数据库 - 删除所有表r   r   r   r#   r%   r*   N)#r,   r   r-   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r/   r$   r0   r1   r&   r'   r(   r)   r2   drop_allr   r4   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r$   r&   r'   r(   r)   r5   r5   r6   	downgrade.   s   Xr9   )r
   N)__doc__typingr   r   alembicr   
sqlalchemysar   str__annotations__r   r   r	   r7   r9   r5   r5   r5   r6   <module>   s    
