atomgit 1.0.3__tar.gz → 1.0.5__tar.gz

atomgit-1.0.5/MANIFEST.in

@@ -0,0 +1 @@
+include requirements.txt

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomgit
-Version: 1.0.3
+Version: 1.0.5
 Summary: AtomGit模型文件上传下载CLI工具
 Home-page: https://atomgit.com/gitcode-ai/atomgit_cli
 Author: AtomGit CLI Team
@@ -29,8 +29,8 @@ Requires-Dist: requests>=2.31.0
 Requires-Dist: tqdm>=4.66.1
 Requires-Dist: colorama>=0.4.6
 Requires-Dist: tabulate>=0.9.0
-Requires-Dist: huggingface-hub>=0.20.0
-Requires-Dist: datasets>=2.16.0
+Requires-Dist: huggingface-hub==1.1.7
+Requires-Dist: datasets==4.4.1
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

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

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

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

@@ -7,18 +7,15 @@ import urllib.error
 
 # 设置Hugging Face Hub的API端点为AtomGit
 os.environ["HF_ENDPOINT"] = "https://hub.atomgit.com"
+# 禁用Xet协议，避免 xet-write-token 请求
+os.environ["HF_HUB_DISABLE_XET"] = "1"
 # 设置缓存目录
 cache_dir = os.path.expanduser("~/.cache/atomgit")
 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 +150,17 @@ 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_kwargs = dict(
                     repo_id=repo_id,
                     folder_path=str(temp_dir),
                     token=credentials['token'],
                     commit_message=commit_message,
-                    upload_timeout=upload_timeout
                 )
+                upload_folder(**upload_kwargs)
+                
                 return True
             finally:
                 # 清理临时目录
@@ -186,15 +185,16 @@ 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_kwargs = dict(
                 repo_id=repo_id,
                 folder_path=str(dir_path),
                 token=credentials['token'],
                 commit_message=commit_message,
-                upload_timeout=upload_timeout
             )
+            upload_folder(**upload_kwargs)
             
             return True
             

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomgit
-Version: 1.0.3
+Version: 1.0.5
 Summary: AtomGit模型文件上传下载CLI工具
 Home-page: https://atomgit.com/gitcode-ai/atomgit_cli
 Author: AtomGit CLI Team
@@ -29,8 +29,8 @@ Requires-Dist: requests>=2.31.0
 Requires-Dist: tqdm>=4.66.1
 Requires-Dist: colorama>=0.4.6
 Requires-Dist: tabulate>=0.9.0
-Requires-Dist: huggingface-hub>=0.20.0
-Requires-Dist: datasets>=2.16.0
+Requires-Dist: huggingface-hub==1.1.7
+Requires-Dist: datasets==4.4.1
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{atomgit-1.0.3 → atomgit-1.0.5}/atomgit.egg-info/SOURCES.txt

@@ -1,3 +1,4 @@
+MANIFEST.in
 README.md
 __init__.py
 __main__.py
@@ -5,6 +6,7 @@ api.py
 atomgit_hub.py
 cli.py
 config.py
+requirements.txt
 setup.py
 test.py
 utils.py

{atomgit-1.0.3 → atomgit-1.0.5}/atomgit.egg-info/requires.txt

@@ -3,5 +3,5 @@ requests>=2.31.0
 tqdm>=4.66.1
 colorama>=0.4.6
 tabulate>=0.9.0
-huggingface-hub>=0.20.0
-datasets>=2.16.0
+huggingface-hub==1.1.7
+datasets==4.4.1

{atomgit-1.0.3 → atomgit-1.0.5}/atomgit_hub.py

@@ -14,6 +14,8 @@ from pathlib import Path
 
 # 设置Hugging Face Hub的API端点为AtomGit
 os.environ["HF_ENDPOINT"] = "https://hub.atomgit.com"
+# 禁用Xet协议，避免 xet-write-token 请求
+os.environ["HF_HUB_DISABLE_XET"] = "1"
 # 设置缓存目录
 cache_dir = os.path.expanduser("~/.cache/atomgit")
 os.makedirs(cache_dir, exist_ok=True)

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

@@ -9,6 +9,8 @@ from getpass import getpass
 
 # 设置Hugging Face Hub的API端点为AtomGit
 os.environ["HF_ENDPOINT"] = "https://hub.atomgit.com"
+# 禁用Xet协议，避免 xet-write-token 请求
+os.environ["HF_HUB_DISABLE_XET"] = "1"
 # 设置缓存目录
 cache_dir = os.path.expanduser("~/.cache/atomgit")
 os.makedirs(cache_dir, exist_ok=True)
@@ -37,7 +39,7 @@ except ImportError:
 
 
 @click.group()
-@click.version_option(version='1.0.3')
+@click.version_option(version='1.0.5')
 def cli():
     """AtomGit CLI - 基于Transformers和Hugging Face Hub的AtomGit平台模型文件上传下载工具"""
     pass

atomgit-1.0.5/requirements.txt

@@ -0,0 +1,7 @@
+click>=8.1.7
+requests>=2.31.0
+tqdm>=4.66.1
+colorama>=0.4.6
+tabulate>=0.9.0
+huggingface-hub==1.1.7
+datasets==4.4.1

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

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