Add more fields to MultipartUploadDetails for CLI to use

kscalelabs · Nov 20, 2024 · c5fb466 · c5fb466
1 parent 019e383
commit c5fb466
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/frontend/src/gen/api.ts b/frontend/src/gen/api.ts
@@ -1752,6 +1752,10 @@ export interface components {
             bucket: string;
             /** Key */
             key: string;
+            /** Part Size */
+            part_size: number;
+            /** Num Parts */
+            num_parts: number;
         };
         /** MyUserInfoResponse */
         MyUserInfoResponse: {

diff --git a/store/app/crud/base.py b/store/app/crud/base.py
@@ -66,6 +66,8 @@ class MultipartUploadDetails(TypedDict):
     presigned_urls: list[dict[str, str | int]]
     bucket: str
     key: str
+    part_size: int
+    num_parts: int
 
 
 class BaseCrud(AsyncContextManager["BaseCrud"]):
@@ -664,4 +666,11 @@ async def _initiate_multipart_upload(
         bucket, upload_id = await self._create_multipart_upload(key, content_type, metadata)
         presigned_urls = await self._generate_presigned_urls(key, upload_id, num_parts, expires_in)
 
-        return MultipartUploadDetails(upload_id=upload_id, presigned_urls=presigned_urls, bucket=bucket, key=key)
+        return MultipartUploadDetails(
+            upload_id=upload_id,
+            presigned_urls=presigned_urls,
+            bucket=bucket,
+            key=key,
+            part_size=part_size,
+            num_parts=num_parts,
+        )