feat(ai-studio): add remote image loading helpers

This commit is contained in:
chengma
2026-07-11 12:31:12 +08:00
parent 10a28c1e77
commit 1267e0a727
4 changed files with 632 additions and 3 deletions
+17
View File
@@ -553,6 +553,23 @@ def mark_asset_status(asset_id, status, path=None, conn=None):
return get_asset(asset_id, conn=database)
def update_asset_local_path(asset_id, local_path, status=ASSET_STATUS_AVAILABLE, path=None, conn=None):
if str(status) not in ASSET_STATUSES:
raise db.DbError("AI工场资产状态必须是 available 或 missing")
abs_local_path = os.path.abspath(local_path) if local_path else None
with _connection(conn, path) as database:
with database:
database.execute(
"""
UPDATE image_studio_assets
SET local_path = ?, status = ?, updated_at = ?
WHERE id = ?
""",
(abs_local_path, str(status), _now(), int(asset_id)),
)
return get_asset(asset_id, conn=database)
def create_job(
project_id,
*,