| @@ -1,6 +1,7 @@ | |||
| from flask import request | |||
| from flask_login import current_user | |||
| from flask_restful import Resource, marshal_with | |||
| from werkzeug.exceptions import Forbidden | |||
| import services | |||
| from configs import dify_config | |||
| @@ -58,6 +59,9 @@ class FileApi(Resource): | |||
| if not file.filename: | |||
| raise FilenameNotExistsError | |||
| if source == "datasets" and not current_user.is_dataset_editor: | |||
| raise Forbidden() | |||
| if source not in ("datasets", None): | |||
| source = None | |||