feat: 上传文档块的时候,文档块保留用户的名称
This commit is contained in:
@@ -3,6 +3,7 @@ from http.client import HTTPException
|
||||
from typing import List
|
||||
|
||||
from fastapi import UploadFile, File, Form
|
||||
from starlette.requests import Request
|
||||
|
||||
from app.utils.db_utils import AsyncSessionDep
|
||||
from app.utils.knowledge_utils import knowledge_service
|
||||
@@ -45,6 +46,7 @@ def add_knowledge_route(app):
|
||||
@app.post("/knowledge/upload_files")
|
||||
async def knowledge_search(
|
||||
session: AsyncSessionDep,
|
||||
request: Request,
|
||||
files: List[UploadFile] = File(...),
|
||||
kb_code: str = Form(..., description="所属知识库的编码"),
|
||||
):
|
||||
@@ -64,7 +66,12 @@ def add_knowledge_route(app):
|
||||
file_dict_list = [item["result"] for item in task_result_list]
|
||||
|
||||
# 插入对应的文档对象记录
|
||||
doc_cls_list = await knowledge_service.save_knowledge_doc_list(session=session, file_dict_list=file_dict_list, kb_code=kb_code)
|
||||
doc_cls_list = await knowledge_service.save_knowledge_doc_list(
|
||||
session=session,
|
||||
file_dict_list=file_dict_list,
|
||||
kb_code=kb_code,
|
||||
user=request.state.user,
|
||||
)
|
||||
|
||||
# 异步处理嵌入文档,不再等待
|
||||
[asyncio.create_task(knowledge_service.process_doc_cls(doc_cls=doc_cls)) for doc_cls in doc_cls_list]
|
||||
|
||||
Reference in New Issue
Block a user