From 4860daab2bde28c78001b418ede336a07aa9eb5f Mon Sep 17 00:00:00 2001 From: Dongyu Li <544104925@qq.com> Date: Thu, 29 May 2025 15:35:08 +0800 Subject: [PATCH] feat(kb api): update python style --- api/controllers/service_api/dataset/segment.py | 1 - api/services/hit_testing_service.py | 5 +++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/api/controllers/service_api/dataset/segment.py b/api/controllers/service_api/dataset/segment.py index 9d8f69ac42..337752275a 100644 --- a/api/controllers/service_api/dataset/segment.py +++ b/api/controllers/service_api/dataset/segment.py @@ -411,7 +411,6 @@ class DatasetChildChunkApi(DatasetApiResource): return {"data": marshal(child_chunk, child_chunk_fields)}, 200 - api.add_resource(SegmentApi, "/datasets//documents//segments") api.add_resource( DatasetSegmentApi, "/datasets//documents//segments/" diff --git a/api/services/hit_testing_service.py b/api/services/hit_testing_service.py index f76e1202e7..13eb3bda57 100644 --- a/api/services/hit_testing_service.py +++ b/api/services/hit_testing_service.py @@ -41,6 +41,7 @@ class HitTestingService: dataset_retrieval = DatasetRetrieval() from core.app.app_config.entities import MetadataFilteringCondition + metadata_filtering_conditions = MetadataFilteringCondition(**metadata_filtering_conditions) metadata_filter_document_ids, metadata_condition = dataset_retrieval.get_metadata_filter_condition( @@ -51,7 +52,7 @@ class HitTestingService: inputs={}, tenant_id=None, user_id=None, - metadata_model_config=None + metadata_model_config=None, ) if metadata_filter_document_ids: document_ids_filter = metadata_filter_document_ids.get(dataset.id, []) @@ -70,7 +71,7 @@ class HitTestingService: else None, reranking_mode=retrieval_model.get("reranking_mode") or "reranking_model", weights=retrieval_model.get("weights", None), - document_ids_filter=document_ids_filter + document_ids_filter=document_ids_filter, ) end = time.perf_counter()