Add support for skill repository type (#121)

* Add skill repo type support.

* update version

---------

Co-authored-by: Jun Luo <jun.luo@opencsg.com>

Author: luojun96

README.md

@@ -25,7 +25,7 @@ Open Source: Dive into the source code, contribute, and customize the SDK to fit
 
 The main functions are:
 
-1. Repo downloading（model/dataset）
+1. Repo downloading（model/dataset/space/code/mcp/skill）
 2. Repo information query（Compatible with huggingface）
 
 **XNet Accelerated Transfer (New!)**: Next-generation storage and version control technology for large-scale AI/ML data.

README_cn.md

@@ -25,7 +25,7 @@ CSGHub SDK 是一个强大的 Python 客户端，专门设计用于与 CSGHub 
 
 主要功能包括：
 
-1. 仓库下载（模型/数据集）
+1. 仓库下载（模型/数据集/应用/代码/MCP/技能）
 2. 仓库信息查询（与huggingface兼容）
 
 **XNet 加速传输（新！）**：面向大规模 AI/ML 数据的下一代存储与版本控制技术。

pycsghub/api_client/__init__.py

@@ -1,6 +1,6 @@
 from typing import Optional
 import logging
-from pycsghub.constants import REPO_TYPE_CODE, REPO_TYPE_MCPSERVER
+from pycsghub.constants import REPO_TYPE_CODE, REPO_TYPE_MCPSERVER, REPO_TYPE_SKILL
 from pycsghub.utils import disable_xnet, get_endpoint, get_token_to_send
 from .api_client import CsghubApi
 from .api_client_interface import HubApi
@@ -16,7 +16,7 @@ def get_csghub_api(repo_type: Optional[RepoType] = None,
     token = get_token_to_send(token)
     endpoint = get_endpoint(endpoint=endpoint)
 
-    if repo_type in [REPO_TYPE_CODE, REPO_TYPE_MCPSERVER]:
+    if repo_type in [REPO_TYPE_CODE, REPO_TYPE_MCPSERVER, REPO_TYPE_SKILL]:
         logger.debug(f"Use CsghubApi for repo_type {repo_type}")
         return CsghubApi(token=token, endpoint=endpoint, user_name=user_name)
 

pycsghub/cmd/repo_types.py

@@ -1,9 +1,10 @@
 from enum import Enum
-from pycsghub.constants import REPO_TYPE_MODEL, REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER
+from pycsghub.constants import REPO_TYPE_MODEL, REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER, REPO_TYPE_SKILL
 
 class RepoType(str, Enum):
     MODEL = REPO_TYPE_MODEL
     DATASET = REPO_TYPE_DATASET
     SPACE = REPO_TYPE_SPACE
     CODE = REPO_TYPE_CODE
     MCP = REPO_TYPE_MCPSERVER
+    SKILL = REPO_TYPE_SKILL

pycsghub/constants.py

@@ -8,7 +8,8 @@
 REPO_TYPE_SPACE = "space"
 REPO_TYPE_CODE = "code"
 REPO_TYPE_MCPSERVER = "mcp"
-REPO_TYPES = [None, REPO_TYPE_MODEL, REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER]
+REPO_TYPE_SKILL = "skill"
+REPO_TYPES = [None, REPO_TYPE_MODEL, REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER, REPO_TYPE_SKILL]
 
 REPO_SOURCE_CSG = "csg"
 REPO_SOURCE_HF = "hf"

pycsghub/test/repo_integration_test.py

@@ -98,6 +98,9 @@ def test_from_env_config(self):
             elif type_str == "code":
                 repo_type = RepoType.CODE
                 filename = f"test_code_{unique_suffix}.txt"
+            elif type_str == "skill":
+                repo_type = RepoType.SKILL
+                filename = f"test_skill_{unique_suffix}.txt"
             else:
                 print(f"Unknown repo type: {type_str}")
                 continue

pycsghub/upload_large_folder/main.py

@@ -7,7 +7,7 @@
 from typing import Optional, Union, List
 from pycsghub.cmd.repo_types import RepoType
 from pycsghub.utils import check_repo_type
-from pycsghub.constants import REPO_TYPE_MODEL, REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER
+from pycsghub.constants import REPO_TYPE_MODEL, REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER, REPO_TYPE_SKILL
 from pycsghub.utils import get_endpoint
 from .path import filter_repo_objects
 from .local_folder import get_local_upload_paths, read_upload_metadata

pycsghub/utils.py

@@ -4,7 +4,7 @@
 import os
 from pycsghub.constants import MODEL_ID_SEPARATOR, DEFAULT_CSG_GROUP, DEFAULT_CSGHUB_DOMAIN
 from pycsghub.constants import OPERATION_ACTION_API, OPERATION_ACTION_GIT, XNET_API_PATH
-from pycsghub.constants import REPO_TYPE_MODEL, REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER
+from pycsghub.constants import REPO_TYPE_MODEL, REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER, REPO_TYPE_SKILL
 from pycsghub.constants import REPO_SOURCE_CSG, REPO_SOURCE_HF, REPO_SOURCE_MS, REPO_SOURCE_XET
 import requests
 from huggingface_hub.hf_api import ModelInfo, DatasetInfo, SpaceInfo
@@ -17,6 +17,7 @@
 import logging
 from huggingface_hub.hf_api import ModelInfo as CodeInfo
 from huggingface_hub.hf_api import ModelInfo as McpserverInfo
+from huggingface_hub.hf_api import ModelInfo as SkillInfo
 
 logger = logging.getLogger(__name__)
 
@@ -159,9 +160,11 @@ def get_repo_info(
         method = code_info
     elif repo_type == REPO_TYPE_MCPSERVER:
         method = mcpserver_info
+    elif repo_type == REPO_TYPE_SKILL:
+        method = skill_info
     else:
         raise ValueError(f"Unsupported repo type {repo_type}.")
-    
+
     logger.debug(f"get repo info {repo_id} {revision} {repo_type} {token} {endpoint} {source}")
 
     return method(
@@ -428,6 +431,35 @@ def mcpserver_info(
     data = r.json()
     return McpserverInfo(**data)
 
+def skill_info(
+    repo_id: str,
+    *,
+    revision: Optional[str] = None,
+    timeout: Optional[float] = None,
+    securityStatus: Optional[bool] = None,
+    files_metadata: bool = False,
+    token: Union[bool, str, None] = None,
+    endpoint: Optional[str] = None,
+    source: Optional[str] = None,
+) -> SkillInfo:
+    headers = build_csg_headers(token=token)
+    path = get_repo_meta_path(repo_type=REPO_TYPE_SKILL,
+                              repo_id=repo_id,
+                              revision=revision,
+                              endpoint=endpoint,
+                              source=source)
+    params = {}
+    if securityStatus:
+        params["securityStatus"] = True
+    if files_metadata:
+        params["blobs"] = True
+    r = requests.get(path, headers=headers, timeout=timeout, params=params)
+    if r.status_code != 200:
+        logger.error(f"get skill meta info from {path} response: {r.text}")
+    r.raise_for_status()
+    data = r.json()
+    return SkillInfo(**data)
+
 def get_repo_meta_path(
     repo_type: str, 
     repo_id: str, 
@@ -583,17 +615,17 @@ def print_download_result(res):
         print(tabulate(items, headers=columns))
 
 def check_repo_type(repo_type: str):
-    if repo_type not in [REPO_TYPE_MODEL, REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER]:
-        raise ValueError(f"invalid repo type, must be one of {REPO_TYPE_MODEL} or {REPO_TYPE_DATASET} or {REPO_TYPE_SPACE} or {REPO_TYPE_CODE} or {REPO_TYPE_MCPSERVER}")
+    if repo_type not in [REPO_TYPE_MODEL, REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER, REPO_TYPE_SKILL]:
+        raise ValueError(f"invalid repo type, must be one of {REPO_TYPE_MODEL} or {REPO_TYPE_DATASET} or {REPO_TYPE_SPACE} or {REPO_TYPE_CODE} or {REPO_TYPE_MCPSERVER} or {REPO_TYPE_SKILL}")
 
 def get_repo_url_prefix(repo_type: Optional[str] = None):
-    if repo_type in [REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER]:
+    if repo_type in [REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_MCPSERVER, REPO_TYPE_SKILL]:
         return f"{repo_type}s"
     else:
         return f"{REPO_TYPE_MODEL}s"
 
 def get_repo_git_prefix(repo_type: Optional[str] = None):
-    if repo_type in [REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE]:
+    if repo_type in [REPO_TYPE_DATASET, REPO_TYPE_SPACE, REPO_TYPE_CODE, REPO_TYPE_SKILL]:
         return f"{repo_type}s"
     elif repo_type in [REPO_TYPE_MCPSERVER]:
         return f"mcpservers"

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "csghub-sdk"
-version = "0.8.3"
+version = "0.8.4"
 description = "CSGHub SDK for downloading and uploading models, datasets, and spaces"
 readme = "README.md"
 license = { text = "Apache-2.0" }