Parcourir la 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 il y a 2 mois
Parent
révision
ed5ed0306e
Aucun compte lié à l'adresse e-mail de l'auteur
1 fichiers modifiés avec 4 ajouts et 2 suppressions
  1. 4
    2
      api/tasks/duplicate_document_indexing_task.py

+ 4
- 2
api/tasks/duplicate_document_indexing_task.py Voir le fichier

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

Chargement…
Annuler
Enregistrer