aap-utils 0.1__tar.gz → 0.1.3__tar.gz

{aap_utils-0.1 → aap_utils-0.1.3}/PKG-INFO

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

{aap_utils-0.1 → aap_utils-0.1.3}/README.md

@@ -4,13 +4,13 @@ A Python library for registering an IP address on a specific server.
 
 ## Installation
 
-To install AAP lib from PyPI, run: pip install pmcd_aap
+To install AAP lib from PyPI, run: pip install aap_utils
 
 
 ## Usage
 
 ```python
-from aap import register_demo
+from aap_utils import register_demo
 
 register_demo(debug=True)
 

aap_utils-0.1.3/aap_utils/register.py

@@ -0,0 +1,80 @@
+import socket
+import requests
+import time
+import os
+from hydra import initialize, compose
+from omegaconf import OmegaConf, DictConfig
+
+def register_demo(debug=False):
+    for attempt in range(10):
+        try:
+            local_ip = socket.gethostbyname(socket.gethostname())
+            response = requests.post(
+                "http://hocmay-svc/aiplatform-api/demo/register/", 
+                json={"demo_ip": local_ip}, 
+                headers={"Content-Type": "application/json"}
+            )
+            if response.ok:
+                response_json = response.json()
+                argo_host = response_json.get('ARGOWORKFLOW_HOST', 'http://argowf.aiplatform.vcntt.tech')
+                domain_name = argo_host.split("argowf.")[-1]
+                url, namespace = create_url(domain_name=domain_name)
+                
+                # 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
+            else:
+                if debug:
+                    print(f"Attempt {attempt + 1}: Failed: {response.text}")
+        except (socket.error, requests.exceptions.RequestException) as e:
+            if debug:  # Print errors only if debug is True
+                print(f"Attempt {attempt + 1}: Error: {e}")
+        
+        time.sleep(2)  # Wait for 2 seconds before retrying
+
+    if debug:
+        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}"
+    return url, transformed_namespace
+
+
+def get_args():
+    kcn_params = "config.yaml"
+    config_path="configs/config.yaml"
+    overrides = []
+
+    if os.environ.get('KCN_PARAMS') and os.environ.get('KCN_PARAMS') != '':
+        full_path = os.getenv('KCN_PARAMS')
+        config_path = os.path.join(config_path, kcn_params)
+        kcn_params = os.path.basename(full_path)  # Extract the filename
+
+    if os.environ.get('KCN_OVERRIDES') and os.environ.get('KCN_OVERRIDES') != '':
+        overrides_env = os.getenv("KCN_OVERRIDES")
+        overrides = overrides_env.split('|') if overrides_env else []
+
+    with initialize(version_base=None, config_path=config_path):
+        args = compose(config_name=kcn_params)
+    return args

{aap_utils-0.1 → aap_utils-0.1.3}/aap_utils.egg-info/PKG-INFO

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

{aap_utils-0.1 → aap_utils-0.1.3}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aap_utils",  # The name of your library
-    version="0.1",
+    version="0.1.3",
     description="A Python library for registering an IP address for AAP",
     author="Minh Dang",
     author_email="danghoangminh86@gmail.com",

aap_utils-0.1/aap_utils/register.py

@@ -1,29 +0,0 @@
-import socket
-import requests
-import time
-
-def register_demo(debug=False):
-    for attempt in range(10):
-        try:
-            local_ip = socket.gethostbyname(socket.gethostname())
-            response = requests.post(
-                "http://hocmay-svc/aiplatform-api/demo/register/", 
-                json={"demo_ip": local_ip}, 
-                headers={"Content-Type": "application/json"}
-            )
-            if response.ok:
-                if debug:
-                    print("IP registered successfully.")
-                return True
-            else:
-                if debug:
-                    print(f"Attempt {attempt + 1}: Failed: {response.text}")
-        except (socket.error, requests.exceptions.RequestException) as e:
-            if debug:  # Print errors only if debug is True
-                print(f"Attempt {attempt + 1}: Error: {e}")
-        
-        time.sleep(2)  # Wait for 2 seconds before retrying
-
-    if debug:
-        print("Failed to register IP after 10 attempts.")
-    return False