Browse Source

minor fix: fix the check of subscription capacity limit (#24991)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
tags/1.8.1
NeatGuyCoding 2 months ago
parent
commit
ed5ed0306e
No account linked to committer's email address
1 changed files with 4 additions and 2 deletions
  1. 4
    2
      api/tasks/duplicate_document_indexing_task.py

+ 4
- 2
api/tasks/duplicate_document_indexing_task.py View File

@@ -45,9 +45,11 @@ def duplicate_document_indexing_task(dataset_id: str, document_ids: list):
batch_upload_limit = int(dify_config.BATCH_UPLOAD_LIMIT)
if count > batch_upload_limit:
raise ValueError(f"You have reached the batch upload limit of {batch_upload_limit}.")
if 0 < vector_space.limit <= vector_space.size:
current = int(getattr(vector_space, "size", 0) or 0)
limit = int(getattr(vector_space, "limit", 0) or 0)
if limit > 0 and (current + count) > limit:
raise ValueError(
"Your total number of documents plus the number of uploads have over the limit of "
"Your total number of documents plus the number of uploads have exceeded the limit of "
"your subscription."
)
except Exception as e:

Loading…
Cancel
Save