Skip to content

Commit e786551

Browse files
liuruibinliuruibin
committed
feat: Support dataset log recorder
1 parent bdad1b4 commit e786551

File tree

6 files changed

+72
-0
lines changed

6 files changed

+72
-0
lines changed

‎apps/dataset/views/dataset.py‎

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from common.auth import TokenAuth, has_permissions
1717
from common.constants.permission_constants import PermissionConstants, CompareConstants, Permission, Group, Operate, \
1818
ViewPermission, RoleConstants
19+
from common.log.log import log
1920
from common.response import result
2021
from common.response.result import get_page_request_params, get_page_api_response, get_api_response
2122
from common.swagger_api.common_api import CommonApi
@@ -42,6 +43,7 @@ class SyncWeb(APIView):
4243
dynamic_tag=keywords.get('dataset_id'))],
4344
compare=CompareConstants.AND), PermissionConstants.DATASET_EDIT,
4445
compare=CompareConstants.AND)
46+
@log(menu=_('Knowledge Base'), operate=_("同步Web站点知识库"))
4547
def put(self, request: Request, dataset_id: str):
4648
return result.success(DataSetSerializers.SyncWeb(
4749
data={'sync_type': request.query_params.get('sync_type'), 'id': dataset_id,
@@ -60,6 +62,7 @@ class CreateQADataset(APIView):
6062
tags=[_('Knowledge Base')]
6163
)
6264
@has_permissions(PermissionConstants.DATASET_CREATE, compare=CompareConstants.AND)
65+
@log(menu=_('Knowledge Base'), operate=_("创建QA知识库"))
6366
def post(self, request: Request):
6467
return result.success(DataSetSerializers.Create(data={'user_id': request.user.id}).save_qa({
6568
'file_list': request.FILES.getlist('file'),
@@ -79,6 +82,7 @@ class CreateWebDataset(APIView):
7982
tags=[_('Knowledge Base')]
8083
)
8184
@has_permissions(PermissionConstants.DATASET_CREATE, compare=CompareConstants.AND)
85+
@log(menu=_('Knowledge Base'), operate=_("Create a web site knowledge base"))
8286
def post(self, request: Request):
8387
return result.success(DataSetSerializers.Create(data={'user_id': request.user.id}).save_web(request.data))
8488

@@ -92,6 +96,7 @@ class Application(APIView):
9296
responses=result.get_api_array_response(
9397
DataSetSerializers.Application.get_response_body_api()),
9498
tags=[_('Knowledge Base')])
99+
@log(menu=_('Knowledge Base'), operate=_("Get a list of applications available in the knowledge base"))
95100
def get(self, request: Request, dataset_id: str):
96101
return result.success(DataSetSerializers.Operate(
97102
data={'id': dataset_id, 'user_id': str(request.user.id)}).list_application())
@@ -103,6 +108,7 @@ def get(self, request: Request, dataset_id: str):
103108
responses=result.get_api_array_response(DataSetSerializers.Query.get_response_body_api()),
104109
tags=[_('Knowledge Base')])
105110
@has_permissions(PermissionConstants.DATASET_READ, compare=CompareConstants.AND)
111+
@log(menu=_('Knowledge Base'), operate=_("Get a list of knowledge bases"))
106112
def get(self, request: Request):
107113
data = {key: str(value) for key, value in request.query_params.items()}
108114
d = DataSetSerializers.Query(data={**data, 'user_id': str(request.user.id)})
@@ -117,6 +123,7 @@ def get(self, request: Request):
117123
tags=[_('Knowledge Base')]
118124
)
119125
@has_permissions(PermissionConstants.DATASET_CREATE, compare=CompareConstants.AND)
126+
@log(menu=_('Knowledge Base'), operate=_("Create a knowledge base"))
120127
def post(self, request: Request):
121128
return result.success(DataSetSerializers.Create(data={'user_id': request.user.id}).save(request.data))
122129

@@ -130,6 +137,7 @@ class HitTest(APIView):
130137
tags=[_('Knowledge Base')])
131138
@has_permissions(lambda r, keywords: Permission(group=Group.DATASET, operate=Operate.USE,
132139
dynamic_tag=keywords.get('dataset_id')))
140+
@log(menu=_('Knowledge Base'), operate=_("Hit test list"))
133141
def get(self, request: Request, dataset_id: str):
134142
return result.success(
135143
DataSetSerializers.HitTest(data={'id': dataset_id, 'user_id': request.user.id,
@@ -150,6 +158,7 @@ class Embedding(APIView):
150158
)
151159
@has_permissions(lambda r, keywords: Permission(group=Group.DATASET, operate=Operate.MANAGE,
152160
dynamic_tag=keywords.get('dataset_id')))
161+
@log(menu=_('Knowledge Base'), operate=_("Re-vectorize"))
153162
def put(self, request: Request, dataset_id: str):
154163
return result.success(
155164
DataSetSerializers.Operate(data={'id': dataset_id, 'user_id': request.user.id}).re_embedding())
@@ -164,6 +173,7 @@ class Export(APIView):
164173
)
165174
@has_permissions(lambda r, keywords: Permission(group=Group.DATASET, operate=Operate.MANAGE,
166175
dynamic_tag=keywords.get('dataset_id')))
176+
@log(menu=_('Knowledge Base'), operate=_("Export knowledge base"))
167177
def get(self, request: Request, dataset_id: str):
168178
return DataSetSerializers.Operate(data={'id': dataset_id, 'user_id': request.user.id}).export_excel()
169179

@@ -178,6 +188,7 @@ class ExportZip(APIView):
178188
)
179189
@has_permissions(lambda r, keywords: Permission(group=Group.DATASET, operate=Operate.MANAGE,
180190
dynamic_tag=keywords.get('dataset_id')))
191+
@log(menu=_('Knowledge Base'), operate=_("Export knowledge base containing images"))
181192
def get(self, request: Request, dataset_id: str):
182193
return DataSetSerializers.Operate(data={'id': dataset_id, 'user_id': request.user.id}).export_zip()
183194

