Skip to content

Commit 754791e

Browse files
authored
fix(file_factory): validate upload_file_id format as UUID (#12084)
Signed-off-by: -LAN- <[email protected]>
1 parent 50b7ec3 commit 754791e

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

api/factories/file_factory.py

+6
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import mimetypes
2+
import uuid
23
from collections.abc import Callable, Mapping, Sequence
34
from typing import Any, cast
45

@@ -119,6 +120,11 @@ def _build_from_local_file(
119120
upload_file_id = mapping.get("upload_file_id")
120121
if not upload_file_id:
121122
raise ValueError("Invalid upload file id")
123+
# check if upload_file_id is a valid uuid
124+
try:
125+
uuid.UUID(upload_file_id)
126+
except ValueError:
127+
raise ValueError("Invalid upload file id format")
122128
stmt = select(UploadFile).where(
123129
UploadFile.id == upload_file_id,
124130
UploadFile.tenant_id == tenant_id,

0 commit comments

Comments
 (0)