43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
from flask_restful import Resource, reqparse
|
|
|
|
from controllers.console import api
|
|
from controllers.console.setup import setup_required
|
|
from controllers.console.wraps import account_initialization_required
|
|
from libs.login import login_required
|
|
from services.external_knowledge_service import ExternalDatasetService
|
|
|
|
|
|
class TestExternalApi(Resource):
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def post(self):
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument(
|
|
"retrieval_setting",
|
|
nullable=False,
|
|
required=True,
|
|
type=dict,
|
|
location="json"
|
|
)
|
|
parser.add_argument(
|
|
"query",
|
|
nullable=False,
|
|
required=True,
|
|
type=str,
|
|
)
|
|
parser.add_argument(
|
|
"knowledge_id",
|
|
nullable=False,
|
|
required=True,
|
|
type=str,
|
|
)
|
|
args = parser.parse_args()
|
|
result = ExternalDatasetService.test_external_knowledge_retrieval(
|
|
args["retrieval_setting"], args["query"], args["knowledge_id"]
|
|
)
|
|
return result, 200
|
|
|
|
|
|
api.add_resource(TestExternalApi, "/retrieval")
|