viettelcloud-aiplatform 0.3.0__py3-none-any.whl

viettelcloud/__init__.py

@@ -0,0 +1 @@
+# Namespace package for viettelcloud

viettelcloud/aiplatform/__init__.py

@@ -0,0 +1,15 @@
+# Copyright 2025 The Kubeflow Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+__version__ = "0.3.0"

viettelcloud/aiplatform/common/constants.py

@@ -0,0 +1,22 @@
+# Copyright 2025 The Kubeflow Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# The default Kubernetes namespace.
+DEFAULT_NAMESPACE = "default"
+
+# How long to wait in seconds for requests to the Kubernetes API Server.
+DEFAULT_TIMEOUT = 120
+
+# Unknown indicates that the value can't be identified.
+UNKNOWN = "Unknown"

viettelcloud/aiplatform/common/types.py

@@ -0,0 +1,28 @@
+# Copyright 2025 The Kubeflow Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from typing import Optional
+
+from kubernetes import client
+from pydantic import BaseModel
+
+
+class KubernetesBackendConfig(BaseModel):
+    namespace: Optional[str] = None
+    config_file: Optional[str] = None
+    context: Optional[str] = None
+    client_configuration: Optional[client.Configuration] = None
+
+    class Config:
+        arbitrary_types_allowed = True

viettelcloud/aiplatform/common/utils.py

@@ -0,0 +1,40 @@
+# Copyright 2025 The Kubeflow Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import os
+from typing import Optional
+
+from kubernetes import config
+
+from viettelcloud.aiplatform.common import constants
+
+
+def is_running_in_k8s() -> bool:
+    return os.path.isdir("/var/run/secrets/kubernetes.io/")
+
+
+def get_default_target_namespace(context: Optional[str] = None) -> str:
+    if not is_running_in_k8s():
+        try:
+            all_contexts, current_context = config.list_kube_config_contexts()
+            # If context is set, we should get namespace from it.
+            if context:
+                for c in all_contexts:
+                    if isinstance(c, dict) and c.get("name") == context:
+                        return c["context"]["namespace"]
+            # Otherwise, try to get namespace from the current context.
+            return current_context["context"]["namespace"]
+        except Exception:
+            return constants.DEFAULT_NAMESPACE
+    with open("/var/run/secrets/kubernetes.io/serviceaccount/namespace") as f:
+        return f.readline()

viettelcloud/aiplatform/hub/OWNERS

@@ -0,0 +1,14 @@
+approvers:
+  - Al-Pragliola
+  - ederign
+  - jonburdo
+  - pboyd
+  - rareddy
+  - tarilabs
+emeritus_approvers:
+  - andreyvelich
+  - ckadner
+  - Tomcli
+  - zijianjoy
+reviewers:
+  - fege

viettelcloud/aiplatform/hub/__init__.py

@@ -0,0 +1,25 @@
+# Copyright 2025 The Kubeflow Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from viettelcloud.aiplatform.hub.api.model_registry_client import ModelRegistryClient
+from viettelcloud.aiplatform.hub.api._proxy_client import (
+    AuthenticationError,
+    PermissionDeniedError,
+)
+
+__all__ = [
+    "ModelRegistryClient",
+    "AuthenticationError",
+    "PermissionDeniedError",
+]

viettelcloud/aiplatform/hub/api/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 2025 The Kubeflow Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

viettelcloud/aiplatform/hub/api/_proxy_client.py

