mirror of
https://github.com/langgenius/dify.git
synced 2026-11-11 13:51:25 +08:00
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
"""Console availability of the KnowledgeFS product surfaces."""
|
|
|
|
from http import HTTPStatus
|
|
|
|
from flask_restx import Resource
|
|
|
|
from controllers.common.schema import register_response_schema_models
|
|
from controllers.console import console_ns
|
|
from controllers.console.wraps import account_initialization_required, setup_required
|
|
from fields.base import ResponseModel
|
|
from libs.helper import dump_response
|
|
from libs.login import login_required
|
|
from services.knowledge_fs.feature_service import get_knowledge_fs_features
|
|
|
|
|
|
class KnowledgeFSFeaturesResponse(ResponseModel):
|
|
upload_enabled: bool
|
|
agent_enabled: bool
|
|
|
|
|
|
register_response_schema_models(console_ns, KnowledgeFSFeaturesResponse)
|
|
|
|
|
|
@console_ns.route("/knowledge-fs/features")
|
|
class KnowledgeFSFeaturesApi(Resource):
|
|
@console_ns.doc(description="Get which Agent Knowledge Base capabilities this deployment offers")
|
|
@console_ns.response(
|
|
HTTPStatus.OK,
|
|
"Agent Knowledge Base availability",
|
|
console_ns.models[KnowledgeFSFeaturesResponse.__name__],
|
|
)
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
def get(self):
|
|
return dump_response(KnowledgeFSFeaturesResponse, get_knowledge_fs_features())
|