3d9e8ffbf0
- Update docker-start.sh to force correct profiles (qdrant, admin) - Fix PostgreSQL port mapping from 5432 to 15432 across all configs - Resolve MCP import conflicts by renaming src/mcp to src/mcp_servers - Fix admin interface StaticFiles mount syntax error - Update LLM client to support both Ollama and OpenAI-compatible APIs - Configure host networking for Discord bot container access - Correct database connection handling for async context managers - Update environment variables and Docker compose configurations - Add missing production dependencies and Dockerfile improvements
224 lines
12 KiB
Python
224 lines
12 KiB
Python
"""Initial tables including creative projects
|
|
|
|
Revision ID: 004
|
|
Revises: None
|
|
Create Date: 2024-12-20 12:00:00.000000
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
# revision identifiers
|
|
revision = '004'
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
def upgrade():
|
|
# Create characters table first
|
|
op.create_table('characters',
|
|
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
|
sa.Column('name', sa.String(100), unique=True, nullable=False, index=True),
|
|
sa.Column('personality', sa.Text(), nullable=False),
|
|
sa.Column('system_prompt', sa.Text(), nullable=False),
|
|
sa.Column('interests', sa.JSON(), nullable=False, default=list),
|
|
sa.Column('speaking_style', sa.Text(), nullable=False),
|
|
sa.Column('background', sa.Text(), nullable=False),
|
|
sa.Column('avatar_url', sa.String(500)),
|
|
sa.Column('is_active', sa.Boolean(), default=True),
|
|
sa.Column('creation_date', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('last_active', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('last_message_id', sa.Integer())
|
|
)
|
|
|
|
# Create conversations table
|
|
op.create_table('conversations',
|
|
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
|
sa.Column('channel_id', sa.String(100), nullable=False, index=True),
|
|
sa.Column('topic', sa.String(200)),
|
|
sa.Column('start_time', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('end_time', sa.DateTime()),
|
|
sa.Column('is_active', sa.Boolean(), default=True),
|
|
sa.Column('participant_count', sa.Integer(), default=0)
|
|
)
|
|
|
|
# Create messages table
|
|
op.create_table('messages',
|
|
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
|
sa.Column('discord_id', sa.String(100), unique=True, index=True),
|
|
sa.Column('conversation_id', sa.Integer(), sa.ForeignKey('conversations.id')),
|
|
sa.Column('character_id', sa.Integer(), sa.ForeignKey('characters.id')),
|
|
sa.Column('content', sa.Text(), nullable=False),
|
|
sa.Column('timestamp', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('response_to_id', sa.Integer(), sa.ForeignKey('messages.id')),
|
|
sa.Column('metadata', sa.JSON(), default=dict)
|
|
)
|
|
|
|
# Add foreign key for last_message_id after messages table is created
|
|
op.create_foreign_key('fk_characters_last_message', 'characters', 'messages', ['last_message_id'], ['id'])
|
|
|
|
# Create memories table
|
|
op.create_table('memories',
|
|
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
|
sa.Column('character_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('memory_type', sa.String(50), nullable=False),
|
|
sa.Column('content', sa.Text(), nullable=False),
|
|
sa.Column('importance', sa.Float(), default=0.5),
|
|
sa.Column('timestamp', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('associated_conversation_id', sa.Integer(), sa.ForeignKey('conversations.id')),
|
|
sa.Column('associated_character_ids', sa.JSON(), default=list),
|
|
sa.Column('metadata', sa.JSON(), default=dict)
|
|
)
|
|
|
|
# Create character_relationships table
|
|
op.create_table('character_relationships',
|
|
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
|
sa.Column('character_a_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('character_b_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('relationship_type', sa.String(50), nullable=False),
|
|
sa.Column('strength', sa.Float(), default=0.0),
|
|
sa.Column('last_interaction', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('interaction_count', sa.Integer(), default=0),
|
|
sa.Column('notes', sa.Text()),
|
|
sa.Column('metadata', sa.JSON(), default=dict)
|
|
)
|
|
|
|
# Create character_evolution table
|
|
op.create_table('character_evolution',
|
|
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
|
sa.Column('character_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('change_type', sa.String(50), nullable=False),
|
|
sa.Column('old_value', sa.Text()),
|
|
sa.Column('new_value', sa.Text(), nullable=False),
|
|
sa.Column('reason', sa.Text()),
|
|
sa.Column('timestamp', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('metadata', sa.JSON(), default=dict)
|
|
)
|
|
|
|
# Create shared_memories table
|
|
op.create_table('shared_memories',
|
|
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
|
sa.Column('memory_id', sa.Integer(), sa.ForeignKey('memories.id'), nullable=False),
|
|
sa.Column('shared_by_character_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('shared_with_character_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('trust_level_at_time_of_sharing', sa.Float(), nullable=False),
|
|
sa.Column('shared_at', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('integration_status', sa.String(20), default='integrated'),
|
|
sa.Column('metadata', sa.JSON(), default=dict)
|
|
)
|
|
|
|
# Create memory_share_requests table
|
|
op.create_table('memory_share_requests',
|
|
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
|
sa.Column('requester_character_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('target_character_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('memory_id', sa.Integer(), sa.ForeignKey('memories.id'), nullable=False),
|
|
sa.Column('request_reason', sa.Text(), nullable=False),
|
|
sa.Column('status', sa.String(20), default='pending'),
|
|
sa.Column('response_reasoning', sa.Text()),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('responded_at', sa.DateTime()),
|
|
sa.Column('metadata', sa.JSON(), default=dict)
|
|
)
|
|
|
|
# Create character_trust_levels table
|
|
op.create_table('character_trust_levels',
|
|
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
|
sa.Column('character_a_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('character_b_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('trust_score', sa.Float(), default=0.3),
|
|
sa.Column('relationship_depth', sa.String(20), default='basic'),
|
|
sa.Column('interaction_count', sa.Integer(), default=0),
|
|
sa.Column('positive_interactions', sa.Integer(), default=0),
|
|
sa.Column('last_updated', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('metadata', sa.JSON(), default=dict)
|
|
)
|
|
|
|
# Create creative_projects table
|
|
op.create_table('creative_projects',
|
|
sa.Column('id', sa.String(255), primary_key=True, index=True),
|
|
sa.Column('title', sa.String(200), nullable=False),
|
|
sa.Column('description', sa.Text(), nullable=False),
|
|
sa.Column('project_type', sa.String(50), nullable=False),
|
|
sa.Column('status', sa.String(50), default='proposed'),
|
|
sa.Column('initiator_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('target_completion', sa.DateTime()),
|
|
sa.Column('project_goals', sa.JSON(), default=list),
|
|
sa.Column('style_guidelines', sa.JSON(), default=dict),
|
|
sa.Column('current_content', sa.Text(), default=''),
|
|
sa.Column('metadata', sa.JSON(), default=dict)
|
|
)
|
|
|
|
# Create indexes for creative_projects
|
|
op.create_index('ix_projects_status', 'creative_projects', ['status'])
|
|
op.create_index('ix_projects_type', 'creative_projects', ['project_type'])
|
|
op.create_index('ix_projects_initiator', 'creative_projects', ['initiator_id', 'created_at'])
|
|
|
|
# Create project_collaborators table
|
|
op.create_table('project_collaborators',
|
|
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
|
sa.Column('project_id', sa.String(255), sa.ForeignKey('creative_projects.id'), nullable=False),
|
|
sa.Column('character_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('role_description', sa.String(200), default='collaborator'),
|
|
sa.Column('joined_at', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('is_active', sa.Boolean(), default=True)
|
|
)
|
|
|
|
# Create indexes for project_collaborators
|
|
op.create_index('ix_collaborators_project', 'project_collaborators', ['project_id'])
|
|
op.create_index('ix_collaborators_character', 'project_collaborators', ['character_id'])
|
|
|
|
# Create project_contributions table
|
|
op.create_table('project_contributions',
|
|
sa.Column('id', sa.String(255), primary_key=True, index=True),
|
|
sa.Column('project_id', sa.String(255), sa.ForeignKey('creative_projects.id'), nullable=False),
|
|
sa.Column('contributor_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('contribution_type', sa.String(50), nullable=False),
|
|
sa.Column('content', sa.Text(), nullable=False),
|
|
sa.Column('timestamp', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('build_on_contribution_id', sa.String(255), sa.ForeignKey('project_contributions.id')),
|
|
sa.Column('feedback_for_contribution_id', sa.String(255), sa.ForeignKey('project_contributions.id')),
|
|
sa.Column('metadata', sa.JSON(), default=dict)
|
|
)
|
|
|
|
# Create indexes for project_contributions
|
|
op.create_index('ix_contributions_project', 'project_contributions', ['project_id', 'timestamp'])
|
|
op.create_index('ix_contributions_contributor', 'project_contributions', ['contributor_id', 'timestamp'])
|
|
op.create_index('ix_contributions_type', 'project_contributions', ['contribution_type'])
|
|
|
|
# Create project_invitations table
|
|
op.create_table('project_invitations',
|
|
sa.Column('id', sa.String(255), primary_key=True, index=True),
|
|
sa.Column('project_id', sa.String(255), sa.ForeignKey('creative_projects.id'), nullable=False),
|
|
sa.Column('inviter_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('invitee_id', sa.Integer(), sa.ForeignKey('characters.id'), nullable=False),
|
|
sa.Column('role_description', sa.String(200), default='collaborator'),
|
|
sa.Column('invitation_message', sa.Text()),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
|
sa.Column('expires_at', sa.DateTime(), nullable=False),
|
|
sa.Column('status', sa.String(50), default='pending'),
|
|
sa.Column('response_message', sa.Text()),
|
|
sa.Column('responded_at', sa.DateTime())
|
|
)
|
|
|
|
# Create indexes for project_invitations
|
|
op.create_index('ix_invitations_invitee', 'project_invitations', ['invitee_id', 'status'])
|
|
op.create_index('ix_invitations_project', 'project_invitations', ['project_id', 'created_at'])
|
|
|
|
def downgrade():
|
|
# Drop tables in reverse order of creation
|
|
op.drop_table('project_invitations')
|
|
op.drop_table('project_contributions')
|
|
op.drop_table('project_collaborators')
|
|
op.drop_table('creative_projects')
|
|
op.drop_table('character_trust_levels')
|
|
op.drop_table('memory_share_requests')
|
|
op.drop_table('shared_memories')
|
|
op.drop_table('character_evolution')
|
|
op.drop_table('character_relationships')
|
|
op.drop_table('memories')
|
|
op.drop_table('messages')
|
|
op.drop_table('conversations')
|
|
op.drop_table('characters') |