From dde5a60357def710d27616ae6a56916731a2cf6c Mon Sep 17 00:00:00 2001 From: Luke Mino-Altherr Date: Sat, 14 Mar 2026 22:13:00 -0400 Subject: [PATCH] Rename prompt_id to job_id on asset_references Rename the column in the DB model, migration, and service schemas. The API response emits both job_id and prompt_id (deprecated alias) for backward compatibility with the cloud API. Amp-Thread-ID: https://ampcode.com/threads/T-019cef41-60b0-752a-aa3c-ed7f20fda2f7 Co-authored-by: Amp --- ..._metadata_prompt.py => 0003_add_metadata_job_id.py} | 10 +++++----- app/assets/api/routes.py | 3 ++- app/assets/api/schemas_out.py | 3 ++- app/assets/database/models.py | 2 +- app/assets/services/schemas.py | 4 ++-- 5 files changed, 12 insertions(+), 10 deletions(-) rename alembic_db/versions/{0003_add_metadata_prompt.py => 0003_add_metadata_job_id.py} (88%) diff --git a/alembic_db/versions/0003_add_metadata_prompt.py b/alembic_db/versions/0003_add_metadata_job_id.py similarity index 88% rename from alembic_db/versions/0003_add_metadata_prompt.py rename to alembic_db/versions/0003_add_metadata_job_id.py index 522d02c33..66c0458ed 100644 --- a/alembic_db/versions/0003_add_metadata_prompt.py +++ b/alembic_db/versions/0003_add_metadata_job_id.py @@ -1,8 +1,8 @@ """ -Add system_metadata and prompt_id columns to asset_references. +Add system_metadata and job_id columns to asset_references. Change preview_id FK from assets.id to asset_references.id. -Revision ID: 0003_add_metadata_prompt +Revision ID: 0003_add_metadata_job_id Revises: 0002_merge_to_asset_references Create Date: 2026-03-09 """ @@ -12,7 +12,7 @@ import sqlalchemy as sa from app.database.models import NAMING_CONVENTION -revision = "0003_add_metadata_prompt" +revision = "0003_add_metadata_job_id" down_revision = "0002_merge_to_asset_references" branch_labels = None depends_on = None @@ -24,7 +24,7 @@ def upgrade() -> None: sa.Column("system_metadata", sa.JSON(), nullable=True) ) batch_op.add_column( - sa.Column("prompt_id", sa.String(length=36), nullable=True) + sa.Column("job_id", sa.String(length=36), nullable=True) ) # Change preview_id FK from assets.id to asset_references.id (self-ref). @@ -62,5 +62,5 @@ def downgrade() -> None: ) with op.batch_alter_table("asset_references") as batch_op: - batch_op.drop_column("prompt_id") + batch_op.drop_column("job_id") batch_op.drop_column("system_metadata") diff --git a/app/assets/api/routes.py b/app/assets/api/routes.py index 46a9af698..02b4d1726 100644 --- a/app/assets/api/routes.py +++ b/app/assets/api/routes.py @@ -171,7 +171,8 @@ def _build_asset_response(result: schemas.AssetDetailResult | schemas.UploadResu preview_id=result.ref.preview_id, user_metadata=result.ref.user_metadata or {}, metadata=result.ref.system_metadata, - prompt_id=result.ref.prompt_id, + job_id=result.ref.job_id, + prompt_id=result.ref.job_id, # deprecated: mirrors job_id for cloud compat created_at=result.ref.created_at, updated_at=result.ref.updated_at, last_access_time=result.ref.last_access_time, diff --git a/app/assets/api/schemas_out.py b/app/assets/api/schemas_out.py index 8db642b45..48d6292e4 100644 --- a/app/assets/api/schemas_out.py +++ b/app/assets/api/schemas_out.py @@ -16,7 +16,8 @@ class Asset(BaseModel): user_metadata: dict[str, Any] = Field(default_factory=dict) is_immutable: bool = False metadata: dict[str, Any] | None = None - prompt_id: str | None = None + job_id: str | None = None + prompt_id: str | None = None # deprecated: use job_id created_at: datetime updated_at: datetime last_access_time: datetime | None = None diff --git a/app/assets/database/models.py b/app/assets/database/models.py index 5534e89d5..c3df8d669 100644 --- a/app/assets/database/models.py +++ b/app/assets/database/models.py @@ -93,7 +93,7 @@ class AssetReference(Base): system_metadata: Mapped[dict[str, Any] | None] = mapped_column( JSON(none_as_null=True), nullable=True, default=None ) - prompt_id: Mapped[str | None] = mapped_column(String(36), nullable=True, default=None) + job_id: Mapped[str | None] = mapped_column(String(36), nullable=True, default=None) created_at: Mapped[datetime] = mapped_column( DateTime(timezone=False), nullable=False, default=get_utc_now ) diff --git a/app/assets/services/schemas.py b/app/assets/services/schemas.py index b744283b2..0eb128f58 100644 --- a/app/assets/services/schemas.py +++ b/app/assets/services/schemas.py @@ -26,7 +26,7 @@ class ReferenceData: created_at: datetime updated_at: datetime system_metadata: dict[str, Any] | None = None - prompt_id: str | None = None + job_id: str | None = None last_access_time: datetime | None = None @@ -96,7 +96,7 @@ def extract_reference_data(ref: AssetReference) -> ReferenceData: user_metadata=ref.user_metadata, preview_id=ref.preview_id, system_metadata=ref.system_metadata, - prompt_id=ref.prompt_id, + job_id=ref.job_id, created_at=ref.created_at, updated_at=ref.updated_at, last_access_time=ref.last_access_time,