mirror of
https://github.com/langgenius/dify.git
synced 2025-11-04 12:53:38 +00:00
18 lines
606 B
Python
18 lines
606 B
Python
|
from controllers.console.datasets.hit_testing_base import DatasetsHitTestingBase
|
||
|
|
from controllers.service_api import api
|
||
|
|
from controllers.service_api.wraps import DatasetApiResource
|
||
|
|
|
||
|
|
|
||
|
|
class HitTestingApi(DatasetApiResource, DatasetsHitTestingBase):
|
||
|
|
def post(self, tenant_id, 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")
|