@@ -193,6 +204,7 @@ class Operate(APIView):
193204
dynamic_tag=keywords.get('dataset_id')),
194205
lambda r, k: Permission(group=Group.DATASET, operate=Operate.DELETE,
195206
dynamic_tag=k.get('dataset_id')), compare=CompareConstants.AND)
207+
@log(menu=_('Knowledge Base'), operate=_("Delete knowledge base"))
196208
def delete(self, request: Request, dataset_id: str):
197209
operate = DataSetSerializers.Operate(data={'id': dataset_id})
198210
return result.success(operate.delete())
@@ -205,6 +217,7 @@ def delete(self, request: Request, dataset_id: str):
205217
tags=[_('Knowledge Base')])
206218
@has_permissions(lambda r, keywords: Permission(group=Group.DATASET, operate=Operate.USE,
207219
dynamic_tag=keywords.get('dataset_id')))
220+
@log(menu=_('Knowledge Base'), operate=_("Query knowledge base details based on knowledge base id"))
208221
def get(self, request: Request, dataset_id: str):
209222
return result.success(DataSetSerializers.Operate(data={'id': dataset_id, 'user_id': request.user.id}).one(
210223
user_id=request.user.id))
@@ -219,6 +232,7 @@ def get(self, request: Request, dataset_id: str):
219232
)
220233
@has_permissions(lambda r, keywords: Permission(group=Group.DATASET, operate=Operate.MANAGE,
221234
dynamic_tag=keywords.get('dataset_id')))
235+
@log(menu=_('Knowledge Base'), operate=_("Modify knowledge base information"))
222236
def put(self, request: Request, dataset_id: str):
223237
return result.success(
224238
DataSetSerializers.Operate(data={'id': dataset_id, 'user_id': request.user.id}).edit(request.data,
@@ -236,6 +250,7 @@ class Page(APIView):
236250
tags=[_('Knowledge Base')]
237251
)
238252
@has_permissions(PermissionConstants.DATASET_READ, compare=CompareConstants.AND)
253+
@log(menu=_('Knowledge Base'), operate=_("Get the knowledge base paginated list"))
239254
def get(self, request: Request, current_page, page_size):
240255
d = DataSetSerializers.Query(
241256
data={'name': request.query_params.get('name', None), 'desc': request.query_params.get("desc", None),
@@ -253,6 +268,7 @@ class Model(APIView):
253268
[lambda r, keywords: Permission(group=Group.DATASET, operate=Operate.MANAGE,
254269
dynamic_tag=keywords.get('dataset_id'))],
255270
compare=CompareConstants.AND))
271+
@log(menu=_('Knowledge Base'), operate=_("Get the model list of the knowledge base"))
256272
def get(self, request: Request, dataset_id: str):
257273
return result.success(
258274
ModelSerializer.Query(

0 commit comments

Comments
 (0)