youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-20 10:47:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: endpoint management

Browse Source
This commit is contained in:
Yeuoly
2024-09-26 12:49:00 +08:00
parent 153dc5b3f3
commit ea497f828f
7 changed files with 268 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
api/controllers/console/workspace/endpoint.py
View File

@ -1,9 +1,12 @@
from flask_restful import Resource
from flask_login import current_user
from flask_restful import Resource, reqparse
from werkzeug.exceptions import Forbidden
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.plugin.endpoint_service import EndpointService
class EndpointCreateApi(Resource):
@ -11,7 +14,24 @@ class EndpointCreateApi(Resource):
@login_required
@account_initialization_required
def post(self):
pass
user = current_user
if not user.is_admin_or_owner:
raise Forbidden()
parser = reqparse.RequestParser()
parser.add_argument("plugin_unique_identifier", type=str, required=True)
parser.add_argument("settings", type=dict, required=True)
args = parser.parse_args()
plugin_unique_identifier = args["plugin_unique_identifier"]
settings = args["settings"]
return EndpointService.create_endpoint(
tenant_id=user.current_tenant_id,
user_id=user.id,
plugin_unique_identifier=plugin_unique_identifier,
settings=settings,
)
class EndpointListApi(Resource):
@ -19,7 +39,12 @@ class EndpointListApi(Resource):
@login_required
@account_initialization_required
def get(self):
pass
user = current_user
return EndpointService.list_endpoints(
tenant_id=user.current_tenant_id,
user_id=user.id,
)
class EndpointDeleteApi(Resource):
@ -27,7 +52,17 @@ class EndpointDeleteApi(Resource):
@login_required
@account_initialization_required
def post(self):
pass
user = current_user
parser = reqparse.RequestParser()
parser.add_argument("endpoint_id", type=str, required=True)
args = parser.parse_args()
endpoint_id = args["endpoint_id"]
return EndpointService.delete_endpoint(
tenant_id=user.current_tenant_id, user_id=user.id, endpoint_id=endpoint_id
)
class EndpointUpdateApi(Resource):
@ -35,7 +70,22 @@ class EndpointUpdateApi(Resource):
@login_required
@account_initialization_required
def post(self):
pass
user = current_user
parser = reqparse.RequestParser()
parser.add_argument("endpoint_id", type=str, required=True)
parser.add_argument("settings", type=dict, required=True)
args = parser.parse_args()
endpoint_id = args["endpoint_id"]
settings = args["settings"]
return EndpointService.update_endpoint(
tenant_id=user.current_tenant_id,
user_id=user.id,
endpoint_id=endpoint_id,
settings=settings,
)
class EndpointEnableApi(Resource):
@ -43,7 +93,17 @@ class EndpointEnableApi(Resource):
@login_required
@account_initialization_required
def post(self):
pass
user = current_user
parser = reqparse.RequestParser()
parser.add_argument("endpoint_id", type=str, required=True)
args = parser.parse_args()
endpoint_id = args["endpoint_id"]
return EndpointService.enable_endpoint(
tenant_id=user.current_tenant_id, user_id=user.id, endpoint_id=endpoint_id
)
class EndpointDisableApi(Resource):
@ -51,7 +111,17 @@ class EndpointDisableApi(Resource):
@login_required
@account_initialization_required
def post(self):
pass
user = current_user
parser = reqparse.RequestParser()
parser.add_argument("endpoint_id", type=str, required=True)
args = parser.parse_args()
endpoint_id = args["endpoint_id"]
return EndpointService.disable_endpoint(
tenant_id=user.current_tenant_id, user_id=user.id, endpoint_id=endpoint_id
)
api.add_resource(EndpointCreateApi, "/workspaces/current/endpoints/create")
@ -59,4 +129,4 @@ api.add_resource(EndpointListApi, "/workspaces/current/endpoints/list")
api.add_resource(EndpointDeleteApi, "/workspaces/current/endpoints/delete")
api.add_resource(EndpointUpdateApi, "/workspaces/current/endpoints/update")
api.add_resource(EndpointEnableApi, "/workspaces/current/endpoints/enable")
api.add_resource(EndpointDisableApi, "/workspaces/current/endpoints/disable")
api.add_resource(EndpointDisableApi, "/workspaces/current/endpoints/disable")