atomgit 1.0.6__tar.gz → 1.0.7__tar.gz

{atomgit-1.0.6 → atomgit-1.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomgit
-Version: 1.0.6
+Version: 1.0.7
 Summary: AtomGit模型文件上传下载CLI工具
 Home-page: https://atomgit.com/gitcode-ai/atomgit_cli
 Author: AtomGit CLI Team

{atomgit-1.0.6 → atomgit-1.0.7}/__init__.py

@@ -8,7 +8,7 @@ AtomGit CLI - 基于Transformers和Hugging Face Hub的模型文件上传下载
 支持模型和数据集的上传、下载等操作。
 """
 
-__version__ = '1.0.6'
+__version__ = '1.0.7'
 __author__ = 'AtomGit CLI Team'
 __description__ = 'AtomGit模型文件上传下载CLI工具'
 

{atomgit-1.0.6 → atomgit-1.0.7}/atomgit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomgit
-Version: 1.0.6
+Version: 1.0.7
 Summary: AtomGit模型文件上传下载CLI工具
 Home-page: https://atomgit.com/gitcode-ai/atomgit_cli
 Author: AtomGit CLI Team

{atomgit-1.0.6 → atomgit-1.0.7}/atomgit_hub.py

@@ -26,7 +26,10 @@ from huggingface_hub import hf_hub_download, upload_folder as hf_upload_folder,
 try:
     from .rate_limiter import install_hf_rate_limiter
 except ImportError:
-    from rate_limiter import install_hf_rate_limiter
+    try:
+        from rate_limiter import install_hf_rate_limiter
+    except ImportError:
+        from atomgit.rate_limiter import install_hf_rate_limiter
 
 install_hf_rate_limiter()
 
@@ -337,8 +340,10 @@ def upload_folder(
         # 如果要上传到根目录，直接使用源文件夹
         upload_path = str(folder_path)
     else:
-        # 如果要上传到特定路径，需要重新组织目录结构
-        with tempfile.TemporaryDirectory() as temp_dir:
+        # 如果要上传到特定路径，需要重新组织目录结构。
+        # 注意：不能提前退出 with TemporaryDirectory()，否则上传时临时目录已被删除。
+        temp_dir = tempfile.mkdtemp(prefix="atomgit_upload_")
+        try:
             temp_path = Path(temp_dir)
             
             # 创建目标路径
@@ -348,7 +353,11 @@ def upload_folder(
             # 复制整个目录树
             shutil.copytree(folder_path, target_path, dirs_exist_ok=True)
             
-            upload_path = str(temp_path)
+        except Exception:
+            shutil.rmtree(temp_dir, ignore_errors=True)
+            raise
+
+        upload_path = str(temp_path)
     # 使用 Monkey Patch 方式临时修改 huggingface_hub 的默认超时配置
     from huggingface_hub import constants as hf_constants
     
@@ -378,6 +387,10 @@ def upload_folder(
             raise Exception(f"仓库不存在：{repo_id}")
         else:
             raise Exception(f"上传失败：{error_msg}")
+    finally:
+        # 清理临时目录（仅重新组织目录结构时创建）
+        if path_in_repo not in ("./", ".", ""):
+            shutil.rmtree(temp_dir, ignore_errors=True)
 
 
 def create_repository(

{atomgit-1.0.6 → atomgit-1.0.7}/cli.py

@@ -39,7 +39,7 @@ except ImportError:
 
 
 @click.group()
-@click.version_option(version='1.0.6')
+@click.version_option(version='1.0.7')
 def cli():
     """AtomGit CLI - 基于Transformers和Hugging Face Hub的AtomGit平台模型文件上传下载工具"""
     pass

{atomgit-1.0.6 → atomgit-1.0.7}/setup.py

@@ -31,7 +31,7 @@ def read_requirements():
 
 setup(
     name='atomgit',
-    version='1.0.6',
+    version='1.0.7',
     author='AtomGit CLI Team',
     author_email='sa@atomgit.com',
     description='AtomGit模型文件上传下载CLI工具',