@weirdfingers/baseboards 0.2.0

package/templates/api/alembic/script.py.mako

@@ -0,0 +1,28 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+${imports if imports else ""}
+
+# revision identifiers, used by Alembic.
+revision: str = ${repr(up_revision)}
+down_revision: Union[str, Sequence[str], None] = ${repr(down_revision)}
+branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
+depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    ${downgrades if downgrades else "pass"}

package/templates/api/alembic/versions/20250101_000000_initial_schema.py

@@ -0,0 +1,448 @@
+"""
+Initial schema with uuid-ossp extension and core tables.
+
+Revision ID: 20250101_000000_initial_schema
+Revises:
+Create Date: 2025-01-01 00:00:00
+"""
+
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+from alembic import op  # type: ignore[reportMissingImports]
+
+# revision identifiers, used by Alembic.
+revision = "20250101_000000_initial_schema"
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # Required extension for uuid_generate_v4
+    op.execute('CREATE EXTENSION IF NOT EXISTS "uuid-ossp";')
+
+    # tenants
+    op.create_table(
+        "tenants",
+        sa.Column(
+            "id",
+            postgresql.UUID(as_uuid=False),
+            server_default=sa.text("uuid_generate_v4()"),
+        ),
+        sa.Column("name", sa.String(length=255), nullable=False),
+        sa.Column("slug", sa.String(length=255), nullable=False),
+        sa.Column(
+            "settings",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+        ),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.PrimaryKeyConstraint("id", name="tenants_pkey"),
+        sa.UniqueConstraint("slug", name="tenants_slug_key"),
+    )
+
+    # provider_configs
+    op.create_table(
+        "provider_configs",
+        sa.Column(
+            "id",
+            postgresql.UUID(as_uuid=False),
+            server_default=sa.text("uuid_generate_v4()"),
+        ),
+        sa.Column("tenant_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("provider_name", sa.String(length=100), nullable=False),
+        sa.Column("config", postgresql.JSONB(astext_type=sa.Text()), nullable=False),
+        sa.Column("is_enabled", sa.Boolean(), server_default=sa.text("true")),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.PrimaryKeyConstraint("id", name="provider_configs_pkey"),
+        sa.ForeignKeyConstraint(
+            ["tenant_id"],
+            ["tenants.id"],
+            ondelete="CASCADE",
+            name="provider_configs_tenant_id_fkey",
+        ),
+        sa.UniqueConstraint(
+            "tenant_id",
+            "provider_name",
+            name="provider_configs_tenant_id_provider_name_key",
+        ),
+    )
+
+    # users
+    op.create_table(
+        "users",
+        sa.Column(
+            "id",
+            postgresql.UUID(as_uuid=False),
+            server_default=sa.text("uuid_generate_v4()"),
+        ),
+        sa.Column("tenant_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("auth_provider", sa.String(length=50), nullable=False),
+        sa.Column("auth_subject", sa.String(length=255), nullable=False),
+        sa.Column("email", sa.String(length=255)),
+        sa.Column("display_name", sa.String(length=255)),
+        sa.Column("avatar_url", sa.Text()),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+        ),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.PrimaryKeyConstraint("id", name="users_pkey"),
+        sa.ForeignKeyConstraint(
+            ["tenant_id"],
+            ["tenants.id"],
+            ondelete="CASCADE",
+            name="users_tenant_id_fkey",
+        ),
+        sa.UniqueConstraint(
+            "tenant_id",
+            "auth_provider",
+            "auth_subject",
+            name="users_tenant_id_auth_provider_auth_subject_key",
+        ),
+    )
+    op.create_index(
+        "idx_users_auth", "users", ["auth_provider", "auth_subject"], unique=False
+    )
+    op.create_index("idx_users_tenant", "users", ["tenant_id"], unique=False)
+
+    # boards
+    op.create_table(
+        "boards",
+        sa.Column(
+            "id",
+            postgresql.UUID(as_uuid=False),
+            server_default=sa.text("uuid_generate_v4()"),
+        ),
+        sa.Column("tenant_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("owner_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("title", sa.String(length=255), nullable=False),
+        sa.Column("description", sa.Text()),
+        sa.Column("is_public", sa.Boolean(), server_default=sa.text("false")),
+        sa.Column(
+            "settings",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+        ),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+        ),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.PrimaryKeyConstraint("id", name="boards_pkey"),
+        sa.ForeignKeyConstraint(
+            ["owner_id"], ["users.id"], ondelete="CASCADE", name="boards_owner_id_fkey"
+        ),
+        sa.ForeignKeyConstraint(
+            ["tenant_id"],
+            ["tenants.id"],
+            ondelete="CASCADE",
+            name="boards_tenant_id_fkey",
+        ),
+    )
+    op.create_index("idx_boards_owner", "boards", ["owner_id"], unique=False)
+    op.create_index("idx_boards_tenant", "boards", ["tenant_id"], unique=False)
+
+    # lora_models
+    op.create_table(
+        "lora_models",
+        sa.Column(
+            "id",
+            postgresql.UUID(as_uuid=False),
+            server_default=sa.text("uuid_generate_v4()"),
+        ),
+        sa.Column("tenant_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("user_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("name", sa.String(length=255), nullable=False),
+        sa.Column("base_model", sa.String(length=100), nullable=False),
+        sa.Column("storage_url", sa.Text(), nullable=False),
+        sa.Column("config", postgresql.JSONB(astext_type=sa.Text()), nullable=False),
+        sa.Column("trigger_word", sa.String(length=100)),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+        ),
+        sa.Column("is_public", sa.Boolean(), server_default=sa.text("false")),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.PrimaryKeyConstraint("id", name="lora_models_pkey"),
+        sa.ForeignKeyConstraint(
+            ["tenant_id"],
+            ["tenants.id"],
+            ondelete="CASCADE",
+            name="lora_models_tenant_id_fkey",
+        ),
+        sa.ForeignKeyConstraint(
+            ["user_id"],
+            ["users.id"],
+            ondelete="CASCADE",
+            name="lora_models_user_id_fkey",
+        ),
+    )
+
+    # board_members
+    op.create_table(
+        "board_members",
+        sa.Column(
+            "id",
+            postgresql.UUID(as_uuid=False),
+            server_default=sa.text("uuid_generate_v4()"),
+        ),
+        sa.Column("board_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("user_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("role", sa.String(length=20), nullable=False),
+        sa.Column("invited_by", postgresql.UUID(as_uuid=False)),
+        sa.Column(
+            "joined_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.PrimaryKeyConstraint("id", name="board_members_pkey"),
+        sa.UniqueConstraint(
+            "board_id", "user_id", name="board_members_board_id_user_id_key"
+        ),
+        sa.CheckConstraint(
+            "role::text = ANY (ARRAY['viewer'::character varying, "
+            "'editor'::character varying, 'admin'::character varying]::text[])",
+            name="board_members_role_check",
+        ),
+        sa.ForeignKeyConstraint(
+            ["board_id"],
+            ["boards.id"],
+            ondelete="CASCADE",
+            name="board_members_board_id_fkey",
+        ),
+        sa.ForeignKeyConstraint(
+            ["invited_by"], ["users.id"], name="board_members_invited_by_fkey"
+        ),
+        sa.ForeignKeyConstraint(
+            ["user_id"],
+            ["users.id"],
+            ondelete="CASCADE",
+            name="board_members_user_id_fkey",
+        ),
+    )
+    op.create_index(
+        "idx_board_members_board", "board_members", ["board_id"], unique=False
+    )
+    op.create_index(
+        "idx_board_members_user", "board_members", ["user_id"], unique=False
+    )
+
+    # generations
+    op.create_table(
+        "generations",
+        sa.Column(
+            "id",
+            postgresql.UUID(as_uuid=False),
+            server_default=sa.text("uuid_generate_v4()"),
+        ),
+        sa.Column("tenant_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("board_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("user_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("generator_name", sa.String(length=100), nullable=False),
+        sa.Column("provider_name", sa.String(length=100), nullable=False),
+        sa.Column("artifact_type", sa.String(length=50), nullable=False),
+        sa.Column(
+            "input_params", postgresql.JSONB(astext_type=sa.Text()), nullable=False
+        ),
+        sa.Column(
+            "status",
+            sa.String(length=50),
+            nullable=False,
+            server_default=sa.text("'pending'::character varying"),
+        ),
+        sa.Column("storage_url", sa.Text()),
+        sa.Column("thumbnail_url", sa.Text()),
+        sa.Column(
+            "additional_files",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'[]'::jsonb"),
+        ),
+        sa.Column(
+            "output_metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+        ),
+        sa.Column("parent_generation_id", postgresql.UUID(as_uuid=False)),
+        sa.Column(
+            "input_generation_ids",
+            postgresql.ARRAY(postgresql.UUID()),
+            server_default=sa.text("'{}'::uuid[]"),
+        ),
+        sa.Column("external_job_id", sa.String(length=255)),
+        sa.Column("progress", sa.Numeric(5, 2), server_default=sa.text("0.0")),
+        sa.Column("error_message", sa.Text()),
+        sa.Column("started_at", sa.DateTime(timezone=True)),
+        sa.Column("completed_at", sa.DateTime(timezone=True)),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.PrimaryKeyConstraint("id", name="generations_pkey"),
+        sa.ForeignKeyConstraint(
+            ["board_id"],
+            ["boards.id"],
+            ondelete="CASCADE",
+            name="generations_board_id_fkey",
+        ),
+        sa.ForeignKeyConstraint(
+            ["parent_generation_id"],
+            ["generations.id"],
+            name="generations_parent_generation_id_fkey",
+        ),
+        sa.ForeignKeyConstraint(
+            ["tenant_id"],
+            ["tenants.id"],
+            ondelete="CASCADE",
+            name="generations_tenant_id_fkey",
+        ),
+        sa.ForeignKeyConstraint(
+            ["user_id"],
+            ["users.id"],
+            ondelete="CASCADE",
+            name="generations_user_id_fkey",
+        ),
+    )
+    op.create_index("idx_generations_board", "generations", ["board_id"], unique=False)
+    op.create_index(
+        "idx_generations_lineage", "generations", ["parent_generation_id"], unique=False
+    )
+    op.create_index("idx_generations_status", "generations", ["status"], unique=False)
+    op.create_index(
+        "idx_generations_tenant", "generations", ["tenant_id"], unique=False
+    )
+    op.create_index("idx_generations_user", "generations", ["user_id"], unique=False)
+
+    # credit_transactions
+    op.create_table(
+        "credit_transactions",
+        sa.Column(
+            "id",
+            postgresql.UUID(as_uuid=False),
+            server_default=sa.text("uuid_generate_v4()"),
+        ),
+        sa.Column("tenant_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("user_id", postgresql.UUID(as_uuid=False), nullable=False),
+        sa.Column("transaction_type", sa.String(length=20), nullable=False),
+        sa.Column("amount", sa.Numeric(10, 4), nullable=False),
+        sa.Column("balance_after", sa.Numeric(10, 4), nullable=False),
+        sa.Column("generation_id", postgresql.UUID(as_uuid=False)),
+        sa.Column("description", sa.Text()),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+        ),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+        ),
+        sa.PrimaryKeyConstraint("id", name="credit_transactions_pkey"),
+        sa.ForeignKeyConstraint(
+            ["generation_id"],
+            ["generations.id"],
+            name="credit_transactions_generation_id_fkey",
+        ),
+        sa.ForeignKeyConstraint(
+            ["tenant_id"],
+            ["tenants.id"],
+            ondelete="CASCADE",
+            name="credit_transactions_tenant_id_fkey",
+        ),
+        sa.ForeignKeyConstraint(
+            ["user_id"],
+            ["users.id"],
+            ondelete="CASCADE",
+            name="credit_transactions_user_id_fkey",
+        ),
+    )
+    op.create_index(
+        "idx_credit_transactions_user", "credit_transactions", ["user_id"], unique=False
+    )
+
+
+def downgrade() -> None:
+    # drop in reverse dependency order
+    op.drop_index("idx_credit_transactions_user", table_name="credit_transactions")
+    op.drop_table("credit_transactions")
+
+    op.drop_index("idx_generations_user", table_name="generations")
+    op.drop_index("idx_generations_tenant", table_name="generations")
+    op.drop_index("idx_generations_status", table_name="generations")
+    op.drop_index("idx_generations_lineage", table_name="generations")
+    op.drop_index("idx_generations_board", table_name="generations")
+    op.drop_table("generations")
+
+    op.drop_index("idx_board_members_user", table_name="board_members")
+    op.drop_index("idx_board_members_board", table_name="board_members")
+    op.drop_table("board_members")
+
+    op.drop_table("lora_models")
+
+    op.drop_index("idx_boards_tenant", table_name="boards")
+    op.drop_index("idx_boards_owner", table_name="boards")
+    op.drop_table("boards")
+
+    op.drop_index("idx_users_tenant", table_name="users")
+    op.drop_index("idx_users_auth", table_name="users")
+    op.drop_table("users")
+
+    op.drop_table("provider_configs")
+
+    op.drop_table("tenants")

package/templates/api/alembic/versions/20251022_174729_remove_provider_name_from_generations.py

@@ -0,0 +1,71 @@
+"""remove provider_name from generations
+
+Revision ID: d7c16af77c9e
+Revises: 553dc6a50a20
+Create Date: 2025-10-22 17:47:29.312285
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision: str = "d7c16af77c9e"
+down_revision: Union[str, Sequence[str], None] = "553dc6a50a20"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    # Drop schema_migrations table if it exists (from legacy migration system)
+    op.execute("DROP TABLE IF EXISTS schema_migrations")
+    op.drop_column("generations", "provider_name")
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    # Add column as nullable first, then update existing rows, then make it NOT NULL
+    op.add_column(
+        "generations",
+        sa.Column(
+            "provider_name", sa.VARCHAR(length=100), autoincrement=False, nullable=True
+        ),
+    )
+    # Set a default value for existing rows (though this downgrade should rarely be used)
+    op.execute(
+        "UPDATE generations SET provider_name = 'unknown' WHERE provider_name IS NULL"
+    )
+    # Now make it NOT NULL
+    op.alter_column("generations", "provider_name", nullable=False)
+    op.create_table(
+        "schema_migrations",
+        sa.Column(
+            "version", sa.VARCHAR(length=255), autoincrement=False, nullable=False
+        ),
+        sa.Column("name", sa.VARCHAR(length=255), autoincrement=False, nullable=False),
+        sa.Column(
+            "applied_at",
+            postgresql.TIMESTAMP(timezone=True),
+            server_default=sa.text("CURRENT_TIMESTAMP"),
+            autoincrement=False,
+            nullable=True,
+        ),
+        sa.Column(
+            "execution_time_ms", sa.INTEGER(), autoincrement=False, nullable=True
+        ),
+        sa.Column(
+            "checksum", sa.VARCHAR(length=64), autoincrement=False, nullable=True
+        ),
+        sa.Column(
+            "applied_by", sa.VARCHAR(length=255), autoincrement=False, nullable=True
+        ),
+        sa.PrimaryKeyConstraint("version", name=op.f("schema_migrations_pkey")),
+    )
+    # ### end Alembic commands ###