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

2
api/controllers/service_api/dataset/document.py
View file

@ -28,6 +28,7 @@ class DocumentAddByTextApi(DatasetApiResource):
"""Resource for documents."""
@cloud_edition_billing_resource_check('vector_space', 'dataset')
@cloud_edition_billing_resource_check('documents', 'dataset')
def post(self, tenant_id, dataset_id):
"""Create document by text."""
parser = reqparse.RequestParser()
@ -153,6 +154,7 @@ class DocumentUpdateByTextApi(DatasetApiResource):
class DocumentAddByFileApi(DatasetApiResource):
"""Resource for documents."""
@cloud_edition_billing_resource_check('vector_space', 'dataset')
@cloud_edition_billing_resource_check('documents', 'dataset')
def post(self, tenant_id, dataset_id):
"""Create document by upload file."""
args = {}