mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-25 21:36:52 +08:00
f292f3a84d
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: sp.wack <83104063+amanape@users.noreply.github.com> Co-authored-by: Engel Nyst <enyst@users.noreply.github.com>
260 lines
8.5 KiB
Python
260 lines
8.5 KiB
Python
"""Sync DB with Models
|
|
|
|
Revision ID: 076
|
|
Revises: 075
|
|
Create Date: 2025-10-05 11:28:41.772294
|
|
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
from openhands.app_server.app_conversation.app_conversation_models import (
|
|
AppConversationStartTaskStatus,
|
|
)
|
|
from openhands.app_server.event_callback.event_callback_result_models import (
|
|
EventCallbackResultStatus,
|
|
)
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '076'
|
|
down_revision: Union[str, Sequence[str], None] = '075'
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Upgrade schema."""
|
|
op.add_column(
|
|
'conversation_metadata',
|
|
sa.Column('max_budget_per_task', sa.Float(), nullable=True),
|
|
)
|
|
op.add_column(
|
|
'conversation_metadata',
|
|
sa.Column('cache_read_tokens', sa.Integer(), server_default='0'),
|
|
)
|
|
op.add_column(
|
|
'conversation_metadata',
|
|
sa.Column('cache_write_tokens', sa.Integer(), server_default='0'),
|
|
)
|
|
op.add_column(
|
|
'conversation_metadata',
|
|
sa.Column('reasoning_tokens', sa.Integer(), server_default='0'),
|
|
)
|
|
op.add_column(
|
|
'conversation_metadata',
|
|
sa.Column('context_window', sa.Integer(), server_default='0'),
|
|
)
|
|
op.add_column(
|
|
'conversation_metadata',
|
|
sa.Column('per_turn_token', sa.Integer(), server_default='0'),
|
|
)
|
|
op.add_column(
|
|
'conversation_metadata',
|
|
sa.Column(
|
|
'conversation_version', sa.String(), nullable=False, server_default='V0'
|
|
),
|
|
)
|
|
op.create_index(
|
|
op.f('ix_conversation_metadata_conversation_version'),
|
|
'conversation_metadata',
|
|
['conversation_version'],
|
|
unique=False,
|
|
)
|
|
op.add_column('conversation_metadata', sa.Column('sandbox_id', sa.String()))
|
|
op.create_index(
|
|
op.f('ix_conversation_metadata_sandbox_id'),
|
|
'conversation_metadata',
|
|
['sandbox_id'],
|
|
unique=False,
|
|
)
|
|
op.create_table(
|
|
'app_conversation_start_task',
|
|
sa.Column('id', sa.UUID(), nullable=False),
|
|
sa.Column('created_by_user_id', sa.String(), nullable=True),
|
|
sa.Column('status', sa.Enum(AppConversationStartTaskStatus), nullable=True),
|
|
sa.Column('detail', sa.String(), nullable=True),
|
|
sa.Column('app_conversation_id', sa.UUID(), nullable=True),
|
|
sa.Column('sandbox_id', sa.String(), nullable=True),
|
|
sa.Column('agent_server_url', sa.String(), nullable=True),
|
|
sa.Column('request', sa.JSON(), nullable=True),
|
|
sa.Column(
|
|
'created_at',
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text('(CURRENT_TIMESTAMP)'),
|
|
nullable=True,
|
|
),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
)
|
|
op.create_index(
|
|
op.f('ix_app_conversation_start_task_created_at'),
|
|
'app_conversation_start_task',
|
|
['created_at'],
|
|
unique=False,
|
|
)
|
|
op.create_index(
|
|
op.f('ix_app_conversation_start_task_created_by_user_id'),
|
|
'app_conversation_start_task',
|
|
['created_by_user_id'],
|
|
unique=False,
|
|
)
|
|
op.create_index(
|
|
op.f('ix_app_conversation_start_task_updated_at'),
|
|
'app_conversation_start_task',
|
|
['updated_at'],
|
|
unique=False,
|
|
)
|
|
op.create_table(
|
|
'event_callback',
|
|
sa.Column('id', sa.UUID(), nullable=False),
|
|
sa.Column('conversation_id', sa.UUID(), nullable=True),
|
|
sa.Column('processor', sa.JSON(), nullable=True),
|
|
sa.Column('event_kind', sa.String(), nullable=True),
|
|
sa.Column(
|
|
'created_at',
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text('(CURRENT_TIMESTAMP)'),
|
|
nullable=True,
|
|
),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
)
|
|
op.create_index(
|
|
op.f('ix_event_callback_created_at'),
|
|
'event_callback',
|
|
['created_at'],
|
|
unique=False,
|
|
)
|
|
op.create_table(
|
|
'event_callback_result',
|
|
sa.Column('id', sa.UUID(), nullable=False),
|
|
sa.Column('status', sa.Enum(EventCallbackResultStatus), nullable=True),
|
|
sa.Column('event_callback_id', sa.UUID(), nullable=True),
|
|
sa.Column('event_id', sa.UUID(), nullable=True),
|
|
sa.Column('conversation_id', sa.UUID(), nullable=True),
|
|
sa.Column('detail', sa.String(), nullable=True),
|
|
sa.Column(
|
|
'created_at',
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text('(CURRENT_TIMESTAMP)'),
|
|
nullable=True,
|
|
),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
)
|
|
op.create_index(
|
|
op.f('ix_event_callback_result_conversation_id'),
|
|
'event_callback_result',
|
|
['conversation_id'],
|
|
unique=False,
|
|
)
|
|
op.create_index(
|
|
op.f('ix_event_callback_result_created_at'),
|
|
'event_callback_result',
|
|
['created_at'],
|
|
unique=False,
|
|
)
|
|
op.create_index(
|
|
op.f('ix_event_callback_result_event_callback_id'),
|
|
'event_callback_result',
|
|
['event_callback_id'],
|
|
unique=False,
|
|
)
|
|
op.create_index(
|
|
op.f('ix_event_callback_result_event_id'),
|
|
'event_callback_result',
|
|
['event_id'],
|
|
unique=False,
|
|
)
|
|
op.create_table(
|
|
'v1_remote_sandbox',
|
|
sa.Column('id', sa.String(), nullable=False),
|
|
sa.Column('created_by_user_id', sa.String(), nullable=True),
|
|
sa.Column('sandbox_spec_id', sa.String(), nullable=True),
|
|
sa.Column(
|
|
'created_at',
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text('(CURRENT_TIMESTAMP)'),
|
|
nullable=True,
|
|
),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
)
|
|
op.create_index(
|
|
op.f('ix_v1_remote_sandbox_created_at'),
|
|
'v1_remote_sandbox',
|
|
['created_at'],
|
|
unique=False,
|
|
)
|
|
op.create_index(
|
|
op.f('ix_v1_remote_sandbox_created_by_user_id'),
|
|
'v1_remote_sandbox',
|
|
['created_by_user_id'],
|
|
unique=False,
|
|
)
|
|
op.create_index(
|
|
op.f('ix_v1_remote_sandbox_sandbox_spec_id'),
|
|
'v1_remote_sandbox',
|
|
['sandbox_spec_id'],
|
|
unique=False,
|
|
)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Downgrade schema."""
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_index(
|
|
op.f('ix_v1_remote_sandbox_sandbox_spec_id'), table_name='v1_remote_sandbox'
|
|
)
|
|
op.drop_index(
|
|
op.f('ix_v1_remote_sandbox_created_by_user_id'), table_name='v1_remote_sandbox'
|
|
)
|
|
op.drop_index(
|
|
op.f('ix_v1_remote_sandbox_created_at'), table_name='v1_remote_sandbox'
|
|
)
|
|
op.drop_table('v1_remote_sandbox')
|
|
op.drop_index(
|
|
op.f('ix_event_callback_result_event_id'),
|
|
table_name='event_callback_result',
|
|
)
|
|
op.drop_index(
|
|
op.f('ix_event_callback_result_event_callback_id'),
|
|
table_name='event_callback_result',
|
|
)
|
|
op.drop_index(
|
|
op.f('ix_event_callback_result_created_at'),
|
|
table_name='event_callback_result',
|
|
)
|
|
op.drop_index(
|
|
op.f('ix_event_callback_result_conversation_id'),
|
|
table_name='event_callback_result',
|
|
)
|
|
op.drop_table('event_callback_result')
|
|
op.drop_index(op.f('ix_event_callback_created_at'), table_name='event_callback')
|
|
op.drop_table('event_callback')
|
|
op.drop_index(
|
|
op.f('ix_app_conversation_start_task_updated_at'),
|
|
table_name='app_conversation_start_task',
|
|
)
|
|
op.drop_index(
|
|
op.f('ix_app_conversation_start_task_created_by_user_id'),
|
|
table_name='app_conversation_start_task',
|
|
)
|
|
op.drop_index(
|
|
op.f('ix_app_conversation_start_task_created_at'),
|
|
table_name='app_conversation_start_task',
|
|
)
|
|
op.drop_table('app_conversation_start_task')
|
|
op.drop_column('conversation_metadata', 'sandbox_id')
|
|
op.drop_column('conversation_metadata', 'conversation_version')
|
|
op.drop_column('conversation_metadata', 'per_turn_token')
|
|
op.drop_column('conversation_metadata', 'context_window')
|
|
op.drop_column('conversation_metadata', 'reasoning_tokens')
|
|
op.drop_column('conversation_metadata', 'cache_write_tokens')
|
|
op.drop_column('conversation_metadata', 'cache_read_tokens')
|
|
op.drop_column('conversation_metadata', 'max_budget_per_task')
|
|
op.execute('DROP TYPE appconversationstarttaskstatus')
|
|
op.execute('DROP TYPE eventcallbackresultstatus')
|
|
# ### end Alembic commands ###
|