mirror of
https://github.com/YFGaia/dify-plus.git
synced 2026-06-14 20:41:21 +08:00
npc0-hue
110 lines
3.5 KiB
Python
110 lines
3.5 KiB
Python
from flask_login import current_user
|
|
from flask_restx import Resource, marshal_with
|
|
from werkzeug.exceptions import Forbidden
|
|
|
|
from controllers.console import api
|
|
from controllers.console.app.wraps import get_app_model
|
|
from controllers.console.wraps import (
|
|
account_initialization_required,
|
|
setup_required,
|
|
)
|
|
from fields.app_fields_extend import (
|
|
recommended_app_list_fields,
|
|
)
|
|
from libs.login import login_required
|
|
from services.account_service_extend import TenantExtendService
|
|
from services.recommended_app_service_extend import RecommendedAppService
|
|
|
|
|
|
# ---------------- start sync app to
|
|
class InstalledSyncAppApi(Resource):
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@marshal_with(recommended_app_list_fields)
|
|
def get(self):
|
|
"""Installed app"""
|
|
|
|
app_service = RecommendedAppService()
|
|
|
|
return app_service.installed_app_list(current_user.current_tenant_id)
|
|
|
|
|
|
class AppSyncApi(Resource):
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@get_app_model
|
|
def put(self, app_model):
|
|
"""Sync app"""
|
|
|
|
# The role of the current user in the ta table must be admin or owner
|
|
tenant_extend_service = TenantExtendService
|
|
super_admin_id = tenant_extend_service.get_super_admin_id().id
|
|
if super_admin_id != current_user.id:
|
|
raise Forbidden()
|
|
|
|
app_service = RecommendedAppService()
|
|
|
|
appId = app_service.sync_recommended_app(app_model.id)
|
|
|
|
return appId, 200
|
|
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@get_app_model
|
|
def delete(self, app_model):
|
|
"""Delete sync app"""
|
|
# The role of the current user in the ta table must be admin or owner
|
|
tenant_extend_service = TenantExtendService
|
|
super_admin_id = tenant_extend_service.get_super_admin_id().id
|
|
if super_admin_id != current_user.id:
|
|
raise Forbidden()
|
|
|
|
app_service = RecommendedAppService()
|
|
|
|
app_service.delete_sync_recommended_app(app_model.id)
|
|
|
|
return "", 200
|
|
|
|
|
|
# Extend: start messages context handling
|
|
class MessageContextApi(Resource):
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def get(self):
|
|
"""Message Context"""
|
|
from flask import request
|
|
conversation_id = request.args.get("conversation_id")
|
|
if not conversation_id:
|
|
from werkzeug.exceptions import BadRequest
|
|
raise BadRequest("conversation_id is required")
|
|
app_service = RecommendedAppService()
|
|
|
|
return app_service.message_context(conversation_id)
|
|
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def delete(self):
|
|
"""Message Context"""
|
|
from flask import request
|
|
message_id = request.args.get("message_id")
|
|
conversation_id = request.args.get("conversation_id")
|
|
if not message_id or not conversation_id:
|
|
from werkzeug.exceptions import BadRequest
|
|
raise BadRequest("message_id and conversation_id are required")
|
|
app_service = RecommendedAppService()
|
|
|
|
return app_service.delete_message_context(conversation_id, message_id)
|
|
# Extend: stop messages context handling
|
|
|
|
|
|
# ----------------start sync app------------------------
|
|
api.add_resource(AppSyncApi, "/apps/<uuid:app_id>/sync")
|
|
api.add_resource(InstalledSyncAppApi, "/installed/apps")
|
|
api.add_resource(MessageContextApi, "/message/context")
|
|
# ---------------- stop sync app ------------------------
|