@weirdfingers/baseboards 0.9.6 → 0.9.7
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/index.js +560 -469
- package/dist/index.js.map +1 -1
- package/package.json +2 -5
- package/templates/README.md +0 -122
- package/templates/api/.env.example +0 -65
- package/templates/api/ARTIFACT_RESOLUTION_GUIDE.md +0 -148
- package/templates/api/Dockerfile +0 -32
- package/templates/api/README.md +0 -264
- package/templates/api/alembic/env.py +0 -114
- package/templates/api/alembic/script.py.mako +0 -28
- package/templates/api/alembic/versions/20250101_000000_initial_schema.py +0 -506
- package/templates/api/alembic/versions/20251022_174729_remove_provider_name_from_generations.py +0 -75
- package/templates/api/alembic/versions/20251023_165852_switch_to_declarative_base_and_mapping.py +0 -467
- package/templates/api/alembic/versions/20251202_000000_add_artifact_lineage.py +0 -134
- package/templates/api/alembic/versions/2025925_62735_add_seed_data_for_default_tenant.py +0 -88
- package/templates/api/alembic.ini +0 -36
- package/templates/api/config/generators.yaml +0 -237
- package/templates/api/config/storage_config.yaml +0 -26
- package/templates/api/docs/ADDING_GENERATORS.md +0 -409
- package/templates/api/docs/GENERATORS_API.md +0 -502
- package/templates/api/docs/MIGRATIONS.md +0 -472
- package/templates/api/docs/TESTING_LIVE_APIS.md +0 -417
- package/templates/api/docs/storage_providers.md +0 -337
- package/templates/api/pyproject.toml +0 -205
- package/templates/api/src/boards/__init__.py +0 -10
- package/templates/api/src/boards/api/app.py +0 -172
- package/templates/api/src/boards/api/auth.py +0 -75
- package/templates/api/src/boards/api/endpoints/__init__.py +0 -3
- package/templates/api/src/boards/api/endpoints/jobs.py +0 -76
- package/templates/api/src/boards/api/endpoints/setup.py +0 -505
- package/templates/api/src/boards/api/endpoints/sse.py +0 -129
- package/templates/api/src/boards/api/endpoints/storage.py +0 -155
- package/templates/api/src/boards/api/endpoints/tenant_registration.py +0 -296
- package/templates/api/src/boards/api/endpoints/uploads.py +0 -149
- package/templates/api/src/boards/api/endpoints/webhooks.py +0 -13
- package/templates/api/src/boards/auth/__init__.py +0 -15
- package/templates/api/src/boards/auth/adapters/__init__.py +0 -27
- package/templates/api/src/boards/auth/adapters/auth0.py +0 -220
- package/templates/api/src/boards/auth/adapters/base.py +0 -73
- package/templates/api/src/boards/auth/adapters/clerk.py +0 -172
- package/templates/api/src/boards/auth/adapters/jwt.py +0 -122
- package/templates/api/src/boards/auth/adapters/none.py +0 -102
- package/templates/api/src/boards/auth/adapters/oidc.py +0 -284
- package/templates/api/src/boards/auth/adapters/supabase.py +0 -110
- package/templates/api/src/boards/auth/context.py +0 -35
- package/templates/api/src/boards/auth/factory.py +0 -129
- package/templates/api/src/boards/auth/middleware.py +0 -221
- package/templates/api/src/boards/auth/provisioning.py +0 -129
- package/templates/api/src/boards/auth/tenant_extraction.py +0 -278
- package/templates/api/src/boards/cli.py +0 -354
- package/templates/api/src/boards/config.py +0 -131
- package/templates/api/src/boards/database/__init__.py +0 -7
- package/templates/api/src/boards/database/cli.py +0 -110
- package/templates/api/src/boards/database/connection.py +0 -292
- package/templates/api/src/boards/database/models.py +0 -19
- package/templates/api/src/boards/database/seed_data.py +0 -182
- package/templates/api/src/boards/dbmodels/__init__.py +0 -441
- package/templates/api/src/boards/generators/__init__.py +0 -57
- package/templates/api/src/boards/generators/artifact_resolution.py +0 -405
- package/templates/api/src/boards/generators/artifacts.py +0 -53
- package/templates/api/src/boards/generators/base.py +0 -144
- package/templates/api/src/boards/generators/implementations/__init__.py +0 -14
- package/templates/api/src/boards/generators/implementations/fal/__init__.py +0 -25
- package/templates/api/src/boards/generators/implementations/fal/audio/__init__.py +0 -23
- package/templates/api/src/boards/generators/implementations/fal/audio/beatoven_music_generation.py +0 -171
- package/templates/api/src/boards/generators/implementations/fal/audio/beatoven_sound_effect_generation.py +0 -167
- package/templates/api/src/boards/generators/implementations/fal/audio/chatterbox_text_to_speech.py +0 -176
- package/templates/api/src/boards/generators/implementations/fal/audio/chatterbox_tts_turbo.py +0 -195
- package/templates/api/src/boards/generators/implementations/fal/audio/elevenlabs_sound_effects_v2.py +0 -194
- package/templates/api/src/boards/generators/implementations/fal/audio/elevenlabs_tts_eleven_v3.py +0 -209
- package/templates/api/src/boards/generators/implementations/fal/audio/fal_elevenlabs_tts_turbo_v2_5.py +0 -206
- package/templates/api/src/boards/generators/implementations/fal/audio/fal_minimax_speech_26_hd.py +0 -237
- package/templates/api/src/boards/generators/implementations/fal/audio/minimax_music_v2.py +0 -173
- package/templates/api/src/boards/generators/implementations/fal/audio/minimax_speech_2_6_turbo.py +0 -221
- package/templates/api/src/boards/generators/implementations/fal/image/__init__.py +0 -63
- package/templates/api/src/boards/generators/implementations/fal/image/bytedance_seedream_v45_edit.py +0 -219
- package/templates/api/src/boards/generators/implementations/fal/image/clarity_upscaler.py +0 -220
- package/templates/api/src/boards/generators/implementations/fal/image/crystal_upscaler.py +0 -173
- package/templates/api/src/boards/generators/implementations/fal/image/fal_ideogram_character.py +0 -227
- package/templates/api/src/boards/generators/implementations/fal/image/flux_2.py +0 -203
- package/templates/api/src/boards/generators/implementations/fal/image/flux_2_edit.py +0 -230
- package/templates/api/src/boards/generators/implementations/fal/image/flux_2_pro.py +0 -204
- package/templates/api/src/boards/generators/implementations/fal/image/flux_2_pro_edit.py +0 -221
- package/templates/api/src/boards/generators/implementations/fal/image/flux_pro_kontext.py +0 -216
- package/templates/api/src/boards/generators/implementations/fal/image/flux_pro_ultra.py +0 -197
- package/templates/api/src/boards/generators/implementations/fal/image/gemini_25_flash_image.py +0 -177
- package/templates/api/src/boards/generators/implementations/fal/image/gemini_25_flash_image_edit.py +0 -208
- package/templates/api/src/boards/generators/implementations/fal/image/gpt_image_15_edit.py +0 -216
- package/templates/api/src/boards/generators/implementations/fal/image/gpt_image_1_5.py +0 -177
- package/templates/api/src/boards/generators/implementations/fal/image/gpt_image_1_edit_image.py +0 -182
- package/templates/api/src/boards/generators/implementations/fal/image/gpt_image_1_mini.py +0 -167
- package/templates/api/src/boards/generators/implementations/fal/image/ideogram_character_edit.py +0 -299
- package/templates/api/src/boards/generators/implementations/fal/image/ideogram_v2.py +0 -190
- package/templates/api/src/boards/generators/implementations/fal/image/imagen4_preview.py +0 -191
- package/templates/api/src/boards/generators/implementations/fal/image/imagen4_preview_fast.py +0 -179
- package/templates/api/src/boards/generators/implementations/fal/image/nano_banana.py +0 -183
- package/templates/api/src/boards/generators/implementations/fal/image/nano_banana_edit.py +0 -212
- package/templates/api/src/boards/generators/implementations/fal/image/nano_banana_pro.py +0 -179
- package/templates/api/src/boards/generators/implementations/fal/image/nano_banana_pro_edit.py +0 -226
- package/templates/api/src/boards/generators/implementations/fal/image/qwen_image.py +0 -249
- package/templates/api/src/boards/generators/implementations/fal/image/qwen_image_edit.py +0 -244
- package/templates/api/src/boards/generators/implementations/fal/image/reve_edit.py +0 -178
- package/templates/api/src/boards/generators/implementations/fal/image/reve_text_to_image.py +0 -155
- package/templates/api/src/boards/generators/implementations/fal/image/seedream_v45_text_to_image.py +0 -180
- package/templates/api/src/boards/generators/implementations/fal/utils.py +0 -61
- package/templates/api/src/boards/generators/implementations/fal/video/__init__.py +0 -77
- package/templates/api/src/boards/generators/implementations/fal/video/bytedance_seedance_v1_pro_text_to_video.py +0 -209
- package/templates/api/src/boards/generators/implementations/fal/video/creatify_lipsync.py +0 -161
- package/templates/api/src/boards/generators/implementations/fal/video/fal_bytedance_seedance_v1_pro_image_to_video.py +0 -222
- package/templates/api/src/boards/generators/implementations/fal/video/fal_minimax_hailuo_02_standard_text_to_video.py +0 -152
- package/templates/api/src/boards/generators/implementations/fal/video/fal_pixverse_lipsync.py +0 -197
- package/templates/api/src/boards/generators/implementations/fal/video/fal_sora_2_text_to_video.py +0 -173
- package/templates/api/src/boards/generators/implementations/fal/video/infinitalk.py +0 -221
- package/templates/api/src/boards/generators/implementations/fal/video/kling_video_ai_avatar_v2_pro.py +0 -168
- package/templates/api/src/boards/generators/implementations/fal/video/kling_video_ai_avatar_v2_standard.py +0 -159
- package/templates/api/src/boards/generators/implementations/fal/video/kling_video_v2_5_turbo_pro_image_to_video.py +0 -175
- package/templates/api/src/boards/generators/implementations/fal/video/kling_video_v2_5_turbo_pro_text_to_video.py +0 -168
- package/templates/api/src/boards/generators/implementations/fal/video/minimax_hailuo_2_3_pro_image_to_video.py +0 -153
- package/templates/api/src/boards/generators/implementations/fal/video/sora2_image_to_video.py +0 -172
- package/templates/api/src/boards/generators/implementations/fal/video/sora_2_image_to_video_pro.py +0 -175
- package/templates/api/src/boards/generators/implementations/fal/video/sora_2_text_to_video_pro.py +0 -163
- package/templates/api/src/boards/generators/implementations/fal/video/sync_lipsync_v2.py +0 -167
- package/templates/api/src/boards/generators/implementations/fal/video/sync_lipsync_v2_pro.py +0 -155
- package/templates/api/src/boards/generators/implementations/fal/video/veed_fabric_1_0.py +0 -180
- package/templates/api/src/boards/generators/implementations/fal/video/veed_lipsync.py +0 -174
- package/templates/api/src/boards/generators/implementations/fal/video/veo3.py +0 -194
- package/templates/api/src/boards/generators/implementations/fal/video/veo31.py +0 -190
- package/templates/api/src/boards/generators/implementations/fal/video/veo31_fast.py +0 -190
- package/templates/api/src/boards/generators/implementations/fal/video/veo31_fast_image_to_video.py +0 -191
- package/templates/api/src/boards/generators/implementations/fal/video/veo31_first_last_frame_to_video.py +0 -187
- package/templates/api/src/boards/generators/implementations/fal/video/veo31_image_to_video.py +0 -183
- package/templates/api/src/boards/generators/implementations/fal/video/veo31_reference_to_video.py +0 -172
- package/templates/api/src/boards/generators/implementations/fal/video/wan_25_preview_image_to_video.py +0 -212
- package/templates/api/src/boards/generators/implementations/fal/video/wan_25_preview_text_to_video.py +0 -208
- package/templates/api/src/boards/generators/implementations/fal/video/wan_pro_image_to_video.py +0 -158
- package/templates/api/src/boards/generators/implementations/kie/__init__.py +0 -11
- package/templates/api/src/boards/generators/implementations/kie/base.py +0 -316
- package/templates/api/src/boards/generators/implementations/kie/image/__init__.py +0 -3
- package/templates/api/src/boards/generators/implementations/kie/image/nano_banana_edit.py +0 -190
- package/templates/api/src/boards/generators/implementations/kie/utils.py +0 -98
- package/templates/api/src/boards/generators/implementations/kie/video/__init__.py +0 -8
- package/templates/api/src/boards/generators/implementations/kie/video/veo3.py +0 -161
- package/templates/api/src/boards/generators/implementations/openai/__init__.py +0 -1
- package/templates/api/src/boards/generators/implementations/openai/audio/__init__.py +0 -1
- package/templates/api/src/boards/generators/implementations/openai/audio/whisper.py +0 -69
- package/templates/api/src/boards/generators/implementations/openai/image/__init__.py +0 -1
- package/templates/api/src/boards/generators/implementations/openai/image/dalle3.py +0 -96
- package/templates/api/src/boards/generators/implementations/replicate/__init__.py +0 -1
- package/templates/api/src/boards/generators/implementations/replicate/image/__init__.py +0 -1
- package/templates/api/src/boards/generators/implementations/replicate/image/flux_pro.py +0 -88
- package/templates/api/src/boards/generators/implementations/replicate/video/__init__.py +0 -1
- package/templates/api/src/boards/generators/implementations/replicate/video/lipsync.py +0 -73
- package/templates/api/src/boards/generators/loader.py +0 -253
- package/templates/api/src/boards/generators/registry.py +0 -114
- package/templates/api/src/boards/generators/resolution.py +0 -632
- package/templates/api/src/boards/generators/testmods/class_gen.py +0 -34
- package/templates/api/src/boards/generators/testmods/import_side_effect.py +0 -35
- package/templates/api/src/boards/graphql/__init__.py +0 -7
- package/templates/api/src/boards/graphql/access_control.py +0 -136
- package/templates/api/src/boards/graphql/mutations/root.py +0 -148
- package/templates/api/src/boards/graphql/queries/root.py +0 -116
- package/templates/api/src/boards/graphql/resolvers/__init__.py +0 -8
- package/templates/api/src/boards/graphql/resolvers/auth.py +0 -12
- package/templates/api/src/boards/graphql/resolvers/board.py +0 -1053
- package/templates/api/src/boards/graphql/resolvers/generation.py +0 -666
- package/templates/api/src/boards/graphql/resolvers/generator.py +0 -50
- package/templates/api/src/boards/graphql/resolvers/lineage.py +0 -381
- package/templates/api/src/boards/graphql/resolvers/upload.py +0 -463
- package/templates/api/src/boards/graphql/resolvers/user.py +0 -25
- package/templates/api/src/boards/graphql/schema.py +0 -81
- package/templates/api/src/boards/graphql/types/board.py +0 -102
- package/templates/api/src/boards/graphql/types/generation.py +0 -166
- package/templates/api/src/boards/graphql/types/generator.py +0 -17
- package/templates/api/src/boards/graphql/types/user.py +0 -47
- package/templates/api/src/boards/jobs/repository.py +0 -153
- package/templates/api/src/boards/logging.py +0 -195
- package/templates/api/src/boards/middleware.py +0 -339
- package/templates/api/src/boards/progress/__init__.py +0 -4
- package/templates/api/src/boards/progress/models.py +0 -25
- package/templates/api/src/boards/progress/publisher.py +0 -64
- package/templates/api/src/boards/py.typed +0 -0
- package/templates/api/src/boards/redis_pool.py +0 -118
- package/templates/api/src/boards/storage/__init__.py +0 -52
- package/templates/api/src/boards/storage/base.py +0 -363
- package/templates/api/src/boards/storage/config.py +0 -187
- package/templates/api/src/boards/storage/factory.py +0 -288
- package/templates/api/src/boards/storage/implementations/__init__.py +0 -27
- package/templates/api/src/boards/storage/implementations/gcs.py +0 -340
- package/templates/api/src/boards/storage/implementations/local.py +0 -201
- package/templates/api/src/boards/storage/implementations/s3.py +0 -294
- package/templates/api/src/boards/storage/implementations/supabase.py +0 -218
- package/templates/api/src/boards/tenant_isolation.py +0 -446
- package/templates/api/src/boards/validation.py +0 -262
- package/templates/api/src/boards/workers/__init__.py +0 -1
- package/templates/api/src/boards/workers/actors.py +0 -274
- package/templates/api/src/boards/workers/cli.py +0 -125
- package/templates/api/src/boards/workers/context.py +0 -348
- package/templates/api/src/boards/workers/middleware.py +0 -58
- package/templates/api/src/py.typed +0 -0
- package/templates/compose.web.yaml +0 -35
- package/templates/compose.yaml +0 -116
- package/templates/docker/env.example +0 -23
- package/templates/web/.env.example +0 -28
- package/templates/web/Dockerfile +0 -51
- package/templates/web/components.json +0 -22
- package/templates/web/imageLoader.js +0 -18
- package/templates/web/next-env.d.ts +0 -5
- package/templates/web/next.config.js +0 -36
- package/templates/web/package.json +0 -41
- package/templates/web/postcss.config.mjs +0 -7
- package/templates/web/public/favicon.ico +0 -0
- package/templates/web/src/app/boards/[boardId]/page.tsx +0 -353
- package/templates/web/src/app/globals.css +0 -123
- package/templates/web/src/app/layout.tsx +0 -31
- package/templates/web/src/app/lineage/[generationId]/page.tsx +0 -235
- package/templates/web/src/app/page.tsx +0 -35
- package/templates/web/src/app/providers.tsx +0 -18
- package/templates/web/src/components/boards/ArtifactInputSlots.tsx +0 -206
- package/templates/web/src/components/boards/ArtifactPreview.tsx +0 -466
- package/templates/web/src/components/boards/GenerationGrid.tsx +0 -282
- package/templates/web/src/components/boards/GenerationInput.tsx +0 -370
- package/templates/web/src/components/boards/GeneratorSelector.tsx +0 -272
- package/templates/web/src/components/boards/UploadArtifact.tsx +0 -563
- package/templates/web/src/components/header.tsx +0 -32
- package/templates/web/src/components/theme-provider.tsx +0 -10
- package/templates/web/src/components/theme-toggle.tsx +0 -75
- package/templates/web/src/components/ui/alert-dialog.tsx +0 -157
- package/templates/web/src/components/ui/button.tsx +0 -58
- package/templates/web/src/components/ui/card.tsx +0 -92
- package/templates/web/src/components/ui/dropdown-menu.tsx +0 -200
- package/templates/web/src/components/ui/navigation-menu.tsx +0 -168
- package/templates/web/src/components/ui/toast.tsx +0 -128
- package/templates/web/src/components/ui/toaster.tsx +0 -35
- package/templates/web/src/components/ui/use-toast.ts +0 -187
- package/templates/web/src/hooks/useGeneratorMRU.ts +0 -57
- package/templates/web/src/lib/utils.ts +0 -6
- package/templates/web/tsconfig.json +0 -41
package/templates/api/alembic/versions/20251022_174729_remove_provider_name_from_generations.py
DELETED
|
@@ -1,75 +0,0 @@
|
|
|
1
|
-
"""remove provider_name from generations
|
|
2
|
-
|
|
3
|
-
Revision ID: d7c16af77c9e
|
|
4
|
-
Revises: 553dc6a50a20
|
|
5
|
-
Create Date: 2025-10-22 17:47:29.312285
|
|
6
|
-
|
|
7
|
-
"""
|
|
8
|
-
|
|
9
|
-
from typing import Sequence, Union
|
|
10
|
-
|
|
11
|
-
from alembic import op
|
|
12
|
-
import sqlalchemy as sa
|
|
13
|
-
from sqlalchemy.dialects import postgresql
|
|
14
|
-
|
|
15
|
-
# revision identifiers, used by Alembic.
|
|
16
|
-
revision: str = "d7c16af77c9e"
|
|
17
|
-
down_revision: Union[str, Sequence[str], None] = "553dc6a50a20"
|
|
18
|
-
branch_labels: Union[str, Sequence[str], None] = None
|
|
19
|
-
depends_on: Union[str, Sequence[str], None] = None
|
|
20
|
-
|
|
21
|
-
# Schema name for all Boards tables
|
|
22
|
-
SCHEMA = "boards"
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
def upgrade() -> None:
|
|
26
|
-
"""Upgrade schema."""
|
|
27
|
-
# ### commands auto generated by Alembic - please adjust! ###
|
|
28
|
-
# Drop schema_migrations table if it exists (from legacy migration system)
|
|
29
|
-
op.execute("DROP TABLE IF EXISTS schema_migrations")
|
|
30
|
-
op.drop_column("generations", "provider_name", schema=SCHEMA)
|
|
31
|
-
# ### end Alembic commands ###
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
def downgrade() -> None:
|
|
35
|
-
"""Downgrade schema."""
|
|
36
|
-
# ### commands auto generated by Alembic - please adjust! ###
|
|
37
|
-
# Add column as nullable first, then update existing rows, then make it NOT NULL
|
|
38
|
-
op.add_column(
|
|
39
|
-
"generations",
|
|
40
|
-
sa.Column(
|
|
41
|
-
"provider_name", sa.VARCHAR(length=100), autoincrement=False, nullable=True
|
|
42
|
-
),
|
|
43
|
-
schema=SCHEMA,
|
|
44
|
-
)
|
|
45
|
-
# Set a default value for existing rows (though this downgrade should rarely be used)
|
|
46
|
-
op.execute(
|
|
47
|
-
f"UPDATE {SCHEMA}.generations SET provider_name = 'unknown' WHERE provider_name IS NULL"
|
|
48
|
-
)
|
|
49
|
-
# Now make it NOT NULL
|
|
50
|
-
op.alter_column("generations", "provider_name", nullable=False, schema=SCHEMA)
|
|
51
|
-
op.create_table(
|
|
52
|
-
"schema_migrations",
|
|
53
|
-
sa.Column(
|
|
54
|
-
"version", sa.VARCHAR(length=255), autoincrement=False, nullable=False
|
|
55
|
-
),
|
|
56
|
-
sa.Column("name", sa.VARCHAR(length=255), autoincrement=False, nullable=False),
|
|
57
|
-
sa.Column(
|
|
58
|
-
"applied_at",
|
|
59
|
-
postgresql.TIMESTAMP(timezone=True),
|
|
60
|
-
server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
61
|
-
autoincrement=False,
|
|
62
|
-
nullable=True,
|
|
63
|
-
),
|
|
64
|
-
sa.Column(
|
|
65
|
-
"execution_time_ms", sa.INTEGER(), autoincrement=False, nullable=True
|
|
66
|
-
),
|
|
67
|
-
sa.Column(
|
|
68
|
-
"checksum", sa.VARCHAR(length=64), autoincrement=False, nullable=True
|
|
69
|
-
),
|
|
70
|
-
sa.Column(
|
|
71
|
-
"applied_by", sa.VARCHAR(length=255), autoincrement=False, nullable=True
|
|
72
|
-
),
|
|
73
|
-
sa.PrimaryKeyConstraint("version", name=op.f("schema_migrations_pkey")),
|
|
74
|
-
)
|
|
75
|
-
# ### end Alembic commands ###
|
package/templates/api/alembic/versions/20251023_165852_switch_to_declarative_base_and_mapping.py
DELETED
|
@@ -1,467 +0,0 @@
|
|
|
1
|
-
"""switch to declarative base and mapping
|
|
2
|
-
|
|
3
|
-
Revision ID: cdad231052d5
|
|
4
|
-
Revises: d7c16af77c9e
|
|
5
|
-
Create Date: 2025-10-23 16:58:52.004927
|
|
6
|
-
|
|
7
|
-
"""
|
|
8
|
-
|
|
9
|
-
from collections.abc import Sequence
|
|
10
|
-
|
|
11
|
-
import sqlalchemy as sa
|
|
12
|
-
from sqlalchemy.dialects import postgresql
|
|
13
|
-
|
|
14
|
-
from alembic import op
|
|
15
|
-
|
|
16
|
-
# revision identifiers, used by Alembic.
|
|
17
|
-
revision: str = "cdad231052d5"
|
|
18
|
-
down_revision: str | Sequence[str] | None = "d7c16af77c9e"
|
|
19
|
-
branch_labels: str | Sequence[str] | None = None
|
|
20
|
-
depends_on: str | Sequence[str] | None = None
|
|
21
|
-
|
|
22
|
-
# Schema name for all Boards tables
|
|
23
|
-
SCHEMA = "boards"
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
def upgrade() -> None:
|
|
27
|
-
"""Upgrade schema."""
|
|
28
|
-
# ### commands auto generated by Alembic - please adjust! ###
|
|
29
|
-
op.alter_column(
|
|
30
|
-
"board_members",
|
|
31
|
-
"joined_at",
|
|
32
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
33
|
-
nullable=False,
|
|
34
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
35
|
-
schema=SCHEMA,
|
|
36
|
-
)
|
|
37
|
-
op.alter_column(
|
|
38
|
-
"boards",
|
|
39
|
-
"is_public",
|
|
40
|
-
existing_type=sa.BOOLEAN(),
|
|
41
|
-
nullable=False,
|
|
42
|
-
existing_server_default=sa.text("false"),
|
|
43
|
-
schema=SCHEMA,
|
|
44
|
-
)
|
|
45
|
-
op.alter_column(
|
|
46
|
-
"boards",
|
|
47
|
-
"settings",
|
|
48
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
49
|
-
nullable=False,
|
|
50
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
51
|
-
schema=SCHEMA,
|
|
52
|
-
)
|
|
53
|
-
op.alter_column(
|
|
54
|
-
"boards",
|
|
55
|
-
"metadata",
|
|
56
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
57
|
-
nullable=False,
|
|
58
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
59
|
-
schema=SCHEMA,
|
|
60
|
-
)
|
|
61
|
-
op.alter_column(
|
|
62
|
-
"boards",
|
|
63
|
-
"created_at",
|
|
64
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
65
|
-
nullable=False,
|
|
66
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
67
|
-
schema=SCHEMA,
|
|
68
|
-
)
|
|
69
|
-
op.alter_column(
|
|
70
|
-
"boards",
|
|
71
|
-
"updated_at",
|
|
72
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
73
|
-
nullable=False,
|
|
74
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
75
|
-
schema=SCHEMA,
|
|
76
|
-
)
|
|
77
|
-
op.alter_column(
|
|
78
|
-
"credit_transactions",
|
|
79
|
-
"metadata",
|
|
80
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
81
|
-
nullable=False,
|
|
82
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
83
|
-
schema=SCHEMA,
|
|
84
|
-
)
|
|
85
|
-
op.alter_column(
|
|
86
|
-
"credit_transactions",
|
|
87
|
-
"created_at",
|
|
88
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
89
|
-
nullable=False,
|
|
90
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
91
|
-
schema=SCHEMA,
|
|
92
|
-
)
|
|
93
|
-
op.alter_column(
|
|
94
|
-
"generations",
|
|
95
|
-
"additional_files",
|
|
96
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
97
|
-
nullable=False,
|
|
98
|
-
existing_server_default=sa.text("'[]'::jsonb"),
|
|
99
|
-
schema=SCHEMA,
|
|
100
|
-
)
|
|
101
|
-
op.alter_column(
|
|
102
|
-
"generations",
|
|
103
|
-
"output_metadata",
|
|
104
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
105
|
-
nullable=False,
|
|
106
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
107
|
-
schema=SCHEMA,
|
|
108
|
-
)
|
|
109
|
-
op.alter_column(
|
|
110
|
-
"generations",
|
|
111
|
-
"input_generation_ids",
|
|
112
|
-
existing_type=postgresql.ARRAY(sa.UUID()),
|
|
113
|
-
nullable=False,
|
|
114
|
-
existing_server_default=sa.text("'{}'::uuid[]"),
|
|
115
|
-
schema=SCHEMA,
|
|
116
|
-
)
|
|
117
|
-
op.alter_column(
|
|
118
|
-
"generations",
|
|
119
|
-
"progress",
|
|
120
|
-
existing_type=sa.NUMERIC(precision=5, scale=2),
|
|
121
|
-
nullable=False,
|
|
122
|
-
existing_server_default=sa.text("0.0"),
|
|
123
|
-
schema=SCHEMA,
|
|
124
|
-
)
|
|
125
|
-
op.alter_column(
|
|
126
|
-
"generations",
|
|
127
|
-
"created_at",
|
|
128
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
129
|
-
nullable=False,
|
|
130
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
131
|
-
schema=SCHEMA,
|
|
132
|
-
)
|
|
133
|
-
op.alter_column(
|
|
134
|
-
"generations",
|
|
135
|
-
"updated_at",
|
|
136
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
137
|
-
nullable=False,
|
|
138
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
139
|
-
schema=SCHEMA,
|
|
140
|
-
)
|
|
141
|
-
op.alter_column(
|
|
142
|
-
"lora_models",
|
|
143
|
-
"metadata",
|
|
144
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
145
|
-
nullable=False,
|
|
146
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
147
|
-
schema=SCHEMA,
|
|
148
|
-
)
|
|
149
|
-
op.alter_column(
|
|
150
|
-
"lora_models",
|
|
151
|
-
"is_public",
|
|
152
|
-
existing_type=sa.BOOLEAN(),
|
|
153
|
-
nullable=False,
|
|
154
|
-
existing_server_default=sa.text("false"),
|
|
155
|
-
schema=SCHEMA,
|
|
156
|
-
)
|
|
157
|
-
op.alter_column(
|
|
158
|
-
"lora_models",
|
|
159
|
-
"created_at",
|
|
160
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
161
|
-
nullable=False,
|
|
162
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
163
|
-
schema=SCHEMA,
|
|
164
|
-
)
|
|
165
|
-
op.alter_column(
|
|
166
|
-
"lora_models",
|
|
167
|
-
"updated_at",
|
|
168
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
169
|
-
nullable=False,
|
|
170
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
171
|
-
schema=SCHEMA,
|
|
172
|
-
)
|
|
173
|
-
op.alter_column(
|
|
174
|
-
"provider_configs",
|
|
175
|
-
"is_enabled",
|
|
176
|
-
existing_type=sa.BOOLEAN(),
|
|
177
|
-
nullable=False,
|
|
178
|
-
existing_server_default=sa.text("true"),
|
|
179
|
-
schema=SCHEMA,
|
|
180
|
-
)
|
|
181
|
-
op.alter_column(
|
|
182
|
-
"provider_configs",
|
|
183
|
-
"created_at",
|
|
184
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
185
|
-
nullable=False,
|
|
186
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
187
|
-
schema=SCHEMA,
|
|
188
|
-
)
|
|
189
|
-
op.alter_column(
|
|
190
|
-
"provider_configs",
|
|
191
|
-
"updated_at",
|
|
192
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
193
|
-
nullable=False,
|
|
194
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
195
|
-
schema=SCHEMA,
|
|
196
|
-
)
|
|
197
|
-
op.alter_column(
|
|
198
|
-
"tenants",
|
|
199
|
-
"settings",
|
|
200
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
201
|
-
nullable=False,
|
|
202
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
203
|
-
schema=SCHEMA,
|
|
204
|
-
)
|
|
205
|
-
op.alter_column(
|
|
206
|
-
"tenants",
|
|
207
|
-
"created_at",
|
|
208
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
209
|
-
nullable=False,
|
|
210
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
211
|
-
schema=SCHEMA,
|
|
212
|
-
)
|
|
213
|
-
op.alter_column(
|
|
214
|
-
"tenants",
|
|
215
|
-
"updated_at",
|
|
216
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
217
|
-
nullable=False,
|
|
218
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
219
|
-
schema=SCHEMA,
|
|
220
|
-
)
|
|
221
|
-
op.alter_column(
|
|
222
|
-
"users",
|
|
223
|
-
"metadata",
|
|
224
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
225
|
-
nullable=False,
|
|
226
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
227
|
-
schema=SCHEMA,
|
|
228
|
-
)
|
|
229
|
-
op.alter_column(
|
|
230
|
-
"users",
|
|
231
|
-
"created_at",
|
|
232
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
233
|
-
nullable=False,
|
|
234
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
235
|
-
schema=SCHEMA,
|
|
236
|
-
)
|
|
237
|
-
op.alter_column(
|
|
238
|
-
"users",
|
|
239
|
-
"updated_at",
|
|
240
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
241
|
-
nullable=False,
|
|
242
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
243
|
-
schema=SCHEMA,
|
|
244
|
-
)
|
|
245
|
-
# ### end Alembic commands ###
|
|
246
|
-
|
|
247
|
-
|
|
248
|
-
def downgrade() -> None:
|
|
249
|
-
"""Downgrade schema."""
|
|
250
|
-
# ### commands auto generated by Alembic - please adjust! ###
|
|
251
|
-
op.alter_column(
|
|
252
|
-
"users",
|
|
253
|
-
"updated_at",
|
|
254
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
255
|
-
nullable=True,
|
|
256
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
257
|
-
schema=SCHEMA,
|
|
258
|
-
)
|
|
259
|
-
op.alter_column(
|
|
260
|
-
"users",
|
|
261
|
-
"created_at",
|
|
262
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
263
|
-
nullable=True,
|
|
264
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
265
|
-
schema=SCHEMA,
|
|
266
|
-
)
|
|
267
|
-
op.alter_column(
|
|
268
|
-
"users",
|
|
269
|
-
"metadata",
|
|
270
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
271
|
-
nullable=True,
|
|
272
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
273
|
-
schema=SCHEMA,
|
|
274
|
-
)
|
|
275
|
-
op.alter_column(
|
|
276
|
-
"tenants",
|
|
277
|
-
"updated_at",
|
|
278
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
279
|
-
nullable=True,
|
|
280
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
281
|
-
schema=SCHEMA,
|
|
282
|
-
)
|
|
283
|
-
op.alter_column(
|
|
284
|
-
"tenants",
|
|
285
|
-
"created_at",
|
|
286
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
287
|
-
nullable=True,
|
|
288
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
289
|
-
schema=SCHEMA,
|
|
290
|
-
)
|
|
291
|
-
op.alter_column(
|
|
292
|
-
"tenants",
|
|
293
|
-
"settings",
|
|
294
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
295
|
-
nullable=True,
|
|
296
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
297
|
-
schema=SCHEMA,
|
|
298
|
-
)
|
|
299
|
-
op.alter_column(
|
|
300
|
-
"provider_configs",
|
|
301
|
-
"updated_at",
|
|
302
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
303
|
-
nullable=True,
|
|
304
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
305
|
-
schema=SCHEMA,
|
|
306
|
-
)
|
|
307
|
-
op.alter_column(
|
|
308
|
-
"provider_configs",
|
|
309
|
-
"created_at",
|
|
310
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
311
|
-
nullable=True,
|
|
312
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
313
|
-
schema=SCHEMA,
|
|
314
|
-
)
|
|
315
|
-
op.alter_column(
|
|
316
|
-
"provider_configs",
|
|
317
|
-
"is_enabled",
|
|
318
|
-
existing_type=sa.BOOLEAN(),
|
|
319
|
-
nullable=True,
|
|
320
|
-
existing_server_default=sa.text("true"),
|
|
321
|
-
schema=SCHEMA,
|
|
322
|
-
)
|
|
323
|
-
op.alter_column(
|
|
324
|
-
"lora_models",
|
|
325
|
-
"updated_at",
|
|
326
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
327
|
-
nullable=True,
|
|
328
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
329
|
-
schema=SCHEMA,
|
|
330
|
-
)
|
|
331
|
-
op.alter_column(
|
|
332
|
-
"lora_models",
|
|
333
|
-
"created_at",
|
|
334
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
335
|
-
nullable=True,
|
|
336
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
337
|
-
schema=SCHEMA,
|
|
338
|
-
)
|
|
339
|
-
op.alter_column(
|
|
340
|
-
"lora_models",
|
|
341
|
-
"is_public",
|
|
342
|
-
existing_type=sa.BOOLEAN(),
|
|
343
|
-
nullable=True,
|
|
344
|
-
existing_server_default=sa.text("false"),
|
|
345
|
-
schema=SCHEMA,
|
|
346
|
-
)
|
|
347
|
-
op.alter_column(
|
|
348
|
-
"lora_models",
|
|
349
|
-
"metadata",
|
|
350
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
351
|
-
nullable=True,
|
|
352
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
353
|
-
schema=SCHEMA,
|
|
354
|
-
)
|
|
355
|
-
op.alter_column(
|
|
356
|
-
"generations",
|
|
357
|
-
"updated_at",
|
|
358
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
359
|
-
nullable=True,
|
|
360
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
361
|
-
schema=SCHEMA,
|
|
362
|
-
)
|
|
363
|
-
op.alter_column(
|
|
364
|
-
"generations",
|
|
365
|
-
"created_at",
|
|
366
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
367
|
-
nullable=True,
|
|
368
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
369
|
-
schema=SCHEMA,
|
|
370
|
-
)
|
|
371
|
-
op.alter_column(
|
|
372
|
-
"generations",
|
|
373
|
-
"progress",
|
|
374
|
-
existing_type=sa.NUMERIC(precision=5, scale=2),
|
|
375
|
-
nullable=True,
|
|
376
|
-
existing_server_default=sa.text("0.0"),
|
|
377
|
-
schema=SCHEMA,
|
|
378
|
-
)
|
|
379
|
-
op.alter_column(
|
|
380
|
-
"generations",
|
|
381
|
-
"input_generation_ids",
|
|
382
|
-
existing_type=postgresql.ARRAY(sa.UUID()),
|
|
383
|
-
nullable=True,
|
|
384
|
-
existing_server_default=sa.text("'{}'::uuid[]"),
|
|
385
|
-
schema=SCHEMA,
|
|
386
|
-
)
|
|
387
|
-
op.alter_column(
|
|
388
|
-
"generations",
|
|
389
|
-
"output_metadata",
|
|
390
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
391
|
-
nullable=True,
|
|
392
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
393
|
-
schema=SCHEMA,
|
|
394
|
-
)
|
|
395
|
-
op.alter_column(
|
|
396
|
-
"generations",
|
|
397
|
-
"additional_files",
|
|
398
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
399
|
-
nullable=True,
|
|
400
|
-
existing_server_default=sa.text("'[]'::jsonb"),
|
|
401
|
-
schema=SCHEMA,
|
|
402
|
-
)
|
|
403
|
-
op.alter_column(
|
|
404
|
-
"credit_transactions",
|
|
405
|
-
"created_at",
|
|
406
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
407
|
-
nullable=True,
|
|
408
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
409
|
-
schema=SCHEMA,
|
|
410
|
-
)
|
|
411
|
-
op.alter_column(
|
|
412
|
-
"credit_transactions",
|
|
413
|
-
"metadata",
|
|
414
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
415
|
-
nullable=True,
|
|
416
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
417
|
-
schema=SCHEMA,
|
|
418
|
-
)
|
|
419
|
-
op.alter_column(
|
|
420
|
-
"boards",
|
|
421
|
-
"updated_at",
|
|
422
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
423
|
-
nullable=True,
|
|
424
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
425
|
-
schema=SCHEMA,
|
|
426
|
-
)
|
|
427
|
-
op.alter_column(
|
|
428
|
-
"boards",
|
|
429
|
-
"created_at",
|
|
430
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
431
|
-
nullable=True,
|
|
432
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
433
|
-
schema=SCHEMA,
|
|
434
|
-
)
|
|
435
|
-
op.alter_column(
|
|
436
|
-
"boards",
|
|
437
|
-
"metadata",
|
|
438
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
439
|
-
nullable=True,
|
|
440
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
441
|
-
schema=SCHEMA,
|
|
442
|
-
)
|
|
443
|
-
op.alter_column(
|
|
444
|
-
"boards",
|
|
445
|
-
"settings",
|
|
446
|
-
existing_type=postgresql.JSONB(astext_type=sa.Text()),
|
|
447
|
-
nullable=True,
|
|
448
|
-
existing_server_default=sa.text("'{}'::jsonb"),
|
|
449
|
-
schema=SCHEMA,
|
|
450
|
-
)
|
|
451
|
-
op.alter_column(
|
|
452
|
-
"boards",
|
|
453
|
-
"is_public",
|
|
454
|
-
existing_type=sa.BOOLEAN(),
|
|
455
|
-
nullable=True,
|
|
456
|
-
existing_server_default=sa.text("false"),
|
|
457
|
-
schema=SCHEMA,
|
|
458
|
-
)
|
|
459
|
-
op.alter_column(
|
|
460
|
-
"board_members",
|
|
461
|
-
"joined_at",
|
|
462
|
-
existing_type=postgresql.TIMESTAMP(timezone=True),
|
|
463
|
-
nullable=True,
|
|
464
|
-
existing_server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
465
|
-
schema=SCHEMA,
|
|
466
|
-
)
|
|
467
|
-
# ### end Alembic commands ###
|
|
@@ -1,134 +0,0 @@
|
|
|
1
|
-
"""add artifact lineage with input_artifacts
|
|
2
|
-
|
|
3
|
-
Revision ID: add_artifact_lineage
|
|
4
|
-
Revises: cdad231052d5
|
|
5
|
-
Create Date: 2025-12-02 00:00:00.000000
|
|
6
|
-
|
|
7
|
-
"""
|
|
8
|
-
|
|
9
|
-
from typing import Sequence, Union
|
|
10
|
-
|
|
11
|
-
from alembic import op
|
|
12
|
-
import sqlalchemy as sa
|
|
13
|
-
from sqlalchemy.dialects import postgresql
|
|
14
|
-
|
|
15
|
-
# revision identifiers, used by Alembic.
|
|
16
|
-
revision: str = "add_artifact_lineage"
|
|
17
|
-
down_revision: Union[str, Sequence[str], None] = "cdad231052d5"
|
|
18
|
-
branch_labels: Union[str, Sequence[str], None] = None
|
|
19
|
-
depends_on: Union[str, Sequence[str], None] = None
|
|
20
|
-
|
|
21
|
-
# Schema name for all Boards tables
|
|
22
|
-
SCHEMA = "boards"
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
def upgrade() -> None:
|
|
26
|
-
"""Upgrade schema to use input_artifacts for lineage tracking."""
|
|
27
|
-
# Add input_artifacts JSONB column
|
|
28
|
-
op.add_column(
|
|
29
|
-
"generations",
|
|
30
|
-
sa.Column(
|
|
31
|
-
"input_artifacts",
|
|
32
|
-
postgresql.JSONB(astext_type=sa.Text()),
|
|
33
|
-
server_default=sa.text("'[]'::jsonb"),
|
|
34
|
-
nullable=False,
|
|
35
|
-
),
|
|
36
|
-
schema=SCHEMA,
|
|
37
|
-
)
|
|
38
|
-
|
|
39
|
-
# Migrate existing data from input_generation_ids to input_artifacts
|
|
40
|
-
# For backwards compatibility, use role="input" for legacy data
|
|
41
|
-
op.execute(f"""
|
|
42
|
-
UPDATE {SCHEMA}.generations
|
|
43
|
-
SET input_artifacts = (
|
|
44
|
-
SELECT COALESCE(jsonb_agg(
|
|
45
|
-
jsonb_build_object(
|
|
46
|
-
'generation_id', gen_id::text,
|
|
47
|
-
'role', 'input',
|
|
48
|
-
'artifact_type', COALESCE(g.artifact_type, 'unknown')
|
|
49
|
-
)
|
|
50
|
-
), '[]'::jsonb)
|
|
51
|
-
FROM unnest(input_generation_ids) AS gen_id
|
|
52
|
-
LEFT JOIN {SCHEMA}.generations g ON g.id = gen_id
|
|
53
|
-
)
|
|
54
|
-
WHERE input_generation_ids IS NOT NULL
|
|
55
|
-
AND array_length(input_generation_ids, 1) > 0
|
|
56
|
-
""")
|
|
57
|
-
|
|
58
|
-
# Add GIN index for JSONB queries
|
|
59
|
-
op.create_index(
|
|
60
|
-
"idx_generations_input_artifacts_gin",
|
|
61
|
-
"generations",
|
|
62
|
-
["input_artifacts"],
|
|
63
|
-
unique=False,
|
|
64
|
-
postgresql_using="gin",
|
|
65
|
-
schema=SCHEMA,
|
|
66
|
-
)
|
|
67
|
-
|
|
68
|
-
# Drop old lineage index
|
|
69
|
-
op.drop_index("idx_generations_lineage", table_name="generations", schema=SCHEMA)
|
|
70
|
-
|
|
71
|
-
# Drop old foreign key constraint on parent_generation_id
|
|
72
|
-
op.drop_constraint(
|
|
73
|
-
"generations_parent_generation_id_fkey",
|
|
74
|
-
"generations",
|
|
75
|
-
type_="foreignkey",
|
|
76
|
-
schema=SCHEMA,
|
|
77
|
-
)
|
|
78
|
-
|
|
79
|
-
# Drop old columns
|
|
80
|
-
op.drop_column("generations", "parent_generation_id", schema=SCHEMA)
|
|
81
|
-
op.drop_column("generations", "input_generation_ids", schema=SCHEMA)
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
def downgrade() -> None:
|
|
85
|
-
"""Downgrade schema back to parent_generation_id and input_generation_ids."""
|
|
86
|
-
# Add back old columns
|
|
87
|
-
op.add_column(
|
|
88
|
-
"generations",
|
|
89
|
-
sa.Column(
|
|
90
|
-
"input_generation_ids",
|
|
91
|
-
postgresql.ARRAY(postgresql.UUID()),
|
|
92
|
-
server_default=sa.text("'{}'::uuid[]"),
|
|
93
|
-
autoincrement=False,
|
|
94
|
-
nullable=False,
|
|
95
|
-
),
|
|
96
|
-
schema=SCHEMA,
|
|
97
|
-
)
|
|
98
|
-
op.add_column(
|
|
99
|
-
"generations",
|
|
100
|
-
sa.Column("parent_generation_id", postgresql.UUID(), autoincrement=False, nullable=True),
|
|
101
|
-
schema=SCHEMA,
|
|
102
|
-
)
|
|
103
|
-
|
|
104
|
-
# Migrate data back from input_artifacts to input_generation_ids
|
|
105
|
-
op.execute(f"""
|
|
106
|
-
UPDATE {SCHEMA}.generations
|
|
107
|
-
SET input_generation_ids = (
|
|
108
|
-
SELECT COALESCE(
|
|
109
|
-
array_agg((elem->>'generation_id')::uuid),
|
|
110
|
-
'{{}}'::uuid[]
|
|
111
|
-
)
|
|
112
|
-
FROM jsonb_array_elements(input_artifacts) elem
|
|
113
|
-
)
|
|
114
|
-
WHERE input_artifacts IS NOT NULL
|
|
115
|
-
AND jsonb_array_length(input_artifacts) > 0
|
|
116
|
-
""")
|
|
117
|
-
|
|
118
|
-
# Recreate foreign key constraint
|
|
119
|
-
op.create_foreign_key(
|
|
120
|
-
"generations_parent_generation_id_fkey",
|
|
121
|
-
"generations",
|
|
122
|
-
"generations",
|
|
123
|
-
["parent_generation_id"],
|
|
124
|
-
["id"],
|
|
125
|
-
source_schema=SCHEMA,
|
|
126
|
-
referent_schema=SCHEMA,
|
|
127
|
-
)
|
|
128
|
-
|
|
129
|
-
# Recreate old index
|
|
130
|
-
op.create_index("idx_generations_lineage", "generations", ["parent_generation_id"], unique=False, schema=SCHEMA)
|
|
131
|
-
|
|
132
|
-
# Drop new index and column
|
|
133
|
-
op.drop_index("idx_generations_input_artifacts_gin", table_name="generations", schema=SCHEMA)
|
|
134
|
-
op.drop_column("generations", "input_artifacts", schema=SCHEMA)
|