feat: add resource library and real image workflow
This commit is contained in:
@@ -29,11 +29,22 @@ async def run_qc_activity(payload: StepActivityInput) -> MockActivityResult:
|
||||
candidate_uri: str | None = None
|
||||
|
||||
if passed:
|
||||
if payload.source_asset_id is None:
|
||||
raise ValueError("run_qc_activity requires source_asset_id")
|
||||
|
||||
source_asset = await session.get(AssetORM, payload.source_asset_id)
|
||||
if source_asset is None:
|
||||
raise ValueError(f"Source asset {payload.source_asset_id} not found")
|
||||
if source_asset.order_id != payload.order_id:
|
||||
raise ValueError(
|
||||
f"Source asset {payload.source_asset_id} does not belong to order {payload.order_id}"
|
||||
)
|
||||
|
||||
candidate = AssetORM(
|
||||
order_id=payload.order_id,
|
||||
asset_type=AssetType.QC_CANDIDATE,
|
||||
step_name=payload.step_name,
|
||||
uri=mock_uri(payload.order_id, payload.step_name.value, "candidate.png"),
|
||||
uri=source_asset.uri,
|
||||
metadata_json=jsonable({"source_asset_id": payload.source_asset_id}),
|
||||
)
|
||||
session.add(candidate)
|
||||
|
||||
Reference in New Issue
Block a user