Do not raise an error in sidecar if uploaded file does not exist

polyaxon · Dec 11, 2024 · f290980 · f290980
1 parent 9d1f68d
commit f290980
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/haupt/haupt/streams/endpoints/artifacts.py b/haupt/haupt/streams/endpoints/artifacts.py
@@ -28,7 +28,12 @@
 async def handle_upload(
     request: ASGIRequest, run_uuid: str, is_file: bool
 ) -> HttpResponse:
-    content_file = request.FILES["upload_file"]
+    content_file = request.FILES.get("upload_file")
+    if not content_file:
+        return HttpResponse(
+            content="A file is required to upload a file",
+            status=status.HTTP_400_BAD_REQUEST,
+        )
     content_json = request.POST.get("json")
     content_json = orjson_loads(content_json) if content_json else {}
     overwrite = content_json.get("overwrite", True)