feat: 上传文档块的时候,文档块保留用户的名称

This commit is contained in:
martsforever
2025-09-07 20:43:17 +08:00
parent 21b7757c14
commit 949b4eda8b
2 changed files with 24 additions and 3 deletions
+16 -2
View File
@@ -8,6 +8,7 @@ from llama_index.core import SimpleDirectoryReader, Document
from app.config.env import env
from app.model.FileModel import FileSaveService
from app.model.KnowledgeDoc import KnowledgeDocModel, KnowledgeDocService
from app.model.UserModel import PublicUser
from app.utils.db_utils import AsyncSessionDep, async_session
from app.utils.milvus_utils import milvus_service
@@ -24,11 +25,24 @@ class KnowledgeService:
return await FileSaveService.saveFile(session=session, file=file, filename=file.filename, file_record={})
# 将 file_dict_list 保存为 doc_cls_list
async def save_knowledge_doc_list(self, session: AsyncSessionDep, file_dict_list: List[dict], kb_code: str):
async def save_knowledge_doc_list(
self,
session: AsyncSessionDep,
file_dict_list: List[dict],
kb_code: str,
user: PublicUser,
):
# 先插入文档对象
tobe_insert_kd_list = [
KnowledgeDocModel(id=file_dict["id"], name=file_dict["name"], path=file_dict["path"], parent_code=kb_code, status="process").model_dump()
KnowledgeDocModel(
id=file_dict["id"],
created_by=user.id,
name=file_dict["name"],
path=file_dict["path"],
parent_code=kb_code,
status="process"
).model_dump()
for file_dict in file_dict_list
]