Merge branch 'feat/external-knowledge-api' of github.com:langgenius/dify into feat/external-knowledge-api

feat/external-knowledge-api
Yi 2 years ago
commit b9b8ec1758

@ -63,8 +63,8 @@ class ExternalApiTemplateListApi(Resource):
) )
parser.add_argument( parser.add_argument(
"description", "description",
nullable=False, nullable=True,
required=True, required=False,
help="Description is required. Description must be between 1 to 400 characters.", help="Description is required. Description must be between 1 to 400 characters.",
type=_validate_description_length, type=_validate_description_length,
) )
@ -97,9 +97,9 @@ class ExternalApiTemplateApi(Resource):
@setup_required @setup_required
@login_required @login_required
@account_initialization_required @account_initialization_required
def get(self, api_template_id): def get(self, external_knowledge_api_id):
api_template_id = str(api_template_id) external_knowledge_api_id = str(external_knowledge_api_id)
api_template = ExternalDatasetService.get_api_template(api_template_id) api_template = ExternalDatasetService.get_api_template(external_knowledge_api_id)
if api_template is None: if api_template is None:
raise NotFound("API template not found.") raise NotFound("API template not found.")
@ -121,8 +121,8 @@ class ExternalApiTemplateApi(Resource):
) )
parser.add_argument( parser.add_argument(
"description", "description",
nullable=False, nullable=True,
required=True, required=False,
help="description is required. Description must be between 1 to 400 characters.", help="description is required. Description must be between 1 to 400 characters.",
type=_validate_description_length, type=_validate_description_length,
) )
@ -163,10 +163,10 @@ class ExternalApiUseCheckApi(Resource):
@setup_required @setup_required
@login_required @login_required
@account_initialization_required @account_initialization_required
def get(self, api_template_id): def get(self, external_knowledge_api_id):
api_template_id = str(api_template_id) external_knowledge_api_id = str(external_knowledge_api_id)
external_api_template_is_using = ExternalDatasetService.external_api_template_use_check(api_template_id) external_api_template_is_using = ExternalDatasetService.external_api_template_use_check(external_knowledge_api_id)
return {"is_using": external_api_template_is_using}, 200 return {"is_using": external_api_template_is_using}, 200
@ -289,6 +289,6 @@ class ExternalKnowledgeHitTestingApi(Resource):
api.add_resource(ExternalKnowledgeHitTestingApi, "/datasets/<uuid:dataset_id>/external-hit-testing") api.add_resource(ExternalKnowledgeHitTestingApi, "/datasets/<uuid:dataset_id>/external-hit-testing")
api.add_resource(ExternalDatasetCreateApi, "/datasets/external") api.add_resource(ExternalDatasetCreateApi, "/datasets/external")
api.add_resource(ExternalApiTemplateListApi, "/datasets/external-api-template") api.add_resource(ExternalApiTemplateListApi, "/datasets/external-knowledge-api")
api.add_resource(ExternalApiTemplateApi, "/datasets/external-api-template/<uuid:api_template_id>") api.add_resource(ExternalApiTemplateApi, "/datasets/external-knowledge-api/<uuid:external_knowledge_api_id>")
api.add_resource(ExternalApiUseCheckApi, "/datasets/external-api-template/<uuid:api_template_id>/use-check") api.add_resource(ExternalApiUseCheckApi, "/datasets/external-knowledge-api/<uuid:external_knowledge_api_id>/use-check")

@ -61,8 +61,8 @@ class ExternalDatasetService:
return api_template return api_template
@staticmethod @staticmethod
def get_api_template(api_template_id: str) -> ExternalApiTemplates: def get_api_template(external_knowledge_api_id: str) -> ExternalApiTemplates:
return ExternalApiTemplates.query.filter_by(id=api_template_id).first() return ExternalApiTemplates.query.filter_by(id=external_knowledge_api_id).first()
@staticmethod @staticmethod
def update_api_template(tenant_id, user_id, api_template_id, args) -> ExternalApiTemplates: def update_api_template(tenant_id, user_id, api_template_id, args) -> ExternalApiTemplates:
@ -89,8 +89,8 @@ class ExternalDatasetService:
db.session.commit() db.session.commit()
@staticmethod @staticmethod
def external_api_template_use_check(api_template_id: str) -> bool: def external_api_template_use_check(external_knowledge_api_id: str) -> bool:
count = ExternalKnowledgeBindings.query.filter_by(external_api_template_id=api_template_id).count() count = ExternalKnowledgeBindings.query.filter_by(external_api_template_id=external_knowledge_api_id).count()
if count > 0: if count > 0:
return True return True
return False return False

Loading…
Cancel
Save