aioli-sdk 1.3.0.dev12__tar.gz → 1.3.0.dev14__tar.gz

{aioli_sdk-1.3.0.dev12 → aioli_sdk-1.3.0.dev14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aioli-sdk
-Version: 1.3.0.dev12
+Version: 1.3.0.dev14
 Summary: Aioli (AI OnLine Inference), a platform for deploying AI models at scale.
 Home-page: https://github.com/determined-ai/aioli
 Author: HPE AI Solutions

{aioli_sdk-1.3.0.dev12 → aioli_sdk-1.3.0.dev14}/aioli/__version__.py

@@ -1,2 +1,2 @@
 # © Copyright 2024 Hewlett Packard Enterprise Development LP
-__version__ = "1.3.0-dev12"
+__version__ = "1.3.0-dev14"

{aioli_sdk-1.3.0.dev12 → aioli_sdk-1.3.0.dev14}/aioli/cli/deployment.py

@@ -48,6 +48,7 @@ def show_deployment(args: Namespace) -> None:
     d = deployment.to_dict()
     # Remove clusterName for now - INF-243
     d.pop("clusterName")
+    d.pop("project")  # something for INF-1516
 
     # Get the model version for the deployment
     deployment_model_versions = packaged_models_api.models_versions_get(deployment.model)
@@ -83,6 +84,7 @@ def format_json(
         # Don't use the d.to_json() method as it adds backslash escapes for double quote
         m_dict = d.to_dict()
         m_dict.pop("modifiedAt")
+        m_dict.pop("project")  # something for INF-1516
         # Use model name instead of id
         model = model_api.models_id_get(d.model)
         deployment_model_versions = model_api.models_versions_get(model.name)

{aioli_sdk-1.3.0.dev12 → aioli_sdk-1.3.0.dev14}/aioli/cli/model.py

@@ -59,6 +59,7 @@ def format_json(
         d = m.to_dict()
         d.pop("id")
         d.pop("modifiedAt")
+        d.pop("project")  # something for INF-1516
         rname = lookup_registry_name_by_id(m.registry, registries_api)
         d["registry"] = rname
         models.append(d)
@@ -215,6 +216,7 @@ def show_model(args: Namespace) -> None:
 
     d = model.to_dict()
     d["registry"] = rname
+    d.pop("project")  # something for INF-1516
 
     if args.json:
         render.print_json(d)

{aioli_sdk-1.3.0.dev12 → aioli_sdk-1.3.0.dev14}/aioli/cli/registry.py

@@ -41,6 +41,7 @@ def format_json(response: List[TrainedModelRegistry]) -> List[Dict[str, str]]:
         d = r.to_dict()
         d.pop("id")
         d.pop("modifiedAt")
+        d.pop("project")  # something for INF-1516
         regs.append(d)
 
     return regs
@@ -121,6 +122,7 @@ def show_registry(args: Namespace) -> None:
     registry = lookup_registry(args.name, api_instance)
 
     d = registry.to_dict()
+    d.pop("project")  # something for INF-1516
     if args.json:
         render.print_json(d)
     else:
@@ -194,8 +196,10 @@ def format_model(r: aiolirest.TrainedModelRegistry, m: aiolirest.ModelResponse)
         # suppose that may change in time.
         # latest_version_id_str is in the calculated uri, so don't show that.
         # format shows a range of values, but for our users all are 'openllm'.
-        uri = f"openllm://{m.name}/{m.latest_version_id_str}"
-        return [uri, description]
+        return [m.url, description]
+    if r.type == "huggingface":
+        downloads = m.metadata.get("downloads") if m.metadata else None
+        return [m.latest_version_id_str, m.url, downloads, m.format]
     if r.type == "ngc":
         return [
             m.display_name,
@@ -265,6 +269,9 @@ def list_models(args: Namespace) -> None:
             response = api_instance.registries_id_models_get(registry.id, model_format)
             for m in response:
                 models.append(m)
+    elif registry.type == "huggingface":
+        headers = ["Name", "URL", "Downloads", "Model Type"]
+        models = api_instance.registries_id_models_get(registry.id, args.search)
     else:
         models = api_instance.registries_id_models_get(registry.id)
     if registry.type == "pfs":
@@ -375,6 +382,11 @@ main_cmd = Cmd(
             "lists the available models in a registry (for certain registry types)",
             [
                 Arg("name", help="The name of the model registry"),
+                Arg(
+                    "--search",
+                    help="The optional string value that will be contained in the returned "
+                    + "model names (huggingface only)",
+                ),
                 Group(
                     Arg("--csv", action="store_true", help="print as CSV"),
                     Arg("--json", action="store_true", help="print as JSON"),

{aioli_sdk-1.3.0.dev12 → aioli_sdk-1.3.0.dev14}/aioli/cli/test/test_cli.py

@@ -160,7 +160,7 @@ class TestCli:
         expected = (
             "Failed to lists the available models in a registry (for certain registry types): "
             "a registry of type s3 is unable to provide a list of supported models. "
-            "Supported registries: (openllm, ngc)"
+            "Supported registries: (openllm, ngc, huggingface, pfs)"
         )
         outcome = subprocess.run(
             ["aioli", "registry", "models", "bento-registry"], capture_output=True

{aioli_sdk-1.3.0.dev12 → aioli_sdk-1.3.0.dev14}/aioli/cli/user.py

@@ -1,7 +1,7 @@
 # © Copyright 2024 Hewlett Packard Enterprise Development LP
 import getpass
 from argparse import Namespace
-from typing import Any, List
+from typing import Any, Dict, List
 
 import aiolirest
 from aioli import cli
@@ -26,15 +26,26 @@ def list_users(args: Namespace) -> None:
         return result
 
     if args.json:
-        render.print_json([r.to_dict() for r in response])
+        render.print_json(format_json(response))
     elif args.yaml:
-        print(render.format_object_as_yaml([r.to_dict() for r in response]))
+        print(render.format_object_as_yaml(format_json(response)))
     else:
         headers = ["Username", "Display Name", "Active", "Remote"]
         values = [format_user(u) for u in response]
         render.tabulate_or_csv(headers, values, args.csv)
 
 
+def format_json(response: List[User]) -> List[Dict[str, str]]:
+    users = []
+    for r in response:
+        # Don't use the r.to_json() method as it adds backslash escapes for double quote
+        d = r.to_dict()
+        d.pop("defaultProject")  # something for INF-1516
+        users.append(d)
+
+    return users
+
+
 @authentication.required
 def activate_user(parsed_args: Namespace) -> None:
     patch = UserPatchRequest(active=True)

{aioli_sdk-1.3.0.dev12 → aioli_sdk-1.3.0.dev14}/aioli_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aioli-sdk
-Version: 1.3.0.dev12
+Version: 1.3.0.dev14
 Summary: Aioli (AI OnLine Inference), a platform for deploying AI models at scale.
 Home-page: https://github.com/determined-ai/aioli
 Author: HPE AI Solutions

{aioli_sdk-1.3.0.dev12 → aioli_sdk-1.3.0.dev14}/aioli_sdk.egg-info/SOURCES.txt

@@ -52,6 +52,7 @@ aiolirest/api/authentication_api.py
 aiolirest/api/deployments_api.py
 aiolirest/api/information_api.py
 aiolirest/api/packaged_models_api.py
+aiolirest/api/projects_api.py
 aiolirest/api/registries_api.py
 aiolirest/api/roles_api.py
 aiolirest/api/templates_api.py
@@ -83,6 +84,8 @@ aiolirest/models/model_response.py
 aiolirest/models/observability.py
 aiolirest/models/packaged_model.py
 aiolirest/models/packaged_model_request.py
+aiolirest/models/project.py
+aiolirest/models/project_request.py
 aiolirest/models/resource_profile.py
 aiolirest/models/resources_template.py
 aiolirest/models/resources_template_request.py
@@ -96,4 +99,5 @@ aiolirest/models/trained_model_registry.py
 aiolirest/models/trained_model_registry_request.py
 aiolirest/models/user.py
 aiolirest/models/user_patch_request.py
-aiolirest/models/user_request.py
+aiolirest/models/user_request.py
+aiolirest/models/user_session.py

{aioli_sdk-1.3.0.dev12 → aioli_sdk-1.3.0.dev14}/aiolirest/__init__.py

@@ -7,7 +7,7 @@
 
     HPE MLIS is *Aioli* -- The AI On-line Inference Platform that enables easy deployment, tracking, and serving of your packaged models regardless of your preferred AI framework.
 
-    The version of the OpenAPI document: 1.3.0-dev12
+    The version of the OpenAPI document: 1.3.0-dev14
     Contact: community@determined-ai
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -22,6 +22,7 @@ from aiolirest.api.authentication_api import AuthenticationApi
 from aiolirest.api.deployments_api import DeploymentsApi
 from aiolirest.api.information_api import InformationApi
 from aiolirest.api.packaged_models_api import PackagedModelsApi
+from aiolirest.api.projects_api import ProjectsApi
 from aiolirest.api.registries_api import RegistriesApi
 from aiolirest.api.roles_api import RolesApi
 from aiolirest.api.templates_api import TemplatesApi
@@ -65,6 +66,8 @@ from aiolirest.models.model_response import ModelResponse
 from aiolirest.models.observability import Observability
 from aiolirest.models.packaged_model import PackagedModel
 from aiolirest.models.packaged_model_request import PackagedModelRequest
+from aiolirest.models.project import Project
+from aiolirest.models.project_request import ProjectRequest
 from aiolirest.models.resource_profile import ResourceProfile
 from aiolirest.models.resources_template import ResourcesTemplate
 from aiolirest.models.resources_template_request import ResourcesTemplateRequest
@@ -79,3 +82,4 @@ from aiolirest.models.trained_model_registry_request import TrainedModelRegistry
 from aiolirest.models.user import User
 from aiolirest.models.user_patch_request import UserPatchRequest
 from aiolirest.models.user_request import UserRequest
+from aiolirest.models.user_session import UserSession

{aioli_sdk-1.3.0.dev12 → aioli_sdk-1.3.0.dev14}/aiolirest/api/__init__.py

@@ -5,6 +5,7 @@ from aiolirest.api.authentication_api import AuthenticationApi
 from aiolirest.api.deployments_api import DeploymentsApi
 from aiolirest.api.information_api import InformationApi
 from aiolirest.api.packaged_models_api import PackagedModelsApi
+from aiolirest.api.projects_api import ProjectsApi
 from aiolirest.api.registries_api import RegistriesApi
 from aiolirest.api.roles_api import RolesApi
 from aiolirest.api.templates_api import TemplatesApi