@@ -0,0 +1,355 @@
+# Copyright 2025 The Kubeflow Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Internal HTTP client for cmp-backend proxy mode."""
+
+from __future__ import annotations
+
+import logging
+from typing import Any, Iterator, Optional
+
+import requests
+from requests.adapters import HTTPAdapter
+from urllib3.util.retry import Retry
+
+LOG = logging.getLogger(__name__)
+
+
+class AuthenticationError(Exception):
+    """Raised when PAT token is invalid or expired."""
+
+    pass
+
+
+class PermissionDeniedError(Exception):
+    """Raised when user lacks permission for the operation."""
+
+    pass
+
+
+class ProxyHTTPClient:
+    """
+    HTTP client for cmp-backend Model Registry proxy.
+
+    This client handles communication with cmp-backend which:
+    - Validates PAT tokens
+    - Enforces RBAC permissions
+    - Routes to the correct MR instance based on project and region
+    """
+
+    def __init__(
+        self,
+        cmp_backend_url: str,
+        pat_token: str,
+        project_id: str,
+        region: str,
+        timeout: float = 30,
+    ):
+        """
+        Initialize proxy client.
+
+        Args:
+            cmp_backend_url: Base URL of cmp-backend (e.g., https://api.viettelcloud.vn)
+            pat_token: Personal Access Token for authentication
+            project_id: Project UUID or slug
+            region: Region UUID or name (e.g., "HN", "HCM")
+            timeout: Request timeout in seconds
+        """
+        self._base_url = f"{cmp_backend_url.rstrip('/')}/v2/model-registry-proxy/proxy"
+        self._project_id = project_id
+        self._region = region
+        self._timeout = timeout
+
+        self._session = requests.Session()
+        self._session.headers.update({
+            "Authorization": f"Token {pat_token}",
+            "X-Project-ID": project_id,
+            "X-Region": region,
+            "Content-Type": "application/json",
+            "Accept": "application/json",
+        })
+
+        # Retry configuration
+        retries = Retry(
+            total=3,
+            backoff_factor=0.5,
+            status_forcelist=[502, 503, 504],
+        )
+        adapter = HTTPAdapter(
+            pool_connections=10,
+            pool_maxsize=10,
+            max_retries=retries,
+        )
+        self._session.mount("http://", adapter)
+        self._session.mount("https://", adapter)
+
+    def __del__(self):
+        if hasattr(self, "_session") and self._session:
+            try:
+                self._session.close()
+            except Exception:
+                pass
+
+    def _request(
+        self,
+        method: str,
+        path: str,
+        body: Optional[dict] = None,
+        params: Optional[dict] = None,
+    ) -> requests.Response:
+        """
+        Send request to cmp-backend proxy.
+
+        Args:
+            method: HTTP method
+            path: API path
+            body: Request body (JSON)
+            params: Query parameters
+
+        Returns:
+            Response object
+
+        Raises:
+            AuthenticationError: If PAT token is invalid/expired
+            PermissionDeniedError: If user lacks permission
+            requests.HTTPError: For other HTTP errors
+        """
+        url = f"{self._base_url}{path}"
+
+        try:
+            LOG.debug(f"Proxy request: {method} {url}")
+            resp = self._session.request(
+                method,
+                url,
+                json=body,
+                params=params,
+                timeout=self._timeout,
+            )
+
+            # Handle auth errors
+            if resp.status_code == 401:
+                raise AuthenticationError("PAT token is invalid or expired")
+            if resp.status_code == 403:
+                error_msg = "Permission denied"
+                try:
+                    error_msg = resp.json().get("detail", error_msg)
+                except Exception:
+                    pass
+                raise PermissionDeniedError(error_msg)
+
+            resp.raise_for_status()
+            return resp
+
+        except requests.exceptions.ConnectionError as e:
+            LOG.error(f"Connection error: {url} - {e}")
+            raise ConnectionError(f"Failed to connect to cmp-backend: {e}") from e
+
+        except requests.exceptions.Timeout as e:
+            LOG.error(f"Timeout: {url} - {e}")
+            raise TimeoutError(f"Request to cmp-backend timed out: {e}") from e
+
+    def _get(self, path: str, params: Optional[dict] = None) -> Any:
+        """GET request, returns JSON."""
+        resp = self._request("GET", path, params=params)
+        return resp.json()
+
+    def _post(self, path: str, body: Optional[dict] = None) -> Any:
+        """POST request, returns JSON."""
+        resp = self._request("POST", path, body=body)
+        return resp.json()
+
+    def _patch(self, path: str, body: Optional[dict] = None) -> Any:
+        """PATCH request, returns JSON."""
+        resp = self._request("PATCH", path, body=body)
+        return resp.json()
+
+    def _delete(self, path: str) -> None:
+        """DELETE request."""
+        self._request("DELETE", path)
+
+    # =========================================================================
+    # Registered Models
+    # =========================================================================
+
+    def list_registered_models(
+        self,
+        page_size: Optional[int] = None,
+        order_by: Optional[str] = None,
+        next_page_token: Optional[str] = None,
+    ) -> dict:
+        """
+        List registered models.
+
+        Args:
+            page_size: Maximum number of items to return
+            order_by: Sort order (e.g., "name", "-createTime")
+            next_page_token: Token for pagination
+
+        Returns:
+            Dict with 'items' and optional 'nextPageToken'
+        """
+        params = {}
+        if page_size:
+            params["pageSize"] = page_size
+        if order_by:
+            params["orderBy"] = order_by
+        if next_page_token:
+            params["nextPageToken"] = next_page_token
+        return self._get("/registered_models/", params=params)
+
+    def get_registered_model(self, model_id: str) -> dict:
+        """Get a registered model by ID or name."""
+        return self._get(f"/registered_models/{model_id}/")
+
+    def create_registered_model(
+        self,
+        name: str,
+        description: Optional[str] = None,
+        owner: Optional[str] = None,
+        custom_properties: Optional[dict] = None,
+    ) -> dict:
+        """Create a registered model."""
+        data = {"name": name}
+        if description:
+            data["description"] = description
+        if owner:
+            data["owner"] = owner
+        if custom_properties:
+            data["customProperties"] = custom_properties
+        return self._post("/registered_models/", body=data)
+
+    def update_registered_model(
+        self,
+        model_id: str,
+        description: Optional[str] = None,
+        custom_properties: Optional[dict] = None,
+    ) -> dict:
+        """Update a registered model."""
+        data = {}
+        if description is not None:
+            data["description"] = description
+        if custom_properties is not None:
+            data["customProperties"] = custom_properties
+        return self._patch(f"/registered_models/{model_id}/", body=data)
+
+    def delete_registered_model(self, model_id: str) -> None:
+        """Delete a registered model."""
+        self._delete(f"/registered_models/{model_id}/")
+
+    # =========================================================================
+    # Model Versions
+    # =========================================================================
+
+    def list_model_versions(
+        self,
+        model_name: str,
+        page_size: Optional[int] = None,
+        order_by: Optional[str] = None,
+        next_page_token: Optional[str] = None,
+    ) -> dict:
+        """List versions of a model."""
+        params = {}
+        if page_size:
+            params["pageSize"] = page_size
+        if order_by:
+            params["orderBy"] = order_by
+        if next_page_token:
+            params["nextPageToken"] = next_page_token
+        return self._get(f"/registered_models/{model_name}/versions/", params=params)
+
+    def get_model_version(self, model_name: str, version: str) -> dict:
+        """Get a specific model version."""
+        return self._get(f"/registered_models/{model_name}/versions/{version}/")
+
+    def create_model_version(
+        self,
+        model_name: str,
+        version_name: str,
+        description: Optional[str] = None,
+        author: Optional[str] = None,
+        custom_properties: Optional[dict] = None,
+    ) -> dict:
+        """Create a model version."""
+        data = {"name": version_name}
+        if description:
+            data["description"] = description
+        if author:
+            data["author"] = author
+        if custom_properties:
+            data["customProperties"] = custom_properties
+        return self._post(f"/registered_models/{model_name}/versions/", body=data)
+
+    def update_model_version(
+        self,
+        model_name: str,
+        version: str,
+        description: Optional[str] = None,
+        custom_properties: Optional[dict] = None,
+    ) -> dict:
+        """Update a model version."""
+        data = {}
+        if description is not None:
+            data["description"] = description
+        if custom_properties is not None:
+            data["customProperties"] = custom_properties
+        return self._patch(f"/registered_models/{model_name}/versions/{version}/", body=data)
+
+    # =========================================================================
+    # Model Artifacts
+    # =========================================================================
+
+    def get_model_artifact(self, artifact_id: str) -> dict:
+        """Get a model artifact."""
+        return self._get(f"/model_artifacts/{artifact_id}/")
+
+    def create_model_artifact(
+        self,
+        name: str,
+        uri: str,
+        description: Optional[str] = None,
+        model_format_name: Optional[str] = None,
+        model_format_version: Optional[str] = None,
+        storage_key: Optional[str] = None,
+        storage_path: Optional[str] = None,
+        custom_properties: Optional[dict] = None,
+    ) -> dict:
+        """Create a model artifact."""
+        data = {"name": name, "uri": uri}
+        if description:
+            data["description"] = description
+        if model_format_name:
+            data["modelFormatName"] = model_format_name
+        if model_format_version:
+            data["modelFormatVersion"] = model_format_version
+        if storage_key:
+            data["storageKey"] = storage_key
+        if storage_path:
+            data["storagePath"] = storage_path
+        if custom_properties:
+            data["customProperties"] = custom_properties
+        return self._post("/model_artifacts/", body=data)
+
+    def update_model_artifact(
+        self,
+        artifact_id: str,
+        description: Optional[str] = None,
+        custom_properties: Optional[dict] = None,
+    ) -> dict:
+        """Update a model artifact."""
+        data = {}
+        if description is not None:
+            data["description"] = description
+        if custom_properties is not None:
+            data["customProperties"] = custom_properties
+        return self._patch(f"/model_artifacts/{artifact_id}/", body=data)