- correct bug for adding document (due to newly introduced hard usage limits)
- correct comment bug in scripts.html - start of selection_specialist
This commit is contained in:
@@ -21,9 +21,24 @@ from common.services.entitlements import LicenseUsageServices
|
||||
MB_CONVERTOR = 1_048_576
|
||||
|
||||
|
||||
def get_file_size(file):
|
||||
try:
|
||||
# Als file een bytes object is of iets anders dat len() ondersteunt
|
||||
file_size = len(file)
|
||||
except TypeError:
|
||||
# Als file een FileStorage object is
|
||||
current_position = file.tell()
|
||||
file.seek(0, os.SEEK_END)
|
||||
file_size = file.tell()
|
||||
file.seek(current_position)
|
||||
|
||||
return file_size
|
||||
|
||||
|
||||
def create_document_stack(api_input, file, filename, extension, tenant_id):
|
||||
# Precheck if we can add a document to the stack
|
||||
LicenseUsageServices.check_storage_and_embedding_quota(tenant_id, len(file)/MB_CONVERTOR)
|
||||
|
||||
LicenseUsageServices.check_storage_and_embedding_quota(tenant_id, get_file_size(file)/MB_CONVERTOR)
|
||||
|
||||
# Create the Document
|
||||
catalog_id = int(api_input.get('catalog_id'))
|
||||
@@ -338,7 +353,7 @@ def refresh_document_with_content(doc_id: int, tenant_id: int, file_content: byt
|
||||
old_doc_vers = DocumentVersion.query.filter_by(doc_id=doc_id).order_by(desc(DocumentVersion.id)).first()
|
||||
|
||||
# Precheck if we have enough quota for the new version
|
||||
LicenseUsageServices.check_storage_and_embedding_quota(tenant_id, len(file_content) / MB_CONVERTOR)
|
||||
LicenseUsageServices.check_storage_and_embedding_quota(tenant_id, get_file_size(file_content) / MB_CONVERTOR)
|
||||
|
||||
# Create new version with same file type as original
|
||||
extension = old_doc_vers.file_type
|
||||
|
||||
Reference in New Issue
Block a user