82 lines
3.9 KiB
Python
82 lines
3.9 KiB
Python
"""add lorebook tables
|
|
|
|
Revision ID: 004
|
|
Revises: 003
|
|
Create Date: 2026-07-24 12:20:38.453954
|
|
|
|
"""
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
|
|
from alembic import op
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '004'
|
|
down_revision: str | Sequence[str] | None = '003'
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Upgrade schema."""
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.create_table('lorebooks',
|
|
sa.Column('id', sa.String(length=80), nullable=False),
|
|
sa.Column('character_id', sa.String(length=80), nullable=True),
|
|
sa.Column('name', sa.String(length=200), nullable=False),
|
|
sa.Column('description', sa.Text(), nullable=False),
|
|
sa.Column('scan_depth', sa.Integer(), nullable=True),
|
|
sa.Column('token_budget', sa.Integer(), nullable=True),
|
|
sa.Column('recursive_scanning', sa.Boolean(), nullable=False),
|
|
sa.Column('scope', sa.String(length=40), nullable=False),
|
|
sa.Column('status', sa.String(length=40), nullable=False),
|
|
sa.Column('owner_scope', sa.String(length=80), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
|
|
sa.ForeignKeyConstraint(['character_id'], ['characters.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_lorebooks_character_id'), 'lorebooks', ['character_id'], unique=False)
|
|
op.create_index(op.f('ix_lorebooks_owner_scope'), 'lorebooks', ['owner_scope'], unique=False)
|
|
op.create_index(op.f('ix_lorebooks_scope'), 'lorebooks', ['scope'], unique=False)
|
|
op.create_index(op.f('ix_lorebooks_status'), 'lorebooks', ['status'], unique=False)
|
|
op.create_table('lorebook_entries',
|
|
sa.Column('id', sa.String(length=80), nullable=False),
|
|
sa.Column('lorebook_id', sa.String(length=80), nullable=False),
|
|
sa.Column('keys_json', sa.Text(), nullable=False),
|
|
sa.Column('content', sa.Text(), nullable=False),
|
|
sa.Column('secondary_keys_json', sa.Text(), nullable=False),
|
|
sa.Column('constant', sa.Boolean(), nullable=False),
|
|
sa.Column('selective', sa.Boolean(), nullable=False),
|
|
sa.Column('enabled', sa.Boolean(), nullable=False),
|
|
sa.Column('case_sensitive', sa.Boolean(), nullable=False),
|
|
sa.Column('use_regex', sa.Boolean(), nullable=False),
|
|
sa.Column('position', sa.String(length=40), nullable=False),
|
|
sa.Column('insertion_order', sa.Integer(), nullable=False),
|
|
sa.Column('priority', sa.Integer(), nullable=False),
|
|
sa.Column('comment', sa.String(length=500), nullable=False),
|
|
sa.Column('bound_entity_id', sa.String(length=80), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
|
|
sa.ForeignKeyConstraint(['lorebook_id'], ['lorebooks.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_lorebook_entries_bound_entity_id'), 'lorebook_entries', ['bound_entity_id'], unique=False)
|
|
op.create_index(op.f('ix_lorebook_entries_lorebook_id'), 'lorebook_entries', ['lorebook_id'], unique=False)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Downgrade schema."""
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_index(op.f('ix_lorebook_entries_lorebook_id'), table_name='lorebook_entries')
|
|
op.drop_index(op.f('ix_lorebook_entries_bound_entity_id'), table_name='lorebook_entries')
|
|
op.drop_table('lorebook_entries')
|
|
op.drop_index(op.f('ix_lorebooks_status'), table_name='lorebooks')
|
|
op.drop_index(op.f('ix_lorebooks_scope'), table_name='lorebooks')
|
|
op.drop_index(op.f('ix_lorebooks_owner_scope'), table_name='lorebooks')
|
|
op.drop_index(op.f('ix_lorebooks_character_id'), table_name='lorebooks')
|
|
op.drop_table('lorebooks')
|
|
# ### end Alembic commands ###
|