fix(batch-exports): Fix missing key in S3 complete multipart upload (#…

…27285)
PostHog · Jan 6, 2025 · e7b412b · e7b412b
1 parent 85ff197
commit e7b412b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/posthog/temporal/batch_exports/s3_batch_export.py b/posthog/temporal/batch_exports/s3_batch_export.py
@@ -327,7 +327,7 @@ def continue_from_state(self, state: S3MultiPartUploadState):
 
         return self.upload_id
 
-    async def complete(self) -> str:
+    async def complete(self) -> str | None:
         if self.is_upload_in_progress() is False:
             raise NoUploadInProgressError()
 
@@ -343,7 +343,7 @@ async def complete(self) -> str:
         self.upload_id = None
         self.parts = []
 
-        return response["Location"]
+        return response.get("Key")
 
     async def abort(self):
         """Abort this S3 multi-part upload."""