aap-utils 0.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

aap_utils/register.py

@@ -1,6 +1,7 @@
 import socket
 import requests
 import time
+import os
 
 def register_demo(debug=False):
     for attempt in range(10):
@@ -12,6 +13,9 @@ def register_demo(debug=False):
                 headers={"Content-Type": "application/json"}
             )
             if response.ok:
+                url, namespace = create_url()
+                # print(f"The demo service is at {namespace}.your_custorm_domain/demo/. Eg: {url}")
+                print(f"The demo service is at {url}")
                 if debug:
                     print("IP registered successfully.")
                 return True
@@ -25,5 +29,27 @@ def register_demo(debug=False):
         time.sleep(2)  # Wait for 2 seconds before retrying
 
     if debug:
-        print("Failed to register IP after 10 attempts.")
+        print("Failed to register IP after 10 attempts.")    
     return False
+
+
+
+def create_url(domain_name = "aiplatform.vcntt.tech"):
+    # Define the static domain name    
+    
+    # Read the KERNEL_NAMESPACE environment variable
+    kernel_namespace = os.getenv("KERNEL_NAMESPACE")
+    
+    # Validate the environment variable
+    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.")
+    
+    # Create the final URL
+    url = f"https://{transformed_namespace}.{domain_name}/demo/"
+    return url, transformed_namespace

{aap_utils-0.1.dist-info → aap_utils-0.1.2.dist-info}/METADATA

@@ -1,11 +1,16 @@
 Metadata-Version: 2.1
-Name: aap_utils
-Version: 0.1
+Name: aap-utils
+Version: 0.1.2
 Summary: A Python library for registering an IP address for AAP
+Home-page: UNKNOWN
 Author: Minh Dang
 Author-email: danghoangminh86@gmail.com
+License: UNKNOWN
+Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Requires-Dist: requests
 
+UNKNOWN
+

aap_utils-0.1.2.dist-info/RECORD

@@ -0,0 +1,6 @@
+aap_utils/__init__.py,sha256=OALCpHMgWzr-3Lu7eKwalsw5GNCwWWkETUPHOqfGXsg,36
+aap_utils/register.py,sha256=aOVv1EhCqUfvIpmLgbWCCMcr-Sk-do3CkhZeVtjdEG0,2046
+aap_utils-0.1.2.dist-info/METADATA,sha256=SB1Dci5aG5JfEEbOnUKCD8EhYCKMfPIf9shIZH6VvKg,409
+aap_utils-0.1.2.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
+aap_utils-0.1.2.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
+aap_utils-0.1.2.dist-info/RECORD,,

{aap_utils-0.1.dist-info → aap_utils-0.1.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: bdist_wheel (0.37.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

aap_utils-0.1.dist-info/RECORD

@@ -1,6 +0,0 @@
-aap_utils/__init__.py,sha256=OALCpHMgWzr-3Lu7eKwalsw5GNCwWWkETUPHOqfGXsg,36
-aap_utils/register.py,sha256=Tv9252qK85Arbl-iZbtu8F2wz7oU7E-EgrH1I2fsqz4,1023
-aap_utils-0.1.dist-info/METADATA,sha256=36CrDww90sppXrnXQg56bIO7OQbJhtjpvrmDsl_fmPk,344
-aap_utils-0.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-aap_utils-0.1.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
-aap_utils-0.1.dist-info/RECORD,,