atomgit 1.0.4__tar.gz → 1.0.5__tar.gz

atomgit-1.0.5/MANIFEST.in

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomgit
-Version: 1.0.4
+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.4 → atomgit-1.0.5}/__init__.py

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

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

@@ -7,6 +7,8 @@ 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)
@@ -151,12 +153,13 @@ class HuggingFaceAPI:
                 # 使用 Monkey Patch 方式临时修改 huggingface_hub 的默认超时配置
                 commit_message = message or "Upload folder using atomgit client"
                 hf_constants.DEFAULT_REQUEST_TIMEOUT = upload_timeout
-                upload_folder(
+                upload_kwargs = dict(
                     repo_id=repo_id,
                     folder_path=str(temp_dir),
                     token=credentials['token'],
-                    commit_message=commit_message
+                    commit_message=commit_message,
                 )
+                upload_folder(**upload_kwargs)
                 
                 return True
             finally:
@@ -185,12 +188,13 @@ class HuggingFaceAPI:
             # 使用 Monkey Patch 方式临时修改 huggingface_hub 的默认超时配置
             commit_message = message or "Upload folder using atomgit client"
             hf_constants.DEFAULT_REQUEST_TIMEOUT = upload_timeout
-            upload_folder(
+            upload_kwargs = dict(
                 repo_id=repo_id,
                 folder_path=str(dir_path),
                 token=credentials['token'],
-                commit_message=commit_message
+                commit_message=commit_message,
             )
+            upload_folder(**upload_kwargs)
             
             return True
             

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomgit
-Version: 1.0.4
+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.4 → 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.4 → 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.4 → 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.4 → 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.4')
+@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.4 → atomgit-1.0.5}/setup.py

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