hailin
/
faiss_rag_enterprise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-05-09 23:33:34 +08:00
parent acc76d573a
commit 92282c2060
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
app/api/upload.py
View File

@ -10,17 +10,19 @@ ALLOWED_SUFFIXES = {".txt", ".md", ".pdf", ".docx"}
def upload_user_file(user_id: str = Form(...), file: UploadFile = File(...)): def upload_user_file(user_id: str = Form(...), file: UploadFile = File(...)):
filename = os.path.basename(file.filename) filename = os.path.basename(file.filename)
suffix = os.path.splitext(filename)[-1].lower() suffix = os.path.splitext(filename)[-1].lower()
if suffix not in ALLOWED_SUFFIXES: if suffix not in ALLOWED_SUFFIXES:
raise HTTPException(status_code=400, detail="不支持的文件类型") raise HTTPException(status_code=400, detail="不支持的文件类型")
# 创建用户文档目录 # 创建用户文档目录
user_doc_dir = os.path.join("docs", user_id) user_doc_dir = os.path.join("docs", user_id)
os.makedirs(user_doc_dir, exist_ok=True) if not os.path.exists(user_doc_dir): # 检查目录是否存在
os.makedirs(user_doc_dir, exist_ok=True) # 如果不存在，创建目录
# 创建索引存储路径，确保目录存在 # 创建索引数据目录，确保路径存在
index_data_dir = os.path.join("index_data", user_id) index_data_dir = os.path.join("index_data", user_id)
os.makedirs(index_data_dir, exist_ok=True) if not os.path.exists(index_data_dir): # 检查目录是否存在
os.makedirs(index_data_dir, exist_ok=True) # 如果不存在，创建目录
# 保存文件 # 保存文件
file_path = os.path.join(user_doc_dir, filename) file_path = os.path.join(user_doc_dir, filename)
@ -29,7 +31,13 @@ def upload_user_file(user_id: str = Form(...), file: UploadFile = File(...)):
shutil.copyfileobj(file.file, f) shutil.copyfileobj(file.file, f)
print(f"[UPLOAD] 文件已保存至 {file_path}") print(f"[UPLOAD] 文件已保存至 {file_path}")
# 创建索引 # 创建索引并保存 docstore.json
docstore_path = os.path.join(index_data_dir, "docstore.json")
if not os.path.exists(docstore_path): # 检查文件是否存在
with open(docstore_path, "w") as docstore_file:
docstore_file.write("{}") # 可根据需要保存实际内容
# 重新构建用户索引
build_user_index(user_id) build_user_index(user_id)
print(f"[UPLOAD] 用户 {user_id} 的索引已重建") print(f"[UPLOAD] 用户 {user_id} 的索引已重建")