xiaoshiai-hub 1.1.1__tar.gz → 1.1.3__tar.gz

{xiaoshiai_hub-1.1.1/xiaoshiai_hub.egg-info → xiaoshiai_hub-1.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xiaoshiai-hub
-Version: 1.1.1
+Version: 1.1.3
 Summary: Python SDK for XiaoShi AI Hub - Upload, download, and manage AI models and datasets with xpai-enc encryption support
 Home-page: https://github.com/poxiaoyun/moha-sdk
 Author: XiaoShi AI
@@ -244,25 +244,65 @@ client = HubClient(
     password="your-password",
 )
 
+# 创建仓库
+repo = client.create_repository(
+    organization="demo",
+    repo_type="models",
+    repo_name="my-model",
+    description="我的模型",
+    visibility="internal",
+    metadata={
+        "license": ["apache-2.0"],
+        "frameworks": ["transformers"],
+    },
+)
+print(f"仓库已创建: {repo.name}")
+
 # 获取仓库信息
 repo_info = client.get_repository_info("demo", "models", "my-model")
 print(f"仓库名称: {repo_info.name}")
 print(f"组织: {repo_info.organization}")
+print(f"所有者: {repo_info.owner}")
+print(f"可见性: {repo_info.visibility}")
+
+# 更新仓库
+client.update_repository(
+    organization="demo",
+    repo_type="models",
+    repo_name="my-model",
+    description="更新后的描述",
+)
 
 # 列出分支
-branches = client.list_branches("demo", "models", "my-model")
-for branch in branches:
-    print(f"分支: {branch.name} (commit: {branch.commit_sha})")
+refs = client.get_repository_refs("demo", "models", "my-model")
+for ref in refs:
+    print(f"分支: {ref.name} (commit: {ref.hash[:8]})")
+
+# 创建分支（幂等操作，已存在则直接返回）
+client.create_branch("demo", "models", "my-model", "dev", "main")
+
+# 删除分支（幂等操作，不存在则直接返回）
+client.delete_branch("demo", "models", "my-model", "dev")
 
 # 浏览仓库内容
 content = client.get_repository_content("demo", "models", "my-model", "main")
 for entry in content.entries:
     print(f"{entry.type}: {entry.name}")
+
+# 删除仓库
+client.delete_repository("demo", "models", "my-model")
 ```
 
 ## 🔐 加密功能
 
-SDK 提供了智能加密功能，使用 AES-256-CTR 算法对大型模型文件进行加密。
+SDK 提供了智能加密功能，支持 **AES** 和 **SM4** 两种加密算法对大型模型文件进行加密。
+
+### 支持的加密算法
+
+| 算法 | 说明 |
+|------|------|
+| `AES` | AES-256-CTR 模式，国际通用标准（默认） |
+| `SM4` | SM4-CTR 模式，国密标准 |
 
 ### 自动加密规则
 
@@ -278,11 +318,21 @@ SDK 提供了智能加密功能，使用 AES-256-CTR 算法对大型模型文件
 ```python
 from xiaoshiai_hub import upload_folder
 
-# 上传文件夹，自动加密大型模型文件
+# 上传文件夹，使用 AES 加密（默认）
 result = upload_folder(
     folder_path="./llama-7b",
     repo_id="demo/llama-7b",
-    encryption_password="my-secure-password-123",  # 设置加密密码
+    encryption_password="my-secure-password-123",
+    username="your-username",
+    password="your-password",
+)
+
+# 使用 SM4 国密算法加密
+result = upload_folder(
+    folder_path="./llama-7b",
+    repo_id="demo/llama-7b",
+    encryption_password="my-secure-password-123",
+    algorithm="SM4",  # 使用 SM4 加密
     username="your-username",
     password="your-password",
 )
@@ -326,7 +376,7 @@ export MOHA_ENCRYPTION_PASSWORD="your-encryption-password"
 
 ## 🖥️ 命令行工具 (CLI)
 
-SDK 提供了 `moha` 命令行工具，支持常见的上传下载操作。
+SDK 提供了 `moha` 命令行工具，支持登录认证、仓库管理、分支管理、上传下载等操作。
 
 ### 基本用法
 
@@ -334,6 +384,75 @@ SDK 提供了 `moha` 命令行工具，支持常见的上传下载操作。
 moha --help
 ```
 
+### 登录认证
+
+```bash
+# 登录（交互式输入用户名和密码）
+moha login
+
+# 直接指定用户名和密码
+moha login --username your-username --password your-password
+
+# 查看当前登录状态
+moha whoami
+
+# 退出登录
+moha logout
+```
+
+登录后，Token 会保存到 `~/.moha/token.json`，后续命令无需重复输入认证信息。
+
+### 仓库管理
+
+```bash
+# 创建仓库
+moha repo-create org/my-model \
+    --description "我的模型" \
+    --visibility internal \
+    --license apache-2.0 \
+    --tasks text-generation \
+    --frameworks transformers
+
+# 创建数据集仓库
+moha repo-create org/my-dataset \
+    --repo-type datasets \
+    --description "我的数据集" \
+    --visibility private
+
+# 查看仓库信息
+moha repo-info org/my-model
+
+# 更新仓库信息
+moha repo-update org/my-model \
+    --description "更新后的描述" \
+    --tags production
+
+# 删除仓库（需要确认）
+moha repo-delete org/my-model
+
+# 跳过确认直接删除
+moha repo-delete org/my-model -y
+```
+
+### 分支管理
+
+```bash
+# 列出仓库的所有分支
+moha branch-list org/my-model
+
+# 创建分支（基于 main 分支）
+moha branch-create org/my-model dev
+
+# 创建分支（基于指定分支）
+moha branch-create org/my-model feature --from dev
+
+# 删除分支
+moha branch-delete org/my-model dev
+
+# 跳过确认直接删除
+moha branch-delete org/my-model dev -y
+```
+
 ### 上传文件夹
 
 ```bash
@@ -353,10 +472,18 @@ moha upload ./my_model org/my-model \
     --username your-username \
     --password your-password
 
-# 启用加密
+# 启用加密（默认使用 AES）
+moha upload ./my_model org/my-model \
+    --encrypt \
+    --encryption-password "your-secret" \
+    --username your-username \
+    --password your-password
+
+# 使用 SM4 国密算法加密
 moha upload ./my_model org/my-model \
     --encrypt \
     --encryption-password "your-secret" \
+    --algorithm SM4 \
     --username your-username \
     --password your-password
 ```
@@ -419,26 +546,75 @@ moha download-file org/my-model model.safetensors \
     --password your-password
 ```
 
+### CLI 命令列表
+
+| 命令 | 说明 |
+|------|------|
+| `moha login` | 登录并保存 Token |
+| `moha logout` | 退出登录并删除 Token |
+| `moha whoami` | 查看当前登录状态 |
+| `moha repo-create` | 创建仓库 |
+| `moha repo-update` | 更新仓库 |
+| `moha repo-delete` | 删除仓库 |
+| `moha repo-info` | 查看仓库信息 |
+| `moha branch-create` | 创建分支 |
+| `moha branch-delete` | 删除分支 |
+| `moha branch-list` | 列出仓库的所有分支 |
+| `moha upload` | 上传文件夹到仓库 |
+| `moha upload-file` | 上传单个文件到仓库 |
+| `moha download` | 下载整个仓库 |
+| `moha download-file` | 从仓库下载单个文件 |
+
 ### CLI 参数说明
 
+#### 通用参数
+
 | 参数 | 说明 | 适用命令 |
 |------|------|----------|
-| `--repo-type, -t` | 仓库类型：`models` 或 `datasets`（默认：models） | 所有 |
-| `--revision, -r` | 分支/标签/提交（默认：main） | 所有 |
+| `--repo-type, -t` | 仓库类型：`models` 或 `datasets`（默认：models） | 大部分命令 |
 | `--base-url` | API 基础 URL（默认：环境变量 MOHA_ENDPOINT） | 所有 |
 | `--token` | 认证令牌 | 所有 |
 | `--username` | 用户名 | 所有 |
 | `--password` | 密码 | 所有 |
+
+#### 上传/下载参数
+
+| 参数 | 说明 | 适用命令 |
+|------|------|----------|
+| `--revision, -r` | 分支/标签/提交（默认：main） | upload, download |
 | `--message, -m` | 提交消息 | upload, upload-file |
 | `--ignore, -i` | 忽略模式（可多次使用） | upload, download |
 | `--include` | 包含模式（可多次使用） | download |
 | `--encrypt, -e` | 启用加密 | upload, upload-file |
 | `--encryption-password` | 加密密码 | upload, upload-file |
+| `--algorithm, -a` | 加密算法：`AES` 或 `SM4`（默认：AES） | upload, upload-file |
 | `--path-in-repo, -p` | 仓库中的文件路径 | upload-file |
 | `--temp-dir` | 加密临时目录 | upload |
 | `--local-dir, -o` | 本地保存目录 | download, download-file |
 | `--quiet, -q` | 禁用进度条 | download, download-file |
 
+#### 仓库管理参数
+
+| 参数 | 说明 | 适用命令 |
+|------|------|----------|
+| `--description, -d` | 仓库描述 | repo-create, repo-update |
+| `--visibility, -v` | 可见性：`public`、`internal`、`private` | repo-create, repo-update |
+| `--license` | 许可证（可多次使用） | repo-create, repo-update |
+| `--tasks` | 任务类型（可多次使用） | repo-create, repo-update |
+| `--languages` | 语言（可多次使用） | repo-create, repo-update |
+| `--tags` | 标签（可多次使用） | repo-create, repo-update |
+| `--frameworks` | 框架（可多次使用） | repo-create, repo-update |
+| `--base-model` | 基础模型（可多次使用） | repo-create, repo-update |
+| `--relationship` | 与基础模型的关系 | repo-create, repo-update |
+| `--yes, -y` | 跳过确认提示 | repo-delete, branch-delete |
+
+#### 分支管理参数
+
+| 参数 | 说明 | 适用命令 |
+|------|------|----------|
+| `--from, -f` | 基于哪个分支创建（默认：main） | branch-create |
+| `--yes, -y` | 跳过确认提示 | branch-delete |
+
 ### 使用环境变量
 
 可以通过环境变量设置认证信息，避免每次输入：

{xiaoshiai_hub-1.1.1 → xiaoshiai_hub-1.1.3}/README.md

@@ -189,25 +189,65 @@ client = HubClient(
     password="your-password",
 )
 
+# 创建仓库
+repo = client.create_repository(
+    organization="demo",
+    repo_type="models",
+    repo_name="my-model",
+    description="我的模型",
+    visibility="internal",
+    metadata={
+        "license": ["apache-2.0"],
+        "frameworks": ["transformers"],
+    },
+)
+print(f"仓库已创建: {repo.name}")
+
 # 获取仓库信息
 repo_info = client.get_repository_info("demo", "models", "my-model")
 print(f"仓库名称: {repo_info.name}")
 print(f"组织: {repo_info.organization}")
+print(f"所有者: {repo_info.owner}")
+print(f"可见性: {repo_info.visibility}")
+
+# 更新仓库
+client.update_repository(
+    organization="demo",
+    repo_type="models",
+    repo_name="my-model",
+    description="更新后的描述",
+)
 
 # 列出分支
-branches = client.list_branches("demo", "models", "my-model")
-for branch in branches:
-    print(f"分支: {branch.name} (commit: {branch.commit_sha})")
+refs = client.get_repository_refs("demo", "models", "my-model")
+for ref in refs:
+    print(f"分支: {ref.name} (commit: {ref.hash[:8]})")
+
+# 创建分支（幂等操作，已存在则直接返回）
+client.create_branch("demo", "models", "my-model", "dev", "main")
+
+# 删除分支（幂等操作，不存在则直接返回）
+client.delete_branch("demo", "models", "my-model", "dev")
 
 # 浏览仓库内容
 content = client.get_repository_content("demo", "models", "my-model", "main")
 for entry in content.entries:
     print(f"{entry.type}: {entry.name}")
+
+# 删除仓库
+client.delete_repository("demo", "models", "my-model")
 ```
 
 ## 🔐 加密功能
 
-SDK 提供了智能加密功能，使用 AES-256-CTR 算法对大型模型文件进行加密。
+SDK 提供了智能加密功能，支持 **AES** 和 **SM4** 两种加密算法对大型模型文件进行加密。
+
+### 支持的加密算法
+
+| 算法 | 说明 |
+|------|------|
+| `AES` | AES-256-CTR 模式，国际通用标准（默认） |
+| `SM4` | SM4-CTR 模式，国密标准 |
 
 ### 自动加密规则
 
@@ -223,11 +263,21 @@ SDK 提供了智能加密功能，使用 AES-256-CTR 算法对大型模型文件
 ```python
 from xiaoshiai_hub import upload_folder
 
-# 上传文件夹，自动加密大型模型文件
+# 上传文件夹，使用 AES 加密（默认）
 result = upload_folder(
     folder_path="./llama-7b",
     repo_id="demo/llama-7b",
-    encryption_password="my-secure-password-123",  # 设置加密密码
+    encryption_password="my-secure-password-123",
+    username="your-username",
+    password="your-password",
+)
+
+# 使用 SM4 国密算法加密
+result = upload_folder(
+    folder_path="./llama-7b",
+    repo_id="demo/llama-7b",
+    encryption_password="my-secure-password-123",
+    algorithm="SM4",  # 使用 SM4 加密
     username="your-username",
     password="your-password",
 )
@@ -271,7 +321,7 @@ export MOHA_ENCRYPTION_PASSWORD="your-encryption-password"
 
 ## 🖥️ 命令行工具 (CLI)
 
-SDK 提供了 `moha` 命令行工具，支持常见的上传下载操作。
+SDK 提供了 `moha` 命令行工具，支持登录认证、仓库管理、分支管理、上传下载等操作。
 
 ### 基本用法
 
@@ -279,6 +329,75 @@ SDK 提供了 `moha` 命令行工具，支持常见的上传下载操作。
 moha --help
 ```
 
+### 登录认证
+
+```bash
+# 登录（交互式输入用户名和密码）
+moha login
+
+# 直接指定用户名和密码
+moha login --username your-username --password your-password
+
+# 查看当前登录状态
+moha whoami
+
+# 退出登录
+moha logout
+```
+
+登录后，Token 会保存到 `~/.moha/token.json`，后续命令无需重复输入认证信息。
+
+### 仓库管理
+
+```bash
+# 创建仓库
+moha repo-create org/my-model \
+    --description "我的模型" \
+    --visibility internal \
+    --license apache-2.0 \
+    --tasks text-generation \
+    --frameworks transformers
+
+# 创建数据集仓库
+moha repo-create org/my-dataset \
+    --repo-type datasets \
+    --description "我的数据集" \
+    --visibility private
+
+# 查看仓库信息
+moha repo-info org/my-model
+
+# 更新仓库信息
+moha repo-update org/my-model \
+    --description "更新后的描述" \
+    --tags production
+
+# 删除仓库（需要确认）
+moha repo-delete org/my-model
+
+# 跳过确认直接删除
+moha repo-delete org/my-model -y
+```
+
+### 分支管理
+
+```bash
+# 列出仓库的所有分支
+moha branch-list org/my-model
+
+# 创建分支（基于 main 分支）
+moha branch-create org/my-model dev
+
+# 创建分支（基于指定分支）
+moha branch-create org/my-model feature --from dev
+
+# 删除分支
+moha branch-delete org/my-model dev
+
+# 跳过确认直接删除
+moha branch-delete org/my-model dev -y
+```
+
 ### 上传文件夹
 
 ```bash
@@ -298,10 +417,18 @@ moha upload ./my_model org/my-model \
     --username your-username \
     --password your-password
 
-# 启用加密
+# 启用加密（默认使用 AES）
+moha upload ./my_model org/my-model \
+    --encrypt \
+    --encryption-password "your-secret" \
+    --username your-username \
+    --password your-password
+
+# 使用 SM4 国密算法加密
 moha upload ./my_model org/my-model \
     --encrypt \
     --encryption-password "your-secret" \
+    --algorithm SM4 \
     --username your-username \
     --password your-password
 ```
@@ -364,26 +491,75 @@ moha download-file org/my-model model.safetensors \
     --password your-password
 ```
 
+### CLI 命令列表
+
+| 命令 | 说明 |
+|------|------|
+| `moha login` | 登录并保存 Token |
+| `moha logout` | 退出登录并删除 Token |
+| `moha whoami` | 查看当前登录状态 |
+| `moha repo-create` | 创建仓库 |
+| `moha repo-update` | 更新仓库 |
+| `moha repo-delete` | 删除仓库 |
+| `moha repo-info` | 查看仓库信息 |
+| `moha branch-create` | 创建分支 |
+| `moha branch-delete` | 删除分支 |
+| `moha branch-list` | 列出仓库的所有分支 |
+| `moha upload` | 上传文件夹到仓库 |
+| `moha upload-file` | 上传单个文件到仓库 |
+| `moha download` | 下载整个仓库 |
+| `moha download-file` | 从仓库下载单个文件 |
+
 ### CLI 参数说明
 
+#### 通用参数
+
 | 参数 | 说明 | 适用命令 |
 |------|------|----------|
-| `--repo-type, -t` | 仓库类型：`models` 或 `datasets`（默认：models） | 所有 |
-| `--revision, -r` | 分支/标签/提交（默认：main） | 所有 |
+| `--repo-type, -t` | 仓库类型：`models` 或 `datasets`（默认：models） | 大部分命令 |
 | `--base-url` | API 基础 URL（默认：环境变量 MOHA_ENDPOINT） | 所有 |
 | `--token` | 认证令牌 | 所有 |
 | `--username` | 用户名 | 所有 |
 | `--password` | 密码 | 所有 |
+
+#### 上传/下载参数
+
+| 参数 | 说明 | 适用命令 |
+|------|------|----------|
+| `--revision, -r` | 分支/标签/提交（默认：main） | upload, download |
 | `--message, -m` | 提交消息 | upload, upload-file |
 | `--ignore, -i` | 忽略模式（可多次使用） | upload, download |
 | `--include` | 包含模式（可多次使用） | download |
 | `--encrypt, -e` | 启用加密 | upload, upload-file |
 | `--encryption-password` | 加密密码 | upload, upload-file |
+| `--algorithm, -a` | 加密算法：`AES` 或 `SM4`（默认：AES） | upload, upload-file |
 | `--path-in-repo, -p` | 仓库中的文件路径 | upload-file |
 | `--temp-dir` | 加密临时目录 | upload |
 | `--local-dir, -o` | 本地保存目录 | download, download-file |
 | `--quiet, -q` | 禁用进度条 | download, download-file |
 
+#### 仓库管理参数
+
+| 参数 | 说明 | 适用命令 |
+|------|------|----------|
+| `--description, -d` | 仓库描述 | repo-create, repo-update |
+| `--visibility, -v` | 可见性：`public`、`internal`、`private` | repo-create, repo-update |
+| `--license` | 许可证（可多次使用） | repo-create, repo-update |
+| `--tasks` | 任务类型（可多次使用） | repo-create, repo-update |
+| `--languages` | 语言（可多次使用） | repo-create, repo-update |
+| `--tags` | 标签（可多次使用） | repo-create, repo-update |
+| `--frameworks` | 框架（可多次使用） | repo-create, repo-update |
+| `--base-model` | 基础模型（可多次使用） | repo-create, repo-update |
+| `--relationship` | 与基础模型的关系 | repo-create, repo-update |
+| `--yes, -y` | 跳过确认提示 | repo-delete, branch-delete |
+
+#### 分支管理参数
+
+| 参数 | 说明 | 适用命令 |
+|------|------|----------|
+| `--from, -f` | 基于哪个分支创建（默认：main） | branch-create |
+| `--yes, -y` | 跳过确认提示 | branch-delete |
+
 ### 使用环境变量
 
 可以通过环境变量设置认证信息，避免每次输入：

{xiaoshiai_hub-1.1.1 → xiaoshiai_hub-1.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "xiaoshiai-hub"
-version = "1.1.1"
+version = "1.1.3"
 description = "Python SDK for XiaoShi AI Hub - Upload, download, and manage AI models and datasets with xpai-enc encryption support"
 readme = "README.md"
 requires-python = ">=3.7"

{xiaoshiai_hub-1.1.1 → xiaoshiai_hub-1.1.3}/setup.py

@@ -17,7 +17,7 @@ else:
     long_description = "Python SDK for XiaoShi AI Hub"
 
 # Read version from __init__.py
-version = "1.1.1"
+version = "1.1.3"
 init_file = Path(__file__).parent / "xiaoshiai_hub" / "__init__.py"
 if init_file.exists():
     with open(init_file, "r", encoding="utf-8") as f:

{xiaoshiai_hub-1.1.1 → xiaoshiai_hub-1.1.3}/xiaoshiai_hub/__init__.py

@@ -28,6 +28,10 @@ from .auth import (
     load_token,
     delete_token,
 )
+from .envelope_crypto import (
+    Algorithm,
+    envelope_enc_file,
+)
 
 # Upload functionality (requires GitPython)
 try:
@@ -58,6 +62,9 @@ __all__ = [
     "save_token",
     "load_token",
     "delete_token",
+    # Encryption
+    "Algorithm",
+    "envelope_enc_file",
     # Exceptions
     "HubException",
     "RepositoryNotFoundError",