feat: 上传文档块的时候,文档块保留用户的名称
This commit is contained in:
@@ -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
|
||||
]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user