添加获取机器人角色接口v1, 返回所有机器人角色和风格
This commit is contained in:
parent
3f32e563ab
commit
35fe932eda
@ -35,6 +35,22 @@ async def get_system_role_pairing_list(
|
||||
|
||||
return ResponseUtil.success(data={"role": pairing_list, "style":style})
|
||||
|
||||
# 获取机器人角色新版
|
||||
@pairingController.get(
|
||||
'/v1/{robot_id}',
|
||||
)
|
||||
async def get_system_role_pairing_list_v1(
|
||||
request: Request,
|
||||
robot_id: int,
|
||||
query_db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
|
||||
pairing_list, style = await PairingService.get_role_pairing_list_services_v1(query_db, robot_id)
|
||||
|
||||
return ResponseUtil.success(data={"role": pairing_list, "style":style})
|
||||
|
||||
|
||||
|
||||
@pairingController.post(
|
||||
'/change',
|
||||
)
|
||||
|
||||
@ -34,19 +34,43 @@ class PairingService:
|
||||
@classmethod
|
||||
async def get_role_pairing_list_services(cls, query_db: AsyncSession, robot_id: int):
|
||||
pairing_list = await PairingDao.get_role_pairing_list(query_db, robot_id)
|
||||
|
||||
|
||||
# style_list = list()
|
||||
style = None
|
||||
# print(pairing_list)
|
||||
for pairing in pairing_list:
|
||||
|
||||
|
||||
if pairing["status"] == '1':
|
||||
style = await Explanation_style_robot_pairDao.get_robot_role_style(query_db, robot_id, pairing["robot_role_id"])
|
||||
# style_list.append(style)
|
||||
|
||||
|
||||
|
||||
return pairing_list, style
|
||||
|
||||
@classmethod
|
||||
async def get_role_pairing_list_services_v1(cls, query_db: AsyncSession, robot_id: int) -> list:
|
||||
""" 获取机器人角色对齐风格列表
|
||||
|
||||
Args:
|
||||
query_db (AsyncSession): db
|
||||
robot_id (int): 机器人id
|
||||
|
||||
Returns:
|
||||
list: 机器人角色风格对齐列表
|
||||
"""
|
||||
|
||||
pairing_list = await PairingDao.get_role_pairing_list(query_db, robot_id)
|
||||
|
||||
style_list = list()
|
||||
|
||||
for pairing in pairing_list:
|
||||
|
||||
style = await Explanation_style_robot_pairDao.get_robot_role_style(
|
||||
query_db, robot_id, pairing["robot_role_id"]
|
||||
)
|
||||
style_list.append(style)
|
||||
|
||||
return pairing_list, style_list
|
||||
|
||||
@classmethod
|
||||
async def get_pairing_list_services(
|
||||
cls, query_db: AsyncSession, query_object: PairingPageQueryModel, is_page: bool = False
|
||||
@ -63,7 +87,6 @@ class PairingService:
|
||||
|
||||
return pairing_list_result
|
||||
|
||||
|
||||
@classmethod
|
||||
async def add_pairing_services(cls, query_db: AsyncSession, page_object: PairingModel):
|
||||
"""
|
||||
|
||||
Loading…
Reference in New Issue
Block a user