@weirdfingers/baseboards 0.5.2 → 0.6.0

package/templates/api/alembic/versions/20251023_165852_switch_to_declarative_base_and_mapping.py

@@ -7,7 +7,6 @@ Create Date: 2025-10-23 16:58:52.004927
 """
 
 from collections.abc import Sequence
-from typing import Union
 
 import sqlalchemy as sa
 from sqlalchemy.dialects import postgresql
@@ -20,6 +19,9 @@ down_revision: str | Sequence[str] | None = "d7c16af77c9e"
 branch_labels: str | Sequence[str] | None = None
 depends_on: str | Sequence[str] | None = None
 
+# Schema name for all Boards tables
+SCHEMA = "boards"
+
 
 def upgrade() -> None:
     """Upgrade schema."""
@@ -30,6 +32,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "boards",
@@ -37,6 +40,7 @@ def upgrade() -> None:
         existing_type=sa.BOOLEAN(),
         nullable=False,
         existing_server_default=sa.text("false"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "boards",
@@ -44,6 +48,7 @@ def upgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=False,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "boards",
@@ -51,6 +56,7 @@ def upgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=False,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "boards",
@@ -58,6 +64,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "boards",
@@ -65,6 +72,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "credit_transactions",
@@ -72,6 +80,7 @@ def upgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=False,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "credit_transactions",
@@ -79,6 +88,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -86,6 +96,7 @@ def upgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=False,
         existing_server_default=sa.text("'[]'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -93,6 +104,7 @@ def upgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=False,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -100,6 +112,7 @@ def upgrade() -> None:
         existing_type=postgresql.ARRAY(sa.UUID()),
         nullable=False,
         existing_server_default=sa.text("'{}'::uuid[]"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -107,6 +120,7 @@ def upgrade() -> None:
         existing_type=sa.NUMERIC(precision=5, scale=2),
         nullable=False,
         existing_server_default=sa.text("0.0"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -114,6 +128,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -121,6 +136,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "lora_models",
@@ -128,6 +144,7 @@ def upgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=False,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "lora_models",
@@ -135,6 +152,7 @@ def upgrade() -> None:
         existing_type=sa.BOOLEAN(),
         nullable=False,
         existing_server_default=sa.text("false"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "lora_models",
@@ -142,6 +160,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "lora_models",
@@ -149,6 +168,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "provider_configs",
@@ -156,6 +176,7 @@ def upgrade() -> None:
         existing_type=sa.BOOLEAN(),
         nullable=False,
         existing_server_default=sa.text("true"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "provider_configs",
@@ -163,6 +184,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "provider_configs",
@@ -170,6 +192,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "tenants",
@@ -177,6 +200,7 @@ def upgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=False,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "tenants",
@@ -184,6 +208,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "tenants",
@@ -191,6 +216,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "users",
@@ -198,6 +224,7 @@ def upgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=False,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "users",
@@ -205,6 +232,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "users",
@@ -212,6 +240,7 @@ def upgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=False,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     # ### end Alembic commands ###
 
@@ -225,6 +254,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "users",
@@ -232,6 +262,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "users",
@@ -239,6 +270,7 @@ def downgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=True,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "tenants",
@@ -246,6 +278,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "tenants",
@@ -253,6 +286,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "tenants",
@@ -260,6 +294,7 @@ def downgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=True,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "provider_configs",
@@ -267,6 +302,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "provider_configs",
@@ -274,6 +310,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "provider_configs",
@@ -281,6 +318,7 @@ def downgrade() -> None:
         existing_type=sa.BOOLEAN(),
         nullable=True,
         existing_server_default=sa.text("true"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "lora_models",
@@ -288,6 +326,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "lora_models",
@@ -295,6 +334,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "lora_models",
@@ -302,6 +342,7 @@ def downgrade() -> None:
         existing_type=sa.BOOLEAN(),
         nullable=True,
         existing_server_default=sa.text("false"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "lora_models",
@@ -309,6 +350,7 @@ def downgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=True,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -316,6 +358,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -323,6 +366,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -330,6 +374,7 @@ def downgrade() -> None:
         existing_type=sa.NUMERIC(precision=5, scale=2),
         nullable=True,
         existing_server_default=sa.text("0.0"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -337,6 +382,7 @@ def downgrade() -> None:
         existing_type=postgresql.ARRAY(sa.UUID()),
         nullable=True,
         existing_server_default=sa.text("'{}'::uuid[]"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -344,6 +390,7 @@ def downgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=True,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "generations",
@@ -351,6 +398,7 @@ def downgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=True,
         existing_server_default=sa.text("'[]'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "credit_transactions",
@@ -358,6 +406,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "credit_transactions",
@@ -365,6 +414,7 @@ def downgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=True,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "boards",
@@ -372,6 +422,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "boards",
@@ -379,6 +430,7 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "boards",
@@ -386,6 +438,7 @@ def downgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=True,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "boards",
@@ -393,6 +446,7 @@ def downgrade() -> None:
         existing_type=postgresql.JSONB(astext_type=sa.Text()),
         nullable=True,
         existing_server_default=sa.text("'{}'::jsonb"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "boards",
@@ -400,6 +454,7 @@ def downgrade() -> None:
         existing_type=sa.BOOLEAN(),
         nullable=True,
         existing_server_default=sa.text("false"),
+        schema=SCHEMA,
     )
     op.alter_column(
         "board_members",
@@ -407,5 +462,6 @@ def downgrade() -> None:
         existing_type=postgresql.TIMESTAMP(timezone=True),
         nullable=True,
         existing_server_default=sa.text("CURRENT_TIMESTAMP"),
+        schema=SCHEMA,
     )
     # ### end Alembic commands ###

package/templates/api/alembic/versions/20251202_000000_add_artifact_lineage.py

@@ -0,0 +1,134 @@
+"""add artifact lineage with input_artifacts
+
+Revision ID: add_artifact_lineage
+Revises: cdad231052d5
+Create Date: 2025-12-02 00:00:00.000000
+
+"""
+
+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 = "add_artifact_lineage"
+down_revision: Union[str, Sequence[str], None] = "cdad231052d5"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+# Schema name for all Boards tables
+SCHEMA = "boards"
+
+
+def upgrade() -> None:
+    """Upgrade schema to use input_artifacts for lineage tracking."""
+    # Add input_artifacts JSONB column
+    op.add_column(
+        "generations",
+        sa.Column(
+            "input_artifacts",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'[]'::jsonb"),
+            nullable=False,
+        ),
+        schema=SCHEMA,
+    )
+
+    # Migrate existing data from input_generation_ids to input_artifacts
+    # For backwards compatibility, use role="input" for legacy data
+    op.execute(f"""
+        UPDATE {SCHEMA}.generations
+        SET input_artifacts = (
+            SELECT COALESCE(jsonb_agg(
+                jsonb_build_object(
+                    'generation_id', gen_id::text,
+                    'role', 'input',
+                    'artifact_type', COALESCE(g.artifact_type, 'unknown')
+                )
+            ), '[]'::jsonb)
+            FROM unnest(input_generation_ids) AS gen_id
+            LEFT JOIN {SCHEMA}.generations g ON g.id = gen_id
+        )
+        WHERE input_generation_ids IS NOT NULL
+        AND array_length(input_generation_ids, 1) > 0
+    """)
+
+    # Add GIN index for JSONB queries
+    op.create_index(
+        "idx_generations_input_artifacts_gin",
+        "generations",
+        ["input_artifacts"],
+        unique=False,
+        postgresql_using="gin",
+        schema=SCHEMA,
+    )
+
+    # Drop old lineage index
+    op.drop_index("idx_generations_lineage", table_name="generations", schema=SCHEMA)
+
+    # Drop old foreign key constraint on parent_generation_id
+    op.drop_constraint(
+        "generations_parent_generation_id_fkey",
+        "generations",
+        type_="foreignkey",
+        schema=SCHEMA,
+    )
+
+    # Drop old columns
+    op.drop_column("generations", "parent_generation_id", schema=SCHEMA)
+    op.drop_column("generations", "input_generation_ids", schema=SCHEMA)
+
+
+def downgrade() -> None:
+    """Downgrade schema back to parent_generation_id and input_generation_ids."""
+    # Add back old columns
+    op.add_column(
+        "generations",
+        sa.Column(
+            "input_generation_ids",
+            postgresql.ARRAY(postgresql.UUID()),
+            server_default=sa.text("'{}'::uuid[]"),
+            autoincrement=False,
+            nullable=False,
+        ),
+        schema=SCHEMA,
+    )
+    op.add_column(
+        "generations",
+        sa.Column("parent_generation_id", postgresql.UUID(), autoincrement=False, nullable=True),
+        schema=SCHEMA,
+    )
+
+    # Migrate data back from input_artifacts to input_generation_ids
+    op.execute(f"""
+        UPDATE {SCHEMA}.generations
+        SET input_generation_ids = (
+            SELECT COALESCE(
+                array_agg((elem->>'generation_id')::uuid),
+                '{{}}'::uuid[]
+            )
+            FROM jsonb_array_elements(input_artifacts) elem
+        )
+        WHERE input_artifacts IS NOT NULL
+        AND jsonb_array_length(input_artifacts) > 0
+    """)
+
+    # Recreate foreign key constraint
+    op.create_foreign_key(
+        "generations_parent_generation_id_fkey",
+        "generations",
+        "generations",
+        ["parent_generation_id"],
+        ["id"],
+        source_schema=SCHEMA,
+        referent_schema=SCHEMA,
+    )
+
+    # Recreate old index
+    op.create_index("idx_generations_lineage", "generations", ["parent_generation_id"], unique=False, schema=SCHEMA)
+
+    # Drop new index and column
+    op.drop_index("idx_generations_input_artifacts_gin", table_name="generations", schema=SCHEMA)
+    op.drop_column("generations", "input_artifacts", schema=SCHEMA)

package/templates/api/alembic/versions/2025925_62735_add_seed_data_for_default_tenant.py

@@ -24,6 +24,9 @@ down_revision: str | Sequence[str] | None = '20250101_000000_initial_schema'
 branch_labels: str | Sequence[str] | None = None
 depends_on: str | Sequence[str] | None = None
 
+# Schema name for all Boards tables
+SCHEMA = "boards"
+
 
 def upgrade() -> None:
     """Upgrade schema and seed default tenant."""
@@ -37,16 +40,16 @@ def upgrade() -> None:
 
     # Check if tenant already exists
     existing_tenant = connection.execute(
-        sa.text("SELECT id FROM tenants WHERE slug = :slug"),
+        sa.text(f"SELECT id FROM {SCHEMA}.tenants WHERE slug = :slug"),
         {"slug": tenant_slug}
     ).fetchone()
 
     if not existing_tenant:
         # Insert the default tenant
         connection.execute(
-            sa.text("""
-                INSERT INTO tenants (name, slug, settings, created_at, updated_at)
-                VALUES (:name, :slug, '{}', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
+            sa.text(f"""
+                INSERT INTO {SCHEMA}.tenants (name, slug, settings, created_at, updated_at)
+                VALUES (:name, :slug, '{{}}'::jsonb, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
             """),
             {
                 "name": tenant_name,
@@ -68,7 +71,7 @@ def downgrade() -> None:
     connection = op.get_bind()
 
     result = connection.execute(
-        sa.text("DELETE FROM tenants WHERE slug = :slug"),
+        sa.text(f"DELETE FROM {SCHEMA}.tenants WHERE slug = :slug"),
         {"slug": tenant_slug}
     )