diff --git a/api/apps/restful_apis/dataset_api.py b/api/apps/restful_apis/dataset_api.py index 03050453f2..cf6ff26cb0 100644 --- a/api/apps/restful_apis/dataset_api.py +++ b/api/apps/restful_apis/dataset_api.py @@ -575,6 +575,7 @@ def delete_knowledge_graph(tenant_id, dataset_id): @add_tenant_id_to_kwargs async def run_index(tenant_id, dataset_id): index_type = request.args.get("type", "") + index_type = index_type.lower() try: success, result = dataset_api_service.run_index(dataset_id, tenant_id, index_type) if success: @@ -593,6 +594,7 @@ async def run_index(tenant_id, dataset_id): @add_tenant_id_to_kwargs def trace_index(tenant_id, dataset_id): index_type = request.args.get("type", "") + index_type = index_type.lower() try: success, result = dataset_api_service.trace_index(dataset_id, tenant_id, index_type) if success: @@ -610,6 +612,7 @@ def trace_index(tenant_id, dataset_id): @login_required @add_tenant_id_to_kwargs def delete_index(tenant_id, dataset_id, index_type): + index_type = index_type.lower() if index_type not in dataset_api_service._VALID_INDEX_TYPES: return get_error_argument_result(f"Invalid index type '{index_type}'") try: