mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-27 02:52:48 +00:00
feat: implement knowledge creation APIs for base, web, lark, and yuque
This commit is contained in:
parent
724686762e
commit
f7a4aca0e8
|
|
@ -11,7 +11,73 @@ class KnowledgeCreateResponse(ResultSerializer):
|
|||
return KnowledgeModelSerializer()
|
||||
|
||||
|
||||
class KnowledgeCreateAPI(APIMixin):
|
||||
class KnowledgeBaseCreateAPI(APIMixin):
|
||||
@staticmethod
|
||||
def get_parameters():
|
||||
return [
|
||||
OpenApiParameter(
|
||||
name="workspace_id",
|
||||
description="工作空间id",
|
||||
type=OpenApiTypes.STR,
|
||||
location='path',
|
||||
required=True,
|
||||
)
|
||||
]
|
||||
|
||||
@staticmethod
|
||||
def get_request():
|
||||
return KnowledgeBaseCreateRequest
|
||||
|
||||
@staticmethod
|
||||
def get_response():
|
||||
return KnowledgeCreateResponse
|
||||
|
||||
|
||||
class KnowledgeWebCreateAPI(APIMixin):
|
||||
@staticmethod
|
||||
def get_parameters():
|
||||
return [
|
||||
OpenApiParameter(
|
||||
name="workspace_id",
|
||||
description="工作空间id",
|
||||
type=OpenApiTypes.STR,
|
||||
location='path',
|
||||
required=True,
|
||||
)
|
||||
]
|
||||
|
||||
@staticmethod
|
||||
def get_request():
|
||||
return KnowledgeBaseCreateRequest
|
||||
|
||||
@staticmethod
|
||||
def get_response():
|
||||
return KnowledgeCreateResponse
|
||||
|
||||
|
||||
class KnowledgeLarkCreateAPI(APIMixin):
|
||||
@staticmethod
|
||||
def get_parameters():
|
||||
return [
|
||||
OpenApiParameter(
|
||||
name="workspace_id",
|
||||
description="工作空间id",
|
||||
type=OpenApiTypes.STR,
|
||||
location='path',
|
||||
required=True,
|
||||
)
|
||||
]
|
||||
|
||||
@staticmethod
|
||||
def get_request():
|
||||
return KnowledgeBaseCreateRequest
|
||||
|
||||
@staticmethod
|
||||
def get_response():
|
||||
return KnowledgeCreateResponse
|
||||
|
||||
|
||||
class KnowledgeYuqueCreateAPI(APIMixin):
|
||||
@staticmethod
|
||||
def get_parameters():
|
||||
return [
|
||||
|
|
|
|||
|
|
@ -5,4 +5,8 @@ from . import views
|
|||
app_name = "knowledge"
|
||||
urlpatterns = [
|
||||
path('workspace/<str:workspace_id>/knowledge', views.KnowledgeView.as_view()),
|
||||
path('workspace/<str:workspace_id>/knowledge/base', views.KnowledgeBaseView.as_view()),
|
||||
path('workspace/<str:workspace_id>/knowledge/web', views.KnowledgeWebView.as_view()),
|
||||
path('workspace/<str:workspace_id>/knowledge/lark', views.KnowledgeLarkView.as_view()),
|
||||
path('workspace/<str:workspace_id>/knowledge/yuque', views.KnowledgeYuqueView.as_view()),
|
||||
]
|
||||
|
|
|
|||
|
|
@ -7,28 +7,14 @@ from common.auth import TokenAuth
|
|||
from common.auth.authentication import has_permissions
|
||||
from common.constants.permission_constants import PermissionConstants
|
||||
from common.result import result
|
||||
from knowledge.api.knowledge import KnowledgeCreateAPI, KnowledgeTreeReadAPI
|
||||
from knowledge.api.knowledge import KnowledgeBaseCreateAPI, KnowledgeLarkCreateAPI, \
|
||||
KnowledgeWebCreateAPI, KnowledgeYuqueCreateAPI, KnowledgeTreeReadAPI
|
||||
from knowledge.serializers.knowledge import KnowledgeSerializer, KnowledgeTreeSerializer
|
||||
|
||||
|
||||
class KnowledgeView(APIView):
|
||||
authentication_classes = [TokenAuth]
|
||||
|
||||
@extend_schema(
|
||||
methods=['POST'],
|
||||
description=_('Create knowledge'),
|
||||
operation_id=_('Create knowledge'),
|
||||
parameters=KnowledgeCreateAPI.get_parameters(),
|
||||
request=KnowledgeCreateAPI.get_request(),
|
||||
responses=KnowledgeCreateAPI.get_response(),
|
||||
tags=[_('Knowledge Base')]
|
||||
)
|
||||
@has_permissions(PermissionConstants.KNOWLEDGE_CREATE.get_workspace_permission())
|
||||
def post(self, request: Request, workspace_id: str):
|
||||
return result.success(KnowledgeSerializer.Create(
|
||||
data={'user_id': request.user.id, 'workspace_id': workspace_id}
|
||||
).insert(request.data))
|
||||
|
||||
@extend_schema(
|
||||
methods=['GET'],
|
||||
description=_('Get knowledge by module'),
|
||||
|
|
@ -42,3 +28,79 @@ class KnowledgeView(APIView):
|
|||
return result.success(KnowledgeTreeSerializer(
|
||||
data={'workspace_id': workspace_id}
|
||||
).get_knowledge_list(request.query_params.get('module_id')))
|
||||
|
||||
|
||||
class KnowledgeBaseView(APIView):
|
||||
authentication_classes = [TokenAuth]
|
||||
|
||||
@extend_schema(
|
||||
methods=['POST'],
|
||||
description=_('Create base knowledge'),
|
||||
operation_id=_('Create base knowledge'),
|
||||
parameters=KnowledgeBaseCreateAPI.get_parameters(),
|
||||
request=KnowledgeBaseCreateAPI.get_request(),
|
||||
responses=KnowledgeBaseCreateAPI.get_response(),
|
||||
tags=[_('Knowledge Base')]
|
||||
)
|
||||
@has_permissions(PermissionConstants.KNOWLEDGE_CREATE.get_workspace_permission())
|
||||
def post(self, request: Request, workspace_id: str):
|
||||
return result.success(KnowledgeSerializer.Create(
|
||||
data={'user_id': request.user.id, 'workspace_id': workspace_id}
|
||||
).insert(request.data))
|
||||
|
||||
|
||||
class KnowledgeWebView(APIView):
|
||||
authentication_classes = [TokenAuth]
|
||||
|
||||
@extend_schema(
|
||||
methods=['POST'],
|
||||
description=_('Create web knowledge'),
|
||||
operation_id=_('Create web knowledge'),
|
||||
parameters=KnowledgeWebCreateAPI.get_parameters(),
|
||||
request=KnowledgeWebCreateAPI.get_request(),
|
||||
responses=KnowledgeWebCreateAPI.get_response(),
|
||||
tags=[_('Knowledge Base')]
|
||||
)
|
||||
@has_permissions(PermissionConstants.KNOWLEDGE_CREATE.get_workspace_permission())
|
||||
def post(self, request: Request, workspace_id: str):
|
||||
return result.success(KnowledgeSerializer.Create(
|
||||
data={'user_id': request.user.id, 'workspace_id': workspace_id}
|
||||
).insert(request.data))
|
||||
|
||||
|
||||
class KnowledgeLarkView(APIView):
|
||||
authentication_classes = [TokenAuth]
|
||||
|
||||
@extend_schema(
|
||||
methods=['POST'],
|
||||
description=_('Create lark knowledge'),
|
||||
operation_id=_('Create lark knowledge'),
|
||||
parameters=KnowledgeLarkCreateAPI.get_parameters(),
|
||||
request=KnowledgeLarkCreateAPI.get_request(),
|
||||
responses=KnowledgeLarkCreateAPI.get_response(),
|
||||
tags=[_('Knowledge Base')]
|
||||
)
|
||||
@has_permissions(PermissionConstants.KNOWLEDGE_CREATE.get_workspace_permission())
|
||||
def post(self, request: Request, workspace_id: str):
|
||||
return result.success(KnowledgeSerializer.Create(
|
||||
data={'user_id': request.user.id, 'workspace_id': workspace_id}
|
||||
).insert(request.data))
|
||||
|
||||
|
||||
class KnowledgeYuqueView(APIView):
|
||||
authentication_classes = [TokenAuth]
|
||||
|
||||
@extend_schema(
|
||||
methods=['POST'],
|
||||
description=_('Create yuque knowledge'),
|
||||
operation_id=_('Create yuque knowledge'),
|
||||
parameters=KnowledgeYuqueCreateAPI.get_parameters(),
|
||||
request=KnowledgeYuqueCreateAPI.get_request(),
|
||||
responses=KnowledgeYuqueCreateAPI.get_response(),
|
||||
tags=[_('Knowledge Base')]
|
||||
)
|
||||
@has_permissions(PermissionConstants.KNOWLEDGE_CREATE.get_workspace_permission())
|
||||
def post(self, request: Request, workspace_id: str):
|
||||
return result.success(KnowledgeSerializer.Create(
|
||||
data={'user_id': request.user.id, 'workspace_id': workspace_id}
|
||||
).insert(request.data))
|
||||
|
|
|
|||
Loading…
Reference in New Issue