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

aap_utils/init_kcn.py

@@ -0,0 +1,19 @@
+import os
+from hydra import initialize, compose
+from omegaconf import OmegaConf, DictConfig
+kcn_params = "config.yaml"
+config_path="configs/"
+overrides = []
+
+if os.environ.get('KCN_PARAMS') and os.environ.get('KCN_PARAMS') != '':
+    full_path = os.getenv('KCN_PARAMS')
+    full_path = "../" + full_path
+    config_path = os.path.dirname(full_path)
+    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)

aap_utils/register.py

@@ -2,6 +2,8 @@ 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):
@@ -13,7 +15,11 @@ def register_demo(debug=False):
                 headers={"Content-Type": "application/json"}
             )
             if response.ok:
-                url, namespace = create_url()
+                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:
@@ -51,5 +57,24 @@ def create_url(domain_name = "aiplatform.vcntt.tech"):
         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/"
+    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.2.dist-info → aap_utils-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aap-utils
-Version: 0.1.2
+Version: 0.1.4
 Summary: A Python library for registering an IP address for AAP
 Home-page: UNKNOWN
 Author: Minh Dang

aap_utils-0.1.4.dist-info/RECORD

@@ -0,0 +1,7 @@
+aap_utils/__init__.py,sha256=OALCpHMgWzr-3Lu7eKwalsw5GNCwWWkETUPHOqfGXsg,36
+aap_utils/init_kcn.py,sha256=FMX9w0WoaP6I3yc3xn8xqGYT2VbtG2uJbwpTKouZpew,718
+aap_utils/register.py,sha256=YDUvUtoICYVMuuzYAueb34ZtGcYI8SJ6YxoNB3POyyQ,3069
+aap_utils-0.1.4.dist-info/METADATA,sha256=zJjIBA1qS5GDppSjX3rtQF8zY455umerTc3pI-jiaJI,409
+aap_utils-0.1.4.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
+aap_utils-0.1.4.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
+aap_utils-0.1.4.dist-info/RECORD,,

aap_utils-0.1.2.dist-info/RECORD

@@ -1,6 +0,0 @@
-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,,