aap-utils 0.1.22__py3-none-any.whl → 0.2.0__py3-none-any.whl

aap_utils/__init__.py

@@ -1,3 +1,4 @@
-from .register import register_demo
-from .register import get_config
-from .register import get_mlflow_url
+from .register import register_demo, get_config, get_mlflow_url
+
+# CLI should not interfere with normal imports
+__all__ = ["register_demo", "get_config", "get_mlflow_url"]

aap_utils/cli.py

@@ -0,0 +1,40 @@
+import os
+
+def extract_default_image():
+    """Extracts the image name from KERNEL_IMAGE if available."""
+    kernel_image = os.getenv("KERNEL_IMAGE", "admin/base_kernel_gpu@sha256:3024")
+    return kernel_image.split("/")[1].split("@")[0]  # Extract "base_kernel_gpu"
+
+def validate_image_name(image_name):
+    """Validate image name format."""
+    return bool(image_name and image_name.strip())
+
+def validate_tags(tags):
+    """Ensure tags are non-empty and separated by semicolon."""
+    tag_list = [tag.strip() for tag in tags.split(";") if tag.strip()]
+    return len(tag_list) > 0
+
+def publish_env():
+    """Handles the 'aap_utils publish env' command."""
+    default_image = extract_default_image()
+    
+    while True:
+        image_name = input(f"Enter image name [{default_image}]: ").strip() or default_image
+        if validate_image_name(image_name):
+            break
+        print("Invalid image name. Please try again.")
+
+    while True:
+        tags = input("Enter tags (separated by ';'): ").strip()
+        if validate_tags(tags):
+            break
+        print("Invalid tags. Please enter at least one tag, separated by ';'.")
+
+    print(f"Publishing environment with Image: {image_name}, Tags: {tags}")
+
+def main():
+    import sys
+    if len(sys.argv) > 2 and sys.argv[1] == "publish" and sys.argv[2] == "env":
+        publish_env()
+    else:
+        print("Usage: aap_utils publish env")

aap_utils/register.py

@@ -92,4 +92,4 @@ def get_mlflow_url(domain_name = "aiplatform.vcntt.tech"):
     kernel_namespace = os.getenv("KERNEL_NAMESPACE", "default_namespace")  # Default if not set
     mlflow_url = f"https://{kernel_namespace.replace('machinelearning', 'appmachine')}.{domain_name}/private/mlflow"
 
-    return os.environ.get('MLFLOW_URI') mlflow_url
+    return os.environ.get('MLFLOW_URI'), mlflow_url

{aap_utils-0.1.22.dist-info → aap_utils-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aap_utils
-Version: 0.1.22
+Version: 0.2.0
 Summary: A Python library for registering an IP address for AAP
 Author: Minh Dang
 Author-email: danghoangminh86@gmail.com

aap_utils-0.2.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+aap_utils/__init__.py,sha256=g6G3i3a5ANDJWqFX2oPsLpP93aTaoTj0gkMQzeqQIJw,172
+aap_utils/cli.py,sha256=TSh2o6CpkiVodF4t88RA3LrAj_aymmb8szl5t7CWMT4,1410
+aap_utils/register.py,sha256=J9tukByU1mcy4SoaJOoGY8jjzL9JMuNTZGtCBSwwon0,3823
+aap_utils-0.2.0.dist-info/METADATA,sha256=SOAQ8YL8coe3-28zEDp7izDbgB5UcLd7GeLSEBnoLFo,346
+aap_utils-0.2.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+aap_utils-0.2.0.dist-info/entry_points.txt,sha256=K-h2toLoqFxAg4Bx2r7FWggKgYnAX9zp9s4Ms4r-d6s,49
+aap_utils-0.2.0.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
+aap_utils-0.2.0.dist-info/RECORD,,

aap_utils-0.2.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+aap_utils = aap_utils.cli:main

aap_utils-0.1.22.dist-info/RECORD

@@ -1,6 +0,0 @@
-aap_utils/__init__.py,sha256=Px9phm-1DU2yq-wUjKg9zEmRWbcLpWAYG-9AlnNVmCc,105
-aap_utils/register.py,sha256=-vftTcE0Ii8TQnaev5cH3vpxPvlDQjnin-OBtgl7be4,3822
-aap_utils-0.1.22.dist-info/METADATA,sha256=ISZYkmaRoSAvlJ0aj1T90OD0pD6SQYWU8O2xa7eaXEY,347
-aap_utils-0.1.22.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-aap_utils-0.1.22.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
-aap_utils-0.1.22.dist-info/RECORD,,