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>
269 lines
9.2 KiB
Python
269 lines
9.2 KiB
Python
"""Sync DB with Models
|
|
|
|
Revision ID: 001
|
|
Revises:
|
|
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 = '001'
|
|
down_revision: Union[str, Sequence[str], None] = None
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Upgrade schema."""
|
|
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,
|
|
)
|
|
op.create_table(
|
|
'conversation_metadata',
|
|
sa.Column('conversation_id', sa.String(), nullable=False),
|
|
sa.Column('github_user_id', sa.String(), nullable=True),
|
|
sa.Column('user_id', sa.String(), nullable=False),
|
|
sa.Column('selected_repository', sa.String(), nullable=True),
|
|
sa.Column('selected_branch', sa.String(), nullable=True),
|
|
sa.Column('git_provider', sa.String(), nullable=True),
|
|
sa.Column('title', sa.String(), nullable=True),
|
|
sa.Column('last_updated_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('trigger', sa.String(), nullable=True),
|
|
sa.Column('pr_number', sa.JSON(), nullable=True),
|
|
sa.Column('accumulated_cost', sa.Float(), nullable=True),
|
|
sa.Column('prompt_tokens', sa.Integer(), nullable=True),
|
|
sa.Column('completion_tokens', sa.Integer(), nullable=True),
|
|
sa.Column('total_tokens', sa.Integer(), nullable=True),
|
|
sa.Column('max_budget_per_task', sa.Float(), nullable=True),
|
|
sa.Column('cache_read_tokens', sa.Integer(), nullable=True),
|
|
sa.Column('cache_write_tokens', sa.Integer(), nullable=True),
|
|
sa.Column('reasoning_tokens', sa.Integer(), nullable=True),
|
|
sa.Column('context_window', sa.Integer(), nullable=True),
|
|
sa.Column('per_turn_token', sa.Integer(), nullable=True),
|
|
sa.Column('llm_model', sa.String(), nullable=True),
|
|
sa.Column('conversation_version', sa.String(), nullable=False),
|
|
sa.Column('sandbox_id', sa.String(), nullable=True),
|
|
sa.PrimaryKeyConstraint('conversation_id'),
|
|
)
|
|
op.create_index(
|
|
op.f('ix_conversation_metadata_conversation_version'),
|
|
'conversation_metadata',
|
|
['conversation_version'],
|
|
unique=False,
|
|
)
|
|
op.create_index(
|
|
op.f('ix_conversation_metadata_sandbox_id'),
|
|
'conversation_metadata',
|
|
['sandbox_id'],
|
|
unique=False,
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Downgrade schema."""
|
|
op.drop_index(
|
|
op.f('ix_conversation_metadata_sandbox_id'), table_name='conversation_metadata'
|
|
)
|
|
op.drop_index(
|
|
op.f('ix_conversation_metadata_conversation_version'),
|
|
table_name='conversation_metadata',
|
|
)
|
|
op.drop_table('conversation_metadata')
|
|
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_index(
|
|
op.f('ix_app_conversation_info_updated_at'),
|
|
table_name='app_conversation_info',
|
|
)
|
|
op.drop_index(
|
|
op.f('ix_app_conversation_info_sandbox_id'),
|
|
table_name='app_conversation_info',
|
|
)
|
|
op.drop_index(
|
|
op.f('ix_app_conversation_info_created_by_user_id'),
|
|
table_name='app_conversation_info',
|
|
)
|
|
op.drop_index(
|
|
op.f('ix_app_conversation_info_created_at'),
|
|
table_name='app_conversation_info',
|
|
)
|
|
op.drop_table('app_conversation_info')
|