yuxuanhui
51 lines
2.2 KiB
Python
51 lines
2.2 KiB
Python
"""Stage three: finite research authorizations and durable steps."""
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision = "0008"
|
|
down_revision = "0007"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
op.create_table(
|
|
"research_flow_runs",
|
|
sa.Column("id", sa.String(36), primary_key=True),
|
|
sa.Column("request_id", sa.String(100), nullable=False, unique=True),
|
|
sa.Column("name", sa.String(200), nullable=False),
|
|
sa.Column("definition", sa.JSON(), nullable=False),
|
|
sa.Column("authorization", sa.JSON(), nullable=False),
|
|
sa.Column("model_revision", sa.Integer(), nullable=True),
|
|
sa.Column("status", sa.String(30), nullable=False),
|
|
sa.Column("version", sa.Integer(), nullable=False),
|
|
sa.Column("round", sa.Integer(), nullable=False),
|
|
sa.Column("simulations_used", sa.Integer(), nullable=False),
|
|
sa.Column("model_calls_used", sa.Integer(), nullable=False),
|
|
sa.Column("error", sa.Text(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
|
|
)
|
|
op.create_index("ix_research_flow_runs_status", "research_flow_runs", ["status"])
|
|
op.create_table(
|
|
"research_step_runs",
|
|
sa.Column("id", sa.String(36), primary_key=True),
|
|
sa.Column("run_id", sa.String(36), sa.ForeignKey("research_flow_runs.id"), nullable=False),
|
|
sa.Column("node_id", sa.String(100), nullable=False),
|
|
sa.Column("round", sa.Integer(), nullable=False),
|
|
sa.Column("status", sa.String(30), nullable=False),
|
|
sa.Column("output", sa.JSON(), nullable=False),
|
|
sa.Column("backtest_run_id", sa.String(36), sa.ForeignKey("backtest_runs.id"), nullable=True),
|
|
sa.Column("error", sa.Text(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.UniqueConstraint("run_id", "node_id", "round"),
|
|
)
|
|
op.create_index("ix_research_step_runs_run_id", "research_step_runs", ["run_id"])
|
|
|
|
|
|
def downgrade():
|
|
op.drop_table("research_step_runs")
|
|
op.drop_table("research_flow_runs")
|