AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-13 20:46:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
crazywoola 2024-10-11 11:28:09 +08:00 committed by GitHub
parent 8bcad002df
commit e0c8189f1a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/services/dataset_service.py
View File

@ -8,6 +8,7 @@ from typing import Optional
from flask_login import current_user from flask_login import current_user
from sqlalchemy import func from sqlalchemy import func
from werkzeug.exceptions import NotFound
from configs import dify_config from configs import dify_config
from core.errors.error import LLMBadRequestError, ProviderTokenNotInitError from core.errors.error import LLMBadRequestError, ProviderTokenNotInitError
@ -975,6 +976,8 @@ class DocumentService:
): ):
DatasetService.check_dataset_model_setting(dataset) DatasetService.check_dataset_model_setting(dataset)
document = DocumentService.get_document(dataset.id, document_data["original_document_id"]) 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": if document.display_status != "available":
raise ValueError("Document is not available") raise ValueError("Document is not available")
# update document name # update document name