feat: implement knowledge creation APIs for base, web, lark, and yuque

This commit is contained in:
CaptainB 2025-04-25 20:31:59 +08:00
parent 724686762e
commit f7a4aca0e8
3 changed files with 149 additions and 17 deletions

View File

@ -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 [

View File

@ -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()),
]

View File

@ -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))