修改删除

dbgpt/app/apps/vadmin/word/views.py

@@ -15,6 +15,8 @@ from dbgpt.app.apps.vadmin.word.params.sensitive import SensitiveParams
 from dbgpt.app.apps.vadmin.word import crud
 from fastapi.encoders import jsonable_encoder
 
+from ...core.dependencies import IdList
+
 router = APIRouter()
 
 
@@ -44,13 +46,8 @@ async def create_similar(qdata: Request, auth: Auth = Depends(OpenAuth())):
     return SuccessResponse(await crud.SimilarDal(auth.db).create_data(data=simi_data))
 
 @router.post("/delete_similars", summary="根据id删除同义词")
-async def delete_similars(data: Request, auth: Auth = Depends(OpenAuth())):
-    if data is None:
-        return ErrorResponse("删除同义词请求参数为空!")
-
-    data_id = data.query_params['word_id']
-    similarids = [data_id]
-    await crud.SimilarDal(auth.db).delete_datas(ids=similarids, v_soft=True)
+async def delete_similars(ids: IdList = Depends(), auth: Auth = Depends(OpenAuth())):
+    await crud.SimilarDal(auth.db).delete_datas(ids=ids.ids, v_soft=True)
     return SuccessResponse("删除成功")
 
 @router.post("/update_similar_byid", summary="根据id修改同义词")
@@ -117,14 +114,6 @@ async def update_sensitive_byid(data: Request, auth: Auth = Depends(OpenAuth()))
     return SuccessResponse("修改成功")
 
 @router.post("/delete_sensitives", summary="根据id删除敏感词")
-async def delete_sensitives(data: Request, auth: Auth = Depends(OpenAuth())):
-    if data is None:
-        print('sensitive request is None')
-        return ErrorResponse("不能删除当前敏感词")
-    else:
-        print(f"word_id：{data.query_params['word_id']} ")
-
-    data_id = data.query_params['word_id']
-    sensitiveids = [data_id]
-    await crud.SensitiveDal(auth.db).delete_datas(ids=sensitiveids, v_soft=True)
+async def delete_sensitives(ids: IdList = Depends(), auth: Auth = Depends(OpenAuth())):
+    await crud.SensitiveDal(auth.db).delete_datas(ids=ids.ids, v_soft=True)
     return SuccessResponse("删除成功")
\ No newline at end of file