aap-utils 0.2.16__py3-none-any.whl → 0.2.17__py3-none-any.whl

aap_utils/cli.py

@@ -91,9 +91,24 @@ def publish_env():
     except requests.RequestException as e:
         print(f"❌ Error connecting to the server: {e}")
 
+        
+def get_ssh_password_if_prompted(prompt_bytes):
+    return "aiplatform"
+
+
 def main():
     import sys
     if len(sys.argv) > 2 and sys.argv[1] == "push" and sys.argv[2] == "env":
         publish_env()
+    elif len(sys.argv) > 1 and sys.argv[1] == "b":
+        # Example: python aap_utils.py b "\raiplatform@172.20.0.11's password:\n"
+        if len(sys.argv) < 3:
+            print("Usage: aap_utils b \"<prompt_bytes>\"")
+            return
+        prompt = sys.argv[2].encode()
+        password = get_ssh_password_if_prompted(prompt)
+        print(password)
     else:
-        print("Usage: aap_utils push env")
+        print("Usage:")
+        print("  aap_utils push env")
+        print("  aap_utils b \"<prompt_bytes>\"")

aap_utils/register.py

@@ -47,14 +47,16 @@ def create_url(domain_name = "aiplatform.vcntt.tech"):
     kernel_namespace = os.getenv("KERNEL_NAMESPACE")
     
     # Validate the environment variable
-    if not kernel_namespace:
-        kernel_namespace="undefined"
+    if not kernel_namespace:        
+        kernel_namespace = os.getenv("ARGOWORKFLOW_NAMESPACE")
+        if not kernel_namespace:        
+            kernel_namespace="undefined"
     
     # Replace 'machinelearning' with 'appmachinepublic' in KERNEL_NAMESPACE
     if kernel_namespace.startswith("machinelearning"):
         transformed_namespace = kernel_namespace.replace("machinelearning", "appmachinepublic", 1)
     else:
-        raise ValueError(f"The KERNEL_NAMESPACE '{kernel_namespace}' does not follow the expected format.")
+        raise ValueError(f"The KERNEL_NAMESPACE or ARGOWORKFLOW_NAMESPACE'{kernel_namespace}' does not follow the expected format.")
     
     # Create the final URL
     url = f"https://{transformed_namespace}.{domain_name}"

{aap_utils-0.2.16.dist-info → aap_utils-0.2.17.dist-info}/METADATA

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

aap_utils-0.2.17.dist-info/RECORD

@@ -0,0 +1,8 @@
+aap_utils/__init__.py,sha256=g6G3i3a5ANDJWqFX2oPsLpP93aTaoTj0gkMQzeqQIJw,172
+aap_utils/cli.py,sha256=h0bwNdqpmrgCvApJx6votPSbEpdPbBZAEgdfKVqHu2w,4229
+aap_utils/register.py,sha256=0okAQ8b8wVtT_bSFCAsV3pcVOTifLHfkyCZzvLSpsYc,3964
+aap_utils-0.2.17.dist-info/METADATA,sha256=vJ9-rAB_9KE-Agy2tIgr8JqMTiUhcLwGMhHG2OIUv40,373
+aap_utils-0.2.17.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+aap_utils-0.2.17.dist-info/entry_points.txt,sha256=K-h2toLoqFxAg4Bx2r7FWggKgYnAX9zp9s4Ms4r-d6s,49
+aap_utils-0.2.17.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
+aap_utils-0.2.17.dist-info/RECORD,,

aap_utils-0.2.16.dist-info/RECORD

@@ -1,8 +0,0 @@
-aap_utils/__init__.py,sha256=g6G3i3a5ANDJWqFX2oPsLpP93aTaoTj0gkMQzeqQIJw,172
-aap_utils/cli.py,sha256=IEpuB7CH9OA8-JlFGuKpd_fZWvDh873RN9Ydl4Qf2dQ,3718
-aap_utils/register.py,sha256=J9tukByU1mcy4SoaJOoGY8jjzL9JMuNTZGtCBSwwon0,3823
-aap_utils-0.2.16.dist-info/METADATA,sha256=PbgLx3I979kUbMsK9VplpvgT5OHgDBvSmQUmsMSVnNs,373
-aap_utils-0.2.16.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-aap_utils-0.2.16.dist-info/entry_points.txt,sha256=K-h2toLoqFxAg4Bx2r7FWggKgYnAX9zp9s4Ms4r-d6s,49
-aap_utils-0.2.16.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
-aap_utils-0.2.16.dist-info/RECORD,,