mirror of
https://github.com/langgenius/dify.git
synced 2026-02-09 12:31:14 +08:00
7828508b30
Some checks are pending
autofix.ci / autofix (push) Waiting to run
Build and Push API & Web / build (api, DIFY_API_IMAGE_NAME, linux/amd64, build-api-amd64) (push) Waiting to run
Build and Push API & Web / build (api, DIFY_API_IMAGE_NAME, linux/arm64, build-api-arm64) (push) Waiting to run
Build and Push API & Web / build (web, DIFY_WEB_IMAGE_NAME, linux/amd64, build-web-amd64) (push) Waiting to run
Build and Push API & Web / build (web, DIFY_WEB_IMAGE_NAME, linux/arm64, build-web-arm64) (push) Waiting to run
Build and Push API & Web / create-manifest (api, DIFY_API_IMAGE_NAME, merge-api-images) (push) Blocked by required conditions
Build and Push API & Web / create-manifest (web, DIFY_WEB_IMAGE_NAME, merge-web-images) (push) Blocked by required conditions
Main CI Pipeline / Check Changed Files (push) Waiting to run
Main CI Pipeline / API Tests (push) Blocked by required conditions
Main CI Pipeline / Web Tests (push) Blocked by required conditions
Main CI Pipeline / Style Check (push) Waiting to run
Main CI Pipeline / VDB Tests (push) Blocked by required conditions
Main CI Pipeline / DB Migration Test (push) Blocked by required conditions
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Co-authored-by: Stephen Zhou <38493346+hyoban@users.noreply.github.com>
35 lines
1.4 KiB
Python
35 lines
1.4 KiB
Python
from controllers.common.schema import register_schema_model
|
|
from controllers.console.datasets.hit_testing_base import DatasetsHitTestingBase, HitTestingPayload
|
|
from controllers.service_api import service_api_ns
|
|
from controllers.service_api.wraps import DatasetApiResource, cloud_edition_billing_rate_limit_check
|
|
|
|
register_schema_model(service_api_ns, HitTestingPayload)
|
|
|
|
|
|
@service_api_ns.route("/datasets/<uuid:dataset_id>/hit-testing", "/datasets/<uuid:dataset_id>/retrieve")
|
|
class HitTestingApi(DatasetApiResource, DatasetsHitTestingBase):
|
|
@service_api_ns.doc("dataset_hit_testing")
|
|
@service_api_ns.doc(description="Perform hit testing on a dataset")
|
|
@service_api_ns.doc(params={"dataset_id": "Dataset ID"})
|
|
@service_api_ns.doc(
|
|
responses={
|
|
200: "Hit testing results",
|
|
401: "Unauthorized - invalid API token",
|
|
404: "Dataset not found",
|
|
}
|
|
)
|
|
@service_api_ns.expect(service_api_ns.models[HitTestingPayload.__name__])
|
|
@cloud_edition_billing_rate_limit_check("knowledge", "dataset")
|
|
def post(self, tenant_id, dataset_id):
|
|
"""Perform hit testing on a dataset.
|
|
|
|
Tests retrieval performance for the specified dataset.
|
|
"""
|
|
dataset_id_str = str(dataset_id)
|
|
|
|
dataset = self.get_and_validate_dataset(dataset_id_str)
|
|
args = self.parse_args(service_api_ns.payload)
|
|
self.hit_testing_args_check(args)
|
|
|
|
return self.perform_hit_testing(dataset, args)
|