Adding new FileResponseParams.

Author: Khurdhula-Harshavardhan

jigsawstack/store.py

@@ -14,6 +14,13 @@ class FileUploadParams(TypedDict):
     content_type: NotRequired[str]
     temp_public_url: NotRequired[bool]
 
+class FileUploadResponse(TypedDict):
+    key: str
+    url: str
+    size: int
+    temp_public_url: NotRequired[str] # Optional, only if temp_public_url is set to True in params
+    
+
 class Store(ClientConfig):
 
     config: RequestConfig
@@ -31,7 +38,7 @@ def __init__(
             disable_request_logging=disable_request_logging,
         )
 
-    def upload(self, file: bytes, options: Union[FileUploadParams, None] = None) -> Any:
+    def upload(self, file: bytes, options: Union[FileUploadParams, None] = None) -> FileUploadResponse:
         if options is None:
             options = {}
 
@@ -88,7 +95,7 @@ def __init__(
         )
 
 
-    async def upload(self, file: bytes, options: Union[FileUploadParams, None] = None) -> Any:
+    async def upload(self, file: bytes, options: Union[FileUploadParams, None] = None) -> FileUploadResponse:
         if options is None:
             options = {}