mirror of
https://github.com/langgenius/dify.git
synced 2024-11-16 19:59:50 +08:00
13be84e4d4
Some checks are pending
Build and Push API & Web / build (api, DIFY_API_IMAGE_NAME, linux/amd64, build-api-amd64) (push) Waiting to run
Build and Push API & Web / build (api, DIFY_API_IMAGE_NAME, linux/arm64, build-api-arm64) (push) Waiting to run
Build and Push API & Web / build (web, DIFY_WEB_IMAGE_NAME, linux/amd64, build-web-amd64) (push) Waiting to run
Build and Push API & Web / build (web, DIFY_WEB_IMAGE_NAME, linux/arm64, build-web-arm64) (push) Waiting to run
Build and Push API & Web / create-manifest (api, DIFY_API_IMAGE_NAME, merge-api-images) (push) Blocked by required conditions
Build and Push API & Web / create-manifest (web, DIFY_WEB_IMAGE_NAME, merge-web-images) (push) Blocked by required conditions
93 lines
3.3 KiB
Python
93 lines
3.3 KiB
Python
from flask_restful import Resource, reqparse
|
|
|
|
from controllers.console import api
|
|
from controllers.console.app.error import TracingConfigCheckError, TracingConfigIsExist, TracingConfigNotExist
|
|
from controllers.console.setup import setup_required
|
|
from controllers.console.wraps import account_initialization_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 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 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 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"}
|
|
except Exception as e:
|
|
raise e
|
|
|
|
|
|
api.add_resource(TraceAppConfigApi, "/apps/<uuid:app_id>/trace-config")
|