mirror of
https://github.com/langgenius/dify.git
synced 2025-11-22 16:06:53 +00:00
2f50f3fd4b
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com>
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
from flask_restx import Resource, fields
|
|
|
|
from controllers.console import api, console_ns
|
|
from controllers.console.wraps import account_initialization_required, setup_required
|
|
from core.model_runtime.utils.encoders import jsonable_encoder
|
|
from libs.login import current_user, login_required
|
|
from models.account import Account
|
|
from services.agent_service import AgentService
|
|
|
|
|
|
@console_ns.route("/workspaces/current/agent-providers")
|
|
class AgentProviderListApi(Resource):
|
|
@api.doc("list_agent_providers")
|
|
@api.doc(description="Get list of available agent providers")
|
|
@api.response(
|
|
200,
|
|
"Success",
|
|
fields.List(fields.Raw(description="Agent provider information")),
|
|
)
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def get(self):
|
|
assert isinstance(current_user, Account)
|
|
user = current_user
|
|
assert user.current_tenant_id is not None
|
|
|
|
user_id = user.id
|
|
tenant_id = user.current_tenant_id
|
|
|
|
return jsonable_encoder(AgentService.list_agent_providers(user_id, tenant_id))
|
|
|
|
|
|
@console_ns.route("/workspaces/current/agent-provider/<path:provider_name>")
|
|
class AgentProviderApi(Resource):
|
|
@api.doc("get_agent_provider")
|
|
@api.doc(description="Get specific agent provider details")
|
|
@api.doc(params={"provider_name": "Agent provider name"})
|
|
@api.response(
|
|
200,
|
|
"Success",
|
|
fields.Raw(description="Agent provider details"),
|
|
)
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def get(self, provider_name: str):
|
|
assert isinstance(current_user, Account)
|
|
user = current_user
|
|
assert user.current_tenant_id is not None
|
|
user_id = user.id
|
|
tenant_id = user.current_tenant_id
|
|
return jsonable_encoder(AgentService.get_agent_provider(user_id, tenant_id, provider_name))
|