feat: run fixed research pipelines within durable budgets

2026-09-08 22:16:05 +08:00
parent eb4850a003
commit 7860434b04
25 changed files with 2357 additions and 37 deletions
@@ -0,0 +1,50 @@
+"""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")