Feat/new saas billing (#12591)

This commit is contained in:
Jyong 2025-01-12 14:50:46 +08:00 • committed by GitHub
commit d8f57bf899
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 144 additions and 3 deletions

View file

@ -2,7 +2,11 @@ from flask_restful import Resource # type: ignore
from controllers.console import api
from controllers.console.datasets.hit_testing_base import DatasetsHitTestingBase
from controllers.console.wraps import account_initialization_required, setup_required
from controllers.console.wraps import (
account_initialization_required,
cloud_edition_billing_rate_limit_check,
setup_required,
)
from libs.login import login_required
@ -10,6 +14,7 @@ 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)