fix: Let ByteSizeParamType to accept the default value given as int (#169)

Author: achimnol

changes/169.fix

@@ -0,0 +1 @@
+Fix regression of `vfolder upload` and other commands using `ByteSizeParamType` with the default values given as `int`

src/ai/backend/client/cli/utils.py

@@ -20,6 +20,8 @@ class ByteSizeParamType(click.ParamType):
     }
 
     def convert(self, value, param, ctx):
+        if isinstance(value, int):
+            return value
         if not isinstance(value, str):
             self.fail(f"expected string, got {value!r} of type {type(value).__name__}", param, ctx)
         m = self._rx_digits.search(value)
@@ -34,6 +36,8 @@ class ByteSizeParamCheckType(ByteSizeParamType):
     name = "byte-check"
 
     def convert(self, value, param, ctx):
+        if isinstance(value, int):
+            return value
         if not isinstance(value, str):
             self.fail(f"expected string, got {value!r} of type {type(value).__name__}", param, ctx)
         m = self._rx_digits.search(value)

src/ai/backend/client/func/vfolder.py

@@ -248,7 +248,7 @@ async def upload(
                 input_file = open(base_path / file_path, "rb")
             else:
                 input_file = open(str(Path(file_path).relative_to(base_path)), "rb")
-            print(f"Uploading {base_path / file_path} ...")
+            print(f"Uploading {base_path / file_path} via {upload_info['url']} ...")
             # TODO: refactor out the progress bar
             uploader = tus_client.async_uploader(
                 file_stream=input_file,