fix: allow VFolder id as name parameter in API

lablup · Jan 15, 2025 · 42f46a4 · 42f46a4
1 parent 76f646c
commit 42f46a4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/ai/backend/manager/api/vfolder.py b/src/ai/backend/manager/api/vfolder.py
@@ -248,7 +248,11 @@ async def resolve_vfolder_rows(
 
     match folder_id_or_name:
         case str():
-            extra_vf_conds = vfolders.c.name == folder_id_or_name
+            try:
+                folder_id = uuid.UUID(folder_id_or_name)
+                extra_vf_conds = vfolders.c.id == folder_id
+            except ValueError:
+                extra_vf_conds = vfolders.c.name == folder_id_or_name
         case uuid.UUID():
             extra_vf_conds = vfolders.c.id == folder_id_or_name
         case _: