Browse Source

Merge pull request #1408 from Mmx233/main

fix: manually check the docs' filename
Timothy Jaeryang Baek 1 year ago
parent
commit
d94bc21ac8
1 changed files with 2 additions and 2 deletions
  1. 2 2
      backend/apps/rag/main.py

+ 2 - 2
backend/apps/rag/main.py

@@ -8,7 +8,7 @@ from fastapi import (
     Form,
     Form,
 )
 )
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.middleware.cors import CORSMiddleware
-import os, shutil, logging
+import os, shutil, logging, re
 
 
 from pathlib import Path
 from pathlib import Path
 from typing import List
 from typing import List
@@ -450,7 +450,7 @@ def store_doc(
     try:
     try:
         is_valid_filename = True
         is_valid_filename = True
         unsanitized_filename = file.filename
         unsanitized_filename = file.filename
-        if not unsanitized_filename.isascii():
+        if re.search(r'[\\/:"\*\?<>|\n\t ]', unsanitized_filename) is not None:
             is_valid_filename = False
             is_valid_filename = False
 
 
         unvalidated_file_path = f"{UPLOAD_DIR}/{unsanitized_filename}"
         unvalidated_file_path = f"{UPLOAD_DIR}/{unsanitized_filename}"