feat(kb api): update python style

pull/20426/head
Dongyu Li 12 months ago
parent 6cd231cfd8
commit 4860daab2b

@ -411,7 +411,6 @@ class DatasetChildChunkApi(DatasetApiResource):
return {"data": marshal(child_chunk, child_chunk_fields)}, 200 return {"data": marshal(child_chunk, child_chunk_fields)}, 200
api.add_resource(SegmentApi, "/datasets/<uuid:dataset_id>/documents/<uuid:document_id>/segments") api.add_resource(SegmentApi, "/datasets/<uuid:dataset_id>/documents/<uuid:document_id>/segments")
api.add_resource( api.add_resource(
DatasetSegmentApi, "/datasets/<uuid:dataset_id>/documents/<uuid:document_id>/segments/<uuid:segment_id>" DatasetSegmentApi, "/datasets/<uuid:dataset_id>/documents/<uuid:document_id>/segments/<uuid:segment_id>"

@ -41,6 +41,7 @@ class HitTestingService:
dataset_retrieval = DatasetRetrieval() dataset_retrieval = DatasetRetrieval()
from core.app.app_config.entities import MetadataFilteringCondition from core.app.app_config.entities import MetadataFilteringCondition
metadata_filtering_conditions = MetadataFilteringCondition(**metadata_filtering_conditions) metadata_filtering_conditions = MetadataFilteringCondition(**metadata_filtering_conditions)
metadata_filter_document_ids, metadata_condition = dataset_retrieval.get_metadata_filter_condition( metadata_filter_document_ids, metadata_condition = dataset_retrieval.get_metadata_filter_condition(
@ -51,7 +52,7 @@ class HitTestingService:
inputs={}, inputs={},
tenant_id=None, tenant_id=None,
user_id=None, user_id=None,
metadata_model_config=None metadata_model_config=None,
) )
if metadata_filter_document_ids: if metadata_filter_document_ids:
document_ids_filter = metadata_filter_document_ids.get(dataset.id, []) document_ids_filter = metadata_filter_document_ids.get(dataset.id, [])
@ -70,7 +71,7 @@ class HitTestingService:
else None, else None,
reranking_mode=retrieval_model.get("reranking_mode") or "reranking_model", reranking_mode=retrieval_model.get("reranking_mode") or "reranking_model",
weights=retrieval_model.get("weights", None), weights=retrieval_model.get("weights", None),
document_ids_filter=document_ids_filter document_ids_filter=document_ids_filter,
) )
end = time.perf_counter() end = time.perf_counter()

Loading…
Cancel
Save