18dce66443
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: -LAN- <laipz8200@outlook.com>
29 lines
869 B
Python
29 lines
869 B
Python
from flask_restx import Resource
|
|
|
|
from controllers.console import api
|
|
from controllers.console.datasets.hit_testing_base import DatasetsHitTestingBase
|
|
from controllers.console.wraps import (
|
|
account_initialization_required,
|
|
cloud_edition_billing_rate_limit_check,
|
|
setup_required,
|
|
)
|
|
from libs.login import login_required
|
|
|
|
|
|
class HitTestingApi(Resource, DatasetsHitTestingBase):
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@cloud_edition_billing_rate_limit_check("knowledge")
|
|
def post(self, dataset_id):
|
|
dataset_id_str = str(dataset_id)
|
|
|
|
dataset = self.get_and_validate_dataset(dataset_id_str)
|
|
args = self.parse_args()
|
|
self.hit_testing_args_check(args)
|
|
|
|
return self.perform_hit_testing(dataset, args)
|
|
|
|
|
|
api.add_resource(HitTestingApi, "/datasets/<uuid:dataset_id>/hit-testing")
|