"""create_knowledge_tables

Revision ID: 75a075297f5c
Revises: 002a9cec4967
Create Date: 2026-06-03 19:29:34.178885
"""
from collections.abc import Sequence

from alembic import op
import sqlalchemy as sa


revision: str = '75a075297f5c'
down_revision: str | None = '002a9cec4967'
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('flovy_knowledge_sources',
    sa.Column('tenant_id', sa.String(length=36), nullable=False),
    sa.Column('source_type', sa.Enum('product', 'faq', 'url', 'pdf', 'manual', name='ks_source_type'), nullable=False),
    sa.Column('title', sa.String(length=500), nullable=True),
    sa.Column('url', sa.String(length=1000), nullable=True),
    sa.Column('content', sa.Text(), nullable=True),
    sa.Column('status', sa.Enum('pending', 'indexing', 'indexed', 'failed', name='ks_status'), server_default='pending', nullable=False),
    sa.Column('chunk_count', sa.Integer(), server_default='0', nullable=False),
    sa.Column('last_indexed_at', sa.DateTime(), nullable=True),
    sa.Column('error_message', sa.Text(), nullable=True),
    sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
    sa.Column('id', sa.String(length=36), nullable=False),
    sa.ForeignKeyConstraint(['tenant_id'], ['tenants.id'], ondelete='CASCADE'),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index(op.f('ix_flovy_knowledge_sources_tenant_id'), 'flovy_knowledge_sources', ['tenant_id'], unique=False)
    op.create_table('flovy_knowledge_chunks',
    sa.Column('tenant_id', sa.String(length=36), nullable=False),
    sa.Column('source_id', sa.String(length=36), nullable=False),
    sa.Column('source_type', sa.String(length=20), nullable=False),
    sa.Column('content', sa.Text(), nullable=False),
    sa.Column('embedding', sa.JSON(), nullable=True),
    sa.Column('embedding_model', sa.String(length=100), nullable=True),
    sa.Column('meta', sa.JSON(), nullable=True),
    sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
    sa.Column('id', sa.String(length=36), nullable=False),
    sa.ForeignKeyConstraint(['source_id'], ['flovy_knowledge_sources.id'], ondelete='CASCADE'),
    sa.ForeignKeyConstraint(['tenant_id'], ['tenants.id'], ondelete='CASCADE'),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index('idx_chunk_tenant_source_type', 'flovy_knowledge_chunks', ['tenant_id', 'source_type'], unique=False)
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_index('idx_chunk_tenant_source_type', table_name='flovy_knowledge_chunks')
    op.drop_table('flovy_knowledge_chunks')
    op.drop_index(op.f('ix_flovy_knowledge_sources_tenant_id'), table_name='flovy_knowledge_sources')
    op.drop_table('flovy_knowledge_sources')
    # ### end Alembic commands ###