From 8bf6e0172f59b932b15e9b42cb75fb96a303339f Mon Sep 17 00:00:00 2001 From: GonzaHM Date: Wed, 21 May 2025 15:50:30 +0900 Subject: [PATCH] fix: correct indentation in dataset retrieval model assignment Fix a bug that caused errors when uploading knowledge through the API by correcting the indentation level of the retrieval model assignment in dataset_service.py. The incorrect indentation was causing the retrieval model to be assigned outside the proper scope, leading to API errors during knowledge upload. --- api/services/dataset_service.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/api/services/dataset_service.py b/api/services/dataset_service.py index bac1bf389f..6957315409 100644 --- a/api/services/dataset_service.py +++ b/api/services/dataset_service.py @@ -960,11 +960,11 @@ class DocumentService: "score_threshold_enabled": False, } - dataset.retrieval_model = ( - knowledge_config.retrieval_model.model_dump() - if knowledge_config.retrieval_model - else default_retrieval_model - ) # type: ignore + dataset.retrieval_model = ( + knowledge_config.retrieval_model.model_dump() + if knowledge_config.retrieval_model + else default_retrieval_model + ) # type: ignore documents = [] if knowledge_config.original_document_id: