From 08c58fe0e612d1b9e5d221c1969471d048221f39 Mon Sep 17 00:00:00 2001 From: stefanfeng Date: Fri, 4 Sep 2026 15:18:39 +0800 Subject: [PATCH] fix(avatar): cap knowledge files at 50MB --- digital-avatar-app/backend/routers/knowledge.py | 4 ++-- digital-avatar-app/backend/tests/test_knowledge_storage.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/digital-avatar-app/backend/routers/knowledge.py b/digital-avatar-app/backend/routers/knowledge.py index 199aa08..85e7b37 100644 --- a/digital-avatar-app/backend/routers/knowledge.py +++ b/digital-avatar-app/backend/routers/knowledge.py @@ -17,7 +17,7 @@ UPLOAD_DIR = os.path.abspath(os.getenv("UPLOAD_DIR", os.path.join(BASE_DIR, "upl os.makedirs(UPLOAD_DIR, exist_ok=True) ALLOWED_EXT = {".md", ".txt", ".pdf", ".doc", ".docx", ".xlsx"} -MAX_UPLOAD_BYTES = 100 * 1024 * 1024 +MAX_UPLOAD_BYTES = 50 * 1024 * 1024 UPLOAD_CHUNK_BYTES = 1024 * 1024 @@ -88,7 +88,7 @@ async def upload_doc(avatar_id: str, file: UploadFile = File(...), authorization while chunk := await file.read(UPLOAD_CHUNK_BYTES): file_size += len(chunk) if file_size > MAX_UPLOAD_BYTES: - raise ValueError("文件不能超过 100MB") + raise ValueError("文件不能超过 50MB") f.write(chunk) except ValueError as exc: if os.path.exists(path): diff --git a/digital-avatar-app/backend/tests/test_knowledge_storage.py b/digital-avatar-app/backend/tests/test_knowledge_storage.py index b13c1a3..f22aeb5 100644 --- a/digital-avatar-app/backend/tests/test_knowledge_storage.py +++ b/digital-avatar-app/backend/tests/test_knowledge_storage.py @@ -82,7 +82,7 @@ def test_upload_rejects_oversize_file_before_queuing_indexing( payload = response.json() assert payload["code"] == 400 - assert payload["message"] == "文件不能超过 100MB" + assert payload["message"] == "文件不能超过 50MB" enqueue.assert_not_called() assert not list((tmp_path / context["avatar"].id).glob("*"))