575 lines
24 KiB
Python
575 lines
24 KiB
Python
|
"""init
|
||
|
|
|
||
|
|
Revision ID: 5630abfa60f8
|
||
|
|
Revises:
|
||
|
|
Create Date: 2026-02-03 17:52:47.887105
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from alembic import op
|
||
|
|
import sqlalchemy as sa
|
||
|
|
import sqlmodel
|
||
|
|
|
||
|
|
|
||
|
|
# revision identifiers, used by Alembic.
|
||
|
|
revision = "5630abfa60f8"
|
||
|
|
down_revision = None
|
||
|
|
branch_labels = None
|
||
|
|
depends_on = None
|
||
|
|
|
||
|
|
|
||
|
|
def upgrade() -> None:
|
||
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
||
|
|
op.create_table(
|
||
|
|
"orgs",
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("slug", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(op.f("ix_orgs_slug"), "orgs", ["slug"], unique=True)
|
||
|
|
op.create_table(
|
||
|
|
"users",
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("clerk_user_id", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("email", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
||
|
|
sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
||
|
|
sa.Column("is_super_admin", sa.Boolean(), nullable=False),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(op.f("ix_users_clerk_user_id"), "users", ["clerk_user_id"], unique=True)
|
||
|
|
op.create_index(op.f("ix_users_email"), "users", ["email"], unique=False)
|
||
|
|
op.create_table(
|
||
|
|
"workspaces",
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("slug", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(op.f("ix_workspaces_org_id"), "workspaces", ["org_id"], unique=False)
|
||
|
|
op.create_index(op.f("ix_workspaces_slug"), "workspaces", ["slug"], unique=False)
|
||
|
|
op.create_table(
|
||
|
|
"agents",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("role", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
||
|
|
sa.Column("status", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("openclaw_session_id", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
||
|
|
sa.Column("api_token_hash", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
||
|
|
sa.Column("api_token_last_used_at", sa.DateTime(), nullable=True),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.Column("updated_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_agents_openclaw_session_id"), "agents", ["openclaw_session_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(op.f("ix_agents_org_id"), "agents", ["org_id"], unique=False)
|
||
|
|
op.create_index(op.f("ix_agents_workspace_id"), "agents", ["workspace_id"], unique=False)
|
||
|
|
op.create_table(
|
||
|
|
"gateway_configs",
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=True),
|
||
|
|
sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("base_url", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("token", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(op.f("ix_gateway_configs_org_id"), "gateway_configs", ["org_id"], unique=False)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_gateway_configs_workspace_id"), "gateway_configs", ["workspace_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_table(
|
||
|
|
"memberships",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("user_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("role", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["user_id"],
|
||
|
|
["users.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(op.f("ix_memberships_org_id"), "memberships", ["org_id"], unique=False)
|
||
|
|
op.create_index(op.f("ix_memberships_user_id"), "memberships", ["user_id"], unique=False)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_memberships_workspace_id"), "memberships", ["workspace_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_table(
|
||
|
|
"orchestration_templates",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("kind", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("title", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("template_markdown", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.Column("updated_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_orchestration_templates_kind"), "orchestration_templates", ["kind"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_orchestration_templates_org_id"),
|
||
|
|
"orchestration_templates",
|
||
|
|
["org_id"],
|
||
|
|
unique=False,
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_orchestration_templates_workspace_id"),
|
||
|
|
"orchestration_templates",
|
||
|
|
["workspace_id"],
|
||
|
|
unique=False,
|
||
|
|
)
|
||
|
|
op.create_table(
|
||
|
|
"projects",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("description", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
||
|
|
sa.Column("status", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.Column("updated_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(op.f("ix_projects_org_id"), "projects", ["org_id"], unique=False)
|
||
|
|
op.create_index(op.f("ix_projects_status"), "projects", ["status"], unique=False)
|
||
|
|
op.create_index(op.f("ix_projects_workspace_id"), "projects", ["workspace_id"], unique=False)
|
||
|
|
op.create_table(
|
||
|
|
"workspace_api_tokens",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("token_hash", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("label", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
||
|
|
sa.Column("last_used_at", sa.DateTime(), nullable=True),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_workspace_api_tokens_org_id"), "workspace_api_tokens", ["org_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_workspace_api_tokens_token_hash"),
|
||
|
|
"workspace_api_tokens",
|
||
|
|
["token_hash"],
|
||
|
|
unique=True,
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_workspace_api_tokens_workspace_id"),
|
||
|
|
"workspace_api_tokens",
|
||
|
|
["workspace_id"],
|
||
|
|
unique=False,
|
||
|
|
)
|
||
|
|
op.create_table(
|
||
|
|
"openclaw_sessions",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("session_id", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("agent_id", sa.Uuid(), nullable=True),
|
||
|
|
sa.Column("status", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("last_seen_at", sa.DateTime(), nullable=True),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["agent_id"],
|
||
|
|
["agents.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_openclaw_sessions_org_id"), "openclaw_sessions", ["org_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_openclaw_sessions_session_id"), "openclaw_sessions", ["session_id"], unique=True
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_openclaw_sessions_status"), "openclaw_sessions", ["status"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_openclaw_sessions_workspace_id"),
|
||
|
|
"openclaw_sessions",
|
||
|
|
["workspace_id"],
|
||
|
|
unique=False,
|
||
|
|
)
|
||
|
|
op.create_table(
|
||
|
|
"tasks",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("project_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("title", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("description", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
||
|
|
sa.Column("status", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("priority", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("due_at", sa.DateTime(), nullable=True),
|
||
|
|
sa.Column("assigned_agent_id", sa.Uuid(), nullable=True),
|
||
|
|
sa.Column("created_by_user_id", sa.Uuid(), nullable=True),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.Column("updated_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["assigned_agent_id"],
|
||
|
|
["agents.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["created_by_user_id"],
|
||
|
|
["users.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["project_id"],
|
||
|
|
["projects.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_tasks_assigned_agent_id"), "tasks", ["assigned_agent_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_tasks_created_by_user_id"), "tasks", ["created_by_user_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(op.f("ix_tasks_org_id"), "tasks", ["org_id"], unique=False)
|
||
|
|
op.create_index(op.f("ix_tasks_priority"), "tasks", ["priority"], unique=False)
|
||
|
|
op.create_index(op.f("ix_tasks_project_id"), "tasks", ["project_id"], unique=False)
|
||
|
|
op.create_index(op.f("ix_tasks_status"), "tasks", ["status"], unique=False)
|
||
|
|
op.create_index(op.f("ix_tasks_workspace_id"), "tasks", ["workspace_id"], unique=False)
|
||
|
|
op.create_table(
|
||
|
|
"task_activities",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("task_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("activity_type", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("message", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("actor_user_id", sa.Uuid(), nullable=True),
|
||
|
|
sa.Column("actor_agent_id", sa.Uuid(), nullable=True),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["actor_agent_id"],
|
||
|
|
["agents.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["actor_user_id"],
|
||
|
|
["users.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["task_id"],
|
||
|
|
["tasks.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(op.f("ix_task_activities_org_id"), "task_activities", ["org_id"], unique=False)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_task_activities_task_id"), "task_activities", ["task_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_task_activities_workspace_id"), "task_activities", ["workspace_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_table(
|
||
|
|
"task_deliverables",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("task_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("title", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("markdown_content", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("created_by_user_id", sa.Uuid(), nullable=True),
|
||
|
|
sa.Column("created_by_agent_id", sa.Uuid(), nullable=True),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["created_by_agent_id"],
|
||
|
|
["agents.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["created_by_user_id"],
|
||
|
|
["users.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["task_id"],
|
||
|
|
["tasks.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_task_deliverables_org_id"), "task_deliverables", ["org_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_task_deliverables_task_id"), "task_deliverables", ["task_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_task_deliverables_workspace_id"),
|
||
|
|
"task_deliverables",
|
||
|
|
["workspace_id"],
|
||
|
|
unique=False,
|
||
|
|
)
|
||
|
|
op.create_table(
|
||
|
|
"task_status_history",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("task_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("from_status", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
||
|
|
sa.Column("to_status", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("actor_user_id", sa.Uuid(), nullable=True),
|
||
|
|
sa.Column("actor_agent_id", sa.Uuid(), nullable=True),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["actor_agent_id"],
|
||
|
|
["agents.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["actor_user_id"],
|
||
|
|
["users.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["task_id"],
|
||
|
|
["tasks.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_task_status_history_org_id"), "task_status_history", ["org_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_task_status_history_task_id"), "task_status_history", ["task_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_task_status_history_workspace_id"),
|
||
|
|
"task_status_history",
|
||
|
|
["workspace_id"],
|
||
|
|
unique=False,
|
||
|
|
)
|
||
|
|
op.create_table(
|
||
|
|
"task_subagents",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("task_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("agent_name", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("openclaw_session_id", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["task_id"],
|
||
|
|
["tasks.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(op.f("ix_task_subagents_org_id"), "task_subagents", ["org_id"], unique=False)
|
||
|
|
op.create_index(op.f("ix_task_subagents_task_id"), "task_subagents", ["task_id"], unique=False)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_task_subagents_workspace_id"), "task_subagents", ["workspace_id"], unique=False
|
||
|
|
)
|
||
|
|
op.create_table(
|
||
|
|
"transcripts",
|
||
|
|
sa.Column("org_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("workspace_id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("id", sa.Uuid(), nullable=False),
|
||
|
|
sa.Column("task_id", sa.Uuid(), nullable=True),
|
||
|
|
sa.Column("session_id", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("full_text", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
||
|
|
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["org_id"],
|
||
|
|
["orgs.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["task_id"],
|
||
|
|
["tasks.id"],
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
["workspace_id"],
|
||
|
|
["workspaces.id"],
|
||
|
|
),
|
||
|
|
sa.PrimaryKeyConstraint("id"),
|
||
|
|
)
|
||
|
|
op.create_index(op.f("ix_transcripts_org_id"), "transcripts", ["org_id"], unique=False)
|
||
|
|
op.create_index(op.f("ix_transcripts_session_id"), "transcripts", ["session_id"], unique=False)
|
||
|
|
op.create_index(op.f("ix_transcripts_task_id"), "transcripts", ["task_id"], unique=False)
|
||
|
|
op.create_index(
|
||
|
|
op.f("ix_transcripts_workspace_id"), "transcripts", ["workspace_id"], unique=False
|
||
|
|
)
|
||
|
|
# ### end Alembic commands ###
|
||
|
|
|
||
|
|
|
||
|
|
def downgrade() -> None:
|
||
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
||
|
|
op.drop_index(op.f("ix_transcripts_workspace_id"), table_name="transcripts")
|
||
|
|
op.drop_index(op.f("ix_transcripts_task_id"), table_name="transcripts")
|
||
|
|
op.drop_index(op.f("ix_transcripts_session_id"), table_name="transcripts")
|
||
|
|
op.drop_index(op.f("ix_transcripts_org_id"), table_name="transcripts")
|
||
|
|
op.drop_table("transcripts")
|
||
|
|
op.drop_index(op.f("ix_task_subagents_workspace_id"), table_name="task_subagents")
|
||
|
|
op.drop_index(op.f("ix_task_subagents_task_id"), table_name="task_subagents")
|
||
|
|
op.drop_index(op.f("ix_task_subagents_org_id"), table_name="task_subagents")
|
||
|
|
op.drop_table("task_subagents")
|
||
|
|
op.drop_index(op.f("ix_task_status_history_workspace_id"), table_name="task_status_history")
|
||
|
|
op.drop_index(op.f("ix_task_status_history_task_id"), table_name="task_status_history")
|
||
|
|
op.drop_index(op.f("ix_task_status_history_org_id"), table_name="task_status_history")
|
||
|
|
op.drop_table("task_status_history")
|
||
|
|
op.drop_index(op.f("ix_task_deliverables_workspace_id"), table_name="task_deliverables")
|
||
|
|
op.drop_index(op.f("ix_task_deliverables_task_id"), table_name="task_deliverables")
|
||
|
|
op.drop_index(op.f("ix_task_deliverables_org_id"), table_name="task_deliverables")
|
||
|
|
op.drop_table("task_deliverables")
|
||
|
|
op.drop_index(op.f("ix_task_activities_workspace_id"), table_name="task_activities")
|
||
|
|
op.drop_index(op.f("ix_task_activities_task_id"), table_name="task_activities")
|
||
|
|
op.drop_index(op.f("ix_task_activities_org_id"), table_name="task_activities")
|
||
|
|
op.drop_table("task_activities")
|
||
|
|
op.drop_index(op.f("ix_tasks_workspace_id"), table_name="tasks")
|
||
|
|
op.drop_index(op.f("ix_tasks_status"), table_name="tasks")
|
||
|
|
op.drop_index(op.f("ix_tasks_project_id"), table_name="tasks")
|
||
|
|
op.drop_index(op.f("ix_tasks_priority"), table_name="tasks")
|
||
|
|
op.drop_index(op.f("ix_tasks_org_id"), table_name="tasks")
|
||
|
|
op.drop_index(op.f("ix_tasks_created_by_user_id"), table_name="tasks")
|
||
|
|
op.drop_index(op.f("ix_tasks_assigned_agent_id"), table_name="tasks")
|
||
|
|
op.drop_table("tasks")
|
||
|
|
op.drop_index(op.f("ix_openclaw_sessions_workspace_id"), table_name="openclaw_sessions")
|
||
|
|
op.drop_index(op.f("ix_openclaw_sessions_status"), table_name="openclaw_sessions")
|
||
|
|
op.drop_index(op.f("ix_openclaw_sessions_session_id"), table_name="openclaw_sessions")
|
||
|
|
op.drop_index(op.f("ix_openclaw_sessions_org_id"), table_name="openclaw_sessions")
|
||
|
|
op.drop_table("openclaw_sessions")
|
||
|
|
op.drop_index(op.f("ix_workspace_api_tokens_workspace_id"), table_name="workspace_api_tokens")
|
||
|
|
op.drop_index(op.f("ix_workspace_api_tokens_token_hash"), table_name="workspace_api_tokens")
|
||
|
|
op.drop_index(op.f("ix_workspace_api_tokens_org_id"), table_name="workspace_api_tokens")
|
||
|
|
op.drop_table("workspace_api_tokens")
|
||
|
|
op.drop_index(op.f("ix_projects_workspace_id"), table_name="projects")
|
||
|
|
op.drop_index(op.f("ix_projects_status"), table_name="projects")
|
||
|
|
op.drop_index(op.f("ix_projects_org_id"), table_name="projects")
|
||
|
|
op.drop_table("projects")
|
||
|
|
op.drop_index(
|
||
|
|
op.f("ix_orchestration_templates_workspace_id"), table_name="orchestration_templates"
|
||
|
|
)
|
||
|
|
op.drop_index(op.f("ix_orchestration_templates_org_id"), table_name="orchestration_templates")
|
||
|
|
op.drop_index(op.f("ix_orchestration_templates_kind"), table_name="orchestration_templates")
|
||
|
|
op.drop_table("orchestration_templates")
|
||
|
|
op.drop_index(op.f("ix_memberships_workspace_id"), table_name="memberships")
|
||
|
|
op.drop_index(op.f("ix_memberships_user_id"), table_name="memberships")
|
||
|
|
op.drop_index(op.f("ix_memberships_org_id"), table_name="memberships")
|
||
|
|
op.drop_table("memberships")
|
||
|
|
op.drop_index(op.f("ix_gateway_configs_workspace_id"), table_name="gateway_configs")
|
||
|
|
op.drop_index(op.f("ix_gateway_configs_org_id"), table_name="gateway_configs")
|
||
|
|
op.drop_table("gateway_configs")
|
||
|
|
op.drop_index(op.f("ix_agents_workspace_id"), table_name="agents")
|
||
|
|
op.drop_index(op.f("ix_agents_org_id"), table_name="agents")
|
||
|
|
op.drop_index(op.f("ix_agents_openclaw_session_id"), table_name="agents")
|
||
|
|
op.drop_table("agents")
|
||
|
|
op.drop_index(op.f("ix_workspaces_slug"), table_name="workspaces")
|
||
|
|
op.drop_index(op.f("ix_workspaces_org_id"), table_name="workspaces")
|
||
|
|
op.drop_table("workspaces")
|
||
|
|
op.drop_index(op.f("ix_users_email"), table_name="users")
|
||
|
|
op.drop_index(op.f("ix_users_clerk_user_id"), table_name="users")
|
||
|
|
op.drop_table("users")
|
||
|
|
op.drop_index(op.f("ix_orgs_slug"), table_name="orgs")
|
||
|
|
op.drop_table("orgs")
|
||
|
|
# ### end Alembic commands ###
|