Upload 云端_定时版本检测引擎.py

云端_定时版本检测引擎.py

@@ -17,7 +17,7 @@ import os
 import tempfile
 import zipfile
 import 数据库连接 as db
-from huggingface_hub import upload_file, HfApi
+from huggingface_hub import CommitOperationAdd, HfApi
 
 logger = logging.getLogger("ComfyUI-Ranking.VersionCheck")
 
@@ -100,6 +100,13 @@ async def fetch_latest_github_hash(repo_url, token):
     return sha[:7] if sha else None
 
 
+def _upload_precache_archive(temp_path, path_in_repo, storage_repo, hf_token):
+    operation = CommitOperationAdd(path_or_fileobj=temp_path, path_in_repo=path_in_repo)
+    return HfApi(token=hf_token).create_commit(
+        repo_id=storage_repo, repo_type="dataset", operations=[operation],
+        commit_message="Cache verified source archive")
+
+
 async def precache_github_zip(repo_url: str, token: Optional[str], item_id: str, version_hash: str):
     """按明确commit下载，上传成功后才发布来源匹配的不可变HF缓存。"""
     temp_path = None
@@ -159,8 +166,7 @@ async def precache_github_zip(repo_url: str, token: Optional[str], item_id: str,
         stage = "archive_validation"
         await run_owned_thread(validate_zip_archive, temp_path, size)
         stage = "hf_upload"
-        uploaded = await run_owned_thread(upload_file, path_or_fileobj=temp_path, path_in_repo=path_in_repo,
-            repo_id=storage_repo, repo_type="dataset", token=hf_token)
+        uploaded = await run_owned_thread(_upload_precache_archive, temp_path, path_in_repo, storage_repo, hf_token)
         stage = "hf_commit"
         hf_commit = commit_sha(getattr(uploaded, 'oid', None))
         stage = "source_recheck"