fix: Missing conversion from str to UUID when get/setting vfolder quo…

…ta (#487)
lablup · Jun 24, 2022 · 126126c · 126126c
1 parent ae1c3c6
commit 126126c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/changes/487.fix.md b/changes/487.fix.md
@@ -0,0 +1 @@
+Fix missing str to UUID conversion for `vfid` parameters in `get_quota` and `set_quota` manager-facing APIs in the storage proxy
diff --git a/src/ai/backend/storage/api/manager.py b/src/ai/backend/storage/api/manager.py
@@ -382,7 +382,7 @@ async def get_quota(request: web.Request) -> web.Response:
         t.Dict(
             {
                 t.Key("volume"): t.String(),
-                t.Key("vfid", default=None): t.Null | t.String,
+                t.Key("vfid", default=None): t.Null | tx.UUID,
             },
         ),
     ) as params:
@@ -399,7 +399,7 @@ async def set_quota(request: web.Request) -> web.Response:
         t.Dict(
             {
                 t.Key("volume"): t.String(),
-                t.Key("vfid", default=None): t.Null | t.String,
+                t.Key("vfid", default=None): t.Null | tx.UUID,
                 t.Key("size_bytes"): tx.BinarySize,
             },
         ),