atomgit 1.0.3__tar.gz → 1.0.4__tar.gz

{atomgit-1.0.3 → atomgit-1.0.4}/PKG-INFO

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

{atomgit-1.0.3 → atomgit-1.0.4}/__init__.py

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

{atomgit-1.0.3 → atomgit-1.0.4}/api.py

@@ -13,12 +13,7 @@ os.makedirs(cache_dir, exist_ok=True)
 os.environ["HF_HOME"] = cache_dir
 
 
-from huggingface_hub import hf_hub_download, create_repo, snapshot_download
-
-try:
-    from .atomgit_hub import upload_folder as atomgit_upload_folder
-except ImportError:
-    from atomgit_hub import upload_folder as atomgit_upload_folder
+from huggingface_hub import hf_hub_download, upload_folder, create_repo, snapshot_download, constants as hf_constants
 
 try:
     from .config import config
@@ -153,15 +148,16 @@ class HuggingFaceAPI:
                 # 复制文件到临时目录
                 import shutil
                 shutil.copy2(file_path, target_file)
-                # 使用AtomGit Hub SDK上传整个目录
+                # 使用 Monkey Patch 方式临时修改 huggingface_hub 的默认超时配置
                 commit_message = message or "Upload folder using atomgit client"
-                atomgit_upload_folder(
+                hf_constants.DEFAULT_REQUEST_TIMEOUT = upload_timeout
+                upload_folder(
                     repo_id=repo_id,
                     folder_path=str(temp_dir),
                     token=credentials['token'],
-                    commit_message=commit_message,
-                    upload_timeout=upload_timeout
+                    commit_message=commit_message
                 )
+                
                 return True
             finally:
                 # 清理临时目录
@@ -186,14 +182,14 @@ class HuggingFaceAPI:
                 print("未找到登录凭证")
                 return False
             
-            # 直接使用AtomGit Hub SDK上传目录
+            # 使用 Monkey Patch 方式临时修改 huggingface_hub 的默认超时配置
             commit_message = message or "Upload folder using atomgit client"
-            atomgit_upload_folder(
+            hf_constants.DEFAULT_REQUEST_TIMEOUT = upload_timeout
+            upload_folder(
                 repo_id=repo_id,
                 folder_path=str(dir_path),
                 token=credentials['token'],
-                commit_message=commit_message,
-                upload_timeout=upload_timeout
+                commit_message=commit_message
             )
             
             return True

{atomgit-1.0.3 → atomgit-1.0.4}/atomgit.egg-info/PKG-INFO

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

{atomgit-1.0.3 → atomgit-1.0.4}/cli.py

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

{atomgit-1.0.3 → atomgit-1.0.4}/setup.py

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