Files

82 lines
3.5 KiB
Python

"""initial schema
Revision ID: 202605071200
Revises:
Create Date: 2026-05-07 12:00:00
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
revision: str = "202605071200"
down_revision: Union[str, None] = None
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
# Topics define durable intelligence goals instead of one-off prompts.
op.create_table(
"topics",
sa.Column("id", sa.Integer(), primary_key=True),
sa.Column("name", sa.String(length=120), nullable=False, unique=True),
sa.Column("description", sa.Text(), nullable=False),
sa.Column("enabled", sa.Boolean(), nullable=False, server_default=sa.true()),
sa.Column("created_at", sa.DateTime(), nullable=False, server_default=sa.func.now()),
sa.Column("updated_at", sa.DateTime(), nullable=False, server_default=sa.func.now()),
)
# Runs are the audit trail for every scheduled or manual agent execution.
op.create_table(
"agent_runs",
sa.Column("id", sa.Integer(), primary_key=True),
sa.Column("status", sa.String(length=32), nullable=False),
sa.Column("started_at", sa.DateTime(), nullable=False, server_default=sa.func.now()),
sa.Column("finished_at", sa.DateTime(), nullable=True),
sa.Column("error", sa.Text(), nullable=True),
sa.Column("request_payload", sa.JSON(), nullable=False),
sa.Column("raw_response", sa.JSON(), nullable=True),
)
# Signals store normalized facts that the reporter and future memory layer can query.
op.create_table(
"signals",
sa.Column("id", sa.Integer(), primary_key=True),
sa.Column("run_id", sa.Integer(), sa.ForeignKey("agent_runs.id"), nullable=False),
sa.Column("topic", sa.String(length=120), nullable=False),
sa.Column("title", sa.String(length=300), nullable=False),
sa.Column("summary", sa.Text(), nullable=False),
sa.Column("source_url", sa.Text(), nullable=False),
sa.Column("source_url_hash", sa.String(length=64), nullable=False),
sa.Column("source_name", sa.String(length=200), nullable=False),
sa.Column("published_at", sa.DateTime(), nullable=True),
sa.Column("importance", sa.Integer(), nullable=False),
sa.Column("entities", sa.JSON(), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False, server_default=sa.func.now()),
sa.UniqueConstraint("source_url_hash", name="uq_signals_source_url_hash"),
)
op.create_index("ix_signals_run_id", "signals", ["run_id"])
op.create_index("ix_signals_topic", "signals", ["topic"])
# Reports are persisted both in MySQL and as markdown files for human reading.
op.create_table(
"reports",
sa.Column("id", sa.Integer(), primary_key=True),
sa.Column("run_id", sa.Integer(), sa.ForeignKey("agent_runs.id"), nullable=False),
sa.Column("title", sa.String(length=200), nullable=False),
sa.Column("content_md", sa.Text(), nullable=False),
sa.Column("file_path", sa.String(length=1024), nullable=True),
sa.Column("created_at", sa.DateTime(), nullable=False, server_default=sa.func.now()),
)
def downgrade() -> None:
op.drop_table("reports")
op.drop_index("ix_signals_topic", table_name="signals")
op.drop_index("ix_signals_run_id", table_name="signals")
op.drop_table("signals")
op.drop_table("agent_runs")
op.drop_table("topics")