mirror of
https://github.com/langgenius/dify.git
synced 2026-05-28 12:53:23 +08:00
49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from uuid import UUID
|
|
|
|
from flask_restx import Resource
|
|
|
|
from controllers.common.schema import register_response_schema_models, register_schema_models
|
|
from fields.hit_testing_fields import HitTestingResponse
|
|
from libs.helper import dump_response
|
|
from libs.login import login_required
|
|
|
|
from .. import console_ns
|
|
from ..datasets.hit_testing_base import DatasetsHitTestingBase, HitTestingPayload
|
|
from ..wraps import (
|
|
account_initialization_required,
|
|
cloud_edition_billing_rate_limit_check,
|
|
setup_required,
|
|
)
|
|
|
|
register_schema_models(console_ns, HitTestingPayload)
|
|
register_response_schema_models(console_ns, HitTestingResponse)
|
|
|
|
|
|
@console_ns.route("/datasets/<uuid:dataset_id>/hit-testing")
|
|
class HitTestingApi(Resource, DatasetsHitTestingBase):
|
|
@console_ns.doc("test_dataset_retrieval")
|
|
@console_ns.doc(description="Test dataset knowledge retrieval")
|
|
@console_ns.doc(params={"dataset_id": "Dataset ID"})
|
|
@console_ns.expect(console_ns.models[HitTestingPayload.__name__])
|
|
@console_ns.response(
|
|
200,
|
|
"Hit testing completed successfully",
|
|
model=console_ns.models[HitTestingResponse.__name__],
|
|
)
|
|
@console_ns.response(404, "Dataset not found")
|
|
@console_ns.response(400, "Invalid parameters")
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@cloud_edition_billing_rate_limit_check("knowledge")
|
|
def post(self, dataset_id: UUID) -> dict[str, object]:
|
|
dataset_id_str = str(dataset_id)
|
|
|
|
dataset = self.get_and_validate_dataset(dataset_id_str)
|
|
args = self.parse_args(console_ns.payload)
|
|
self.hit_testing_args_check(args)
|
|
|
|
return dump_response(HitTestingResponse, self.perform_hit_testing(dataset, args))
|