瀏覽代碼

fix(file_factory): validate upload_file_id before querying UploadFile (#11937)

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- 4 月之前
父節點
當前提交
8f73670925
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      api/factories/file_factory.py

+ 4 - 1
api/factories/file_factory.py

@@ -116,8 +116,11 @@ def _build_from_local_file(
     tenant_id: str,
     transfer_method: FileTransferMethod,
 ) -> File:
+    upload_file_id = mapping.get("upload_file_id")
+    if not upload_file_id:
+        raise ValueError("Invalid upload file id")
     stmt = select(UploadFile).where(
-        UploadFile.id == mapping.get("upload_file_id"),
+        UploadFile.id == upload_file_id,
         UploadFile.tenant_id == tenant_id,
     )