feat: add route /knowledge/delete
This commit is contained in:
@@ -6,6 +6,7 @@ from fastapi import UploadFile, File, Form
|
|||||||
|
|
||||||
from app.utils.db_utils import AsyncSessionDep
|
from app.utils.db_utils import AsyncSessionDep
|
||||||
from app.utils.knowledge_utils import knowledge_service
|
from app.utils.knowledge_utils import knowledge_service
|
||||||
|
from app.utils.milvus_utils import milvus_service
|
||||||
|
|
||||||
|
|
||||||
def add_knowledge_route(app):
|
def add_knowledge_route(app):
|
||||||
@@ -34,11 +35,13 @@ def add_knowledge_route(app):
|
|||||||
# "search_documents": search_result,
|
# "search_documents": search_result,
|
||||||
# }
|
# }
|
||||||
#
|
#
|
||||||
# @app.post("/knowledge/delete")
|
@app.post("/knowledge/delete")
|
||||||
# async def knowledge_embed_text(body: dict):
|
async def knowledge_embed_text(body: dict):
|
||||||
# await milvus_service.async_delete(body.get('id'))
|
await milvus_service.async_delete(body.get('id'))
|
||||||
# return {
|
return {
|
||||||
# "result": "删除成功",
|
"result": "删除成功",
|
||||||
|
}
|
||||||
|
|
||||||
@app.post("/knowledge/upload_files")
|
@app.post("/knowledge/upload_files")
|
||||||
async def knowledge_search(
|
async def knowledge_search(
|
||||||
session: AsyncSessionDep,
|
session: AsyncSessionDep,
|
||||||
@@ -64,7 +67,7 @@ def add_knowledge_route(app):
|
|||||||
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)
|
||||||
|
|
||||||
# 异步处理嵌入文档,不再等待
|
# 异步处理嵌入文档,不再等待
|
||||||
[asyncio.create_task(knowledge_service.process_file_dict(file_dict=file_dict, kb_code=kb_code)) for file_dict in file_dict_list]
|
[asyncio.create_task(knowledge_service.process_doc_cls(doc_cls=doc_cls)) for doc_cls in doc_cls_list]
|
||||||
|
|
||||||
# 直接返回文档对象
|
# 直接返回文档对象
|
||||||
return {"message": f"正在处理 {len(doc_cls_list)} 个文档,请刷新列表查看文档状态。", "result": [item.model_dump() for item in doc_cls_list]}
|
return {"message": f"正在处理 {len(doc_cls_list)} 个文档,请刷新列表查看文档状态。", "result": [item.model_dump() for item in doc_cls_list]}
|
||||||
|
|||||||
Reference in New Issue
Block a user