Przeglądaj źródła

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

tags/0.9.2
crazywoola 1 rok temu
rodzic
commit
e0c8189f1a
No account linked to committer's email address
1 zmienionych plików z 3 dodań i 0 usunięć
  1. 3
    0
      api/services/dataset_service.py

+ 3
- 0
api/services/dataset_service.py Wyświetl plik



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
): ):
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

Ładowanie…
Anuluj
Zapisz