mirror of
https://github.com/langgenius/dify.git
synced 2025-10-31 02:42:59 +00:00
93 lines
3.4 KiB
Python
93 lines
3.4 KiB
Python
from flask_restful import Resource, reqparse
|
|
from werkzeug.exceptions import BadRequest
|
|
|
|
from controllers.console import api
|
|
from controllers.console.app.error import TracingConfigCheckError, TracingConfigIsExist, TracingConfigNotExist
|
|
from controllers.console.wraps import account_initialization_required, setup_required
|
|
from libs.login import login_required
|
|
from services.ops_service import OpsService
|
|
|
|
|
|
class TraceAppConfigApi(Resource):
|
|
"""
|
|
Manage trace app configurations
|
|
"""
|
|
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def get(self, app_id):
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument("tracing_provider", type=str, required=True, location="args")
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
trace_config = OpsService.get_tracing_app_config(app_id=app_id, tracing_provider=args["tracing_provider"])
|
|
if not trace_config:
|
|
return {"has_not_configured": True}
|
|
return trace_config
|
|
except Exception as e:
|
|
raise BadRequest(str(e))
|
|
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def post(self, app_id):
|
|
"""Create a new trace app configuration"""
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument("tracing_provider", type=str, required=True, location="json")
|
|
parser.add_argument("tracing_config", type=dict, required=True, location="json")
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
result = OpsService.create_tracing_app_config(
|
|
app_id=app_id, tracing_provider=args["tracing_provider"], tracing_config=args["tracing_config"]
|
|
)
|
|
if not result:
|
|
raise TracingConfigIsExist()
|
|
if result.get("error"):
|
|
raise TracingConfigCheckError()
|
|
return result
|
|
except Exception as e:
|
|
raise BadRequest(str(e))
|
|
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def patch(self, app_id):
|
|
"""Update an existing trace app configuration"""
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument("tracing_provider", type=str, required=True, location="json")
|
|
parser.add_argument("tracing_config", type=dict, required=True, location="json")
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
result = OpsService.update_tracing_app_config(
|
|
app_id=app_id, tracing_provider=args["tracing_provider"], tracing_config=args["tracing_config"]
|
|
)
|
|
if not result:
|
|
raise TracingConfigNotExist()
|
|
return {"result": "success"}
|
|
except Exception as e:
|
|
raise BadRequest(str(e))
|
|
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def delete(self, app_id):
|
|
"""Delete an existing trace app configuration"""
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument("tracing_provider", type=str, required=True, location="args")
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
result = OpsService.delete_tracing_app_config(app_id=app_id, tracing_provider=args["tracing_provider"])
|
|
if not result:
|
|
raise TracingConfigNotExist()
|
|
return {"result": "success"}, 204
|
|
except Exception as e:
|
|
raise BadRequest(str(e))
|
|
|
|
|
|
api.add_resource(TraceAppConfigApi, "/apps/<uuid:app_id>/trace-config")
|