瀏覽代碼

feat: add NotFound error for dataset service (#9215)

tags/0.9.2
crazywoola 1 年之前
父節點
當前提交
e0c8189f1a
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 3 行新增0 行删除
  1. 3
    0
      api/services/dataset_service.py

+ 3
- 0
api/services/dataset_service.py 查看文件

@@ -8,6 +8,7 @@ from typing import Optional

from flask_login import current_user
from sqlalchemy import func
from werkzeug.exceptions import NotFound

from configs import dify_config
from core.errors.error import LLMBadRequestError, ProviderTokenNotInitError
@@ -975,6 +976,8 @@ class DocumentService:
):
DatasetService.check_dataset_model_setting(dataset)
document = DocumentService.get_document(dataset.id, document_data["original_document_id"])
if document is None:
raise NotFound("Document not found")
if document.display_status != "available":
raise ValueError("Document is not available")
# update document name

Loading…
取消
儲存