hotfix: fix docker link parsing

ai4os · Oct 21, 2024 · 7772654 · 7772654
1 parent e9ac53d
commit 7772654
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/ai4papi/routers/v1/catalog/common.py b/ai4papi/routers/v1/catalog/common.py
@@ -242,6 +242,7 @@ def get_metadata(
             # TODO: when the migration is finished, we have to generate the url from the module name
             # (ie. ignore the value coming from the metadata)
             metadata['links']['docker_image'] = f"https://hub.docker.com/r/{metadata['links']['docker_image']}"
+            metadata['links']['docker_image'] = metadata['links']['docker_image'].strip('/ ')
 
         return metadata
 

diff --git a/ai4papi/routers/v1/catalog/modules.py b/ai4papi/routers/v1/catalog/modules.py
@@ -51,7 +51,7 @@ def get_config(
 
     # Parse docker registry
     registry = metadata['links']['docker_image']
-    repo, image = registry.split('/')[:2]
+    repo, image = registry.split('/')[-2:]
     if repo not in ['deephdc', 'ai4oshub']:
         repo = 'ai4oshub'
 

diff --git a/ai4papi/routers/v1/catalog/tools.py b/ai4papi/routers/v1/catalog/tools.py
@@ -47,7 +47,7 @@ def get_config(
 
     # Parse docker registry
     registry = metadata['links']['docker_image']
-    repo, image = registry.split('/')[:2]
+    repo, image = registry.split('/')[-2:]
     if repo not in ['deephdc', 'ai4oshub']:
         repo = 'ai4oshub'
 

diff --git a/ai4papi/routers/v1/try_me/nomad.py b/ai4papi/routers/v1/try_me/nomad.py
@@ -38,7 +38,8 @@ def create_deployment(
 
     # Retrieve docker_image from module_name
     meta = Modules.get_metadata(module_name)
-    docker_image = meta['links']['docker_image']
+    registry = meta['links']['docker_image']
+    docker_image = '/'.join(registry.split('/')[-2:])
 
     # Load module configuration
     nomad_conf = deepcopy(papiconf.TRY_ME['nomad'])