jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Feat: Add documents limitation (#2662)

Browse source
This commit is contained in:
Garfield Dai 2024-03-03 12:45:06 +08:00 • committed by GitHub
commit 8e66b96221
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
14 changed files with 86 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

3
api/controllers/console/datasets/file.py
View file

@ -11,7 +11,7 @@ from controllers.console.datasets.error import (
UnsupportedFileTypeError,
)
from controllers.console.setup import setup_required
from controllers.console.wraps import account_initialization_required
from controllers.console.wraps import account_initialization_required, cloud_edition_billing_resource_check
from fields.file_fields import file_fields, upload_config_fields
from libs.login import login_required
from services.file_service import ALLOWED_EXTENSIONS, UNSTRUSTURED_ALLOWED_EXTENSIONS, FileService
@ -39,6 +39,7 @@ class FileApi(Resource):
@login_required
@account_initialization_required
@marshal_with(file_fields)
@cloud_edition_billing_resource_check(resource='documents')
def post(self):
# get file from request