瀏覽代碼

Feat: When Delete Chunk Will Also Delete Chunk Related Image (#7656)

### What problem does this PR solve?

When Delete Chunk Will Also Delete Chunk Related Image

### Type of change
- [x] New Feature (non-breaking change which adds functionality)
tags/v0.19.0
Stephen Hu 5 月之前
父節點
當前提交
d2b346cf9e
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 5 行新增0 行删除
  1. 5
    0
      api/apps/chunk_app.py

+ 5
- 0
api/apps/chunk_app.py 查看文件

@@ -37,6 +37,7 @@ import xxhash
import re



@manager.route('/list', methods=['POST']) # noqa: F821
@login_required
@validate_request("doc_id")
@@ -194,6 +195,7 @@ def switch():
@login_required
@validate_request("chunk_ids", "doc_id")
def rm():
from rag.utils.storage_factory import STORAGE_IMPL
req = request.json
try:
e, doc = DocumentService.get_by_id(req["doc_id"])
@@ -204,6 +206,9 @@ def rm():
deleted_chunk_ids = req["chunk_ids"]
chunk_number = len(deleted_chunk_ids)
DocumentService.decrement_chunk_num(doc.id, doc.kb_id, 1, chunk_number, 0)
for cid in deleted_chunk_ids:
if STORAGE_IMPL.obj_exist(doc.kb_id, cid):
STORAGE_IMPL.rm(doc.kb_id, cid)
return get_json_result(data=True)
except Exception as e:
return server_error_response(e)

Loading…
取消
儲存