xiaoshiai-hub 1.1.2__tar.gz → 1.1.4__tar.gz

{xiaoshiai_hub-1.1.2/xiaoshiai_hub.egg-info → xiaoshiai_hub-1.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xiaoshiai-hub
-Version: 1.1.2
+Version: 1.1.4
 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
@@ -31,6 +31,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: requests>=2.20.0
 Requires-Dist: tqdm>=4.62.0
+Requires-Dist: cryptography>=46.0.3
 Provides-Extra: upload
 Requires-Dist: gitpython>=3.1.0; extra == "upload"
 Provides-Extra: encryption
@@ -244,20 +245,53 @@ 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")
 ```
 
 ## 🔐 加密功能
@@ -343,7 +377,7 @@ export MOHA_ENCRYPTION_PASSWORD="your-encryption-password"
 
 ## 🖥️ 命令行工具 (CLI)
 
-SDK 提供了 `moha` 命令行工具，支持常见的上传下载操作。
+SDK 提供了 `moha` 命令行工具，支持登录认证、仓库管理、分支管理、上传下载等操作。
 
 ### 基本用法
 
@@ -351,6 +385,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
@@ -444,16 +547,42 @@ 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 |
@@ -465,6 +594,28 @@ moha download-file org/my-model model.safetensors \
 | `--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.2 → xiaoshiai_hub-1.1.4}/README.md

@@ -189,20 +189,53 @@ 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")
 ```
 
 ## 🔐 加密功能
@@ -288,7 +321,7 @@ export MOHA_ENCRYPTION_PASSWORD="your-encryption-password"
 
 ## 🖥️ 命令行工具 (CLI)
 
-SDK 提供了 `moha` 命令行工具，支持常见的上传下载操作。
+SDK 提供了 `moha` 命令行工具，支持登录认证、仓库管理、分支管理、上传下载等操作。
 
 ### 基本用法
 
@@ -296,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
@@ -389,16 +491,42 @@ 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 |
@@ -410,6 +538,28 @@ moha download-file org/my-model model.safetensors \
 | `--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.2 → xiaoshiai_hub-1.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "xiaoshiai-hub"
-version = "1.1.2"
+version = "1.1.4"
 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"
@@ -47,6 +47,7 @@ classifiers = [
 dependencies = [
     "requests>=2.20.0",
     "tqdm>=4.62.0",
+    "cryptography>=46.0.3"
 ]
 
 [project.optional-dependencies]

xiaoshiai_hub-1.1.4/requirements.txt

@@ -0,0 +1,3 @@
+requests>=2.20.0,<3.0.0
+tqdm>=4.62.0,<5.0.0
+cryptography>=46.0.3,<47.0.0

{xiaoshiai_hub-1.1.2 → xiaoshiai_hub-1.1.4}/setup.py

@@ -17,7 +17,7 @@ else:
     long_description = "Python SDK for XiaoShi AI Hub"
 
 # Read version from __init__.py
-version = "1.1.2"
+version = "1.1.4"
 init_file = Path(__file__).parent / "xiaoshiai_hub" / "__init__.py"
 if init_file.exists():
     with open(init_file, "r", encoding="utf-8") as f:
@@ -30,6 +30,7 @@ if init_file.exists():
 install_requires = [
     "requests>=2.20.0,<3.0.0",
     "tqdm>=4.62.0,<5.0.0",
+    "cryptography>=46.0.3,<47.0.0",
 ]
 
 # Optional dependencies