diff --git a/api/db/services/document_service.py b/api/db/services/document_service.py index 0abf1b1f3..514b3fd87 100644 --- a/api/db/services/document_service.py +++ b/api/db/services/document_service.py @@ -41,6 +41,7 @@ from rag.utils.redis_conn import REDIS_CONN from rag.utils.doc_store_conn import OrderByExpr from common import settings + class DocumentService(CommonService): model = Document diff --git a/api/db/services/knowledgebase_service.py b/api/db/services/knowledgebase_service.py index fbd193509..6b39a4fe3 100644 --- a/api/db/services/knowledgebase_service.py +++ b/api/db/services/knowledgebase_service.py @@ -28,6 +28,7 @@ from common.constants import StatusEnum from api.constants import DATASET_NAME_LIMIT from api.utils.api_utils import get_parser_config, get_data_error_result + class KnowledgebaseService(CommonService): """Service class for managing knowledge base operations. @@ -396,7 +397,7 @@ class KnowledgebaseService(CommonService): if dataset_name == "": return False, get_data_error_result(message="Dataset name can't be empty.") if len(dataset_name.encode("utf-8")) > DATASET_NAME_LIMIT: - return get_data_error_result(message=f"Dataset name length is {len(dataset_name)} which is larger than {DATASET_NAME_LIMIT}") + return False, get_data_error_result(message=f"Dataset name length is {len(dataset_name)} which is larger than {DATASET_NAME_LIMIT}") # Deduplicate name within tenant dataset_name = duplicate_name(