添加获取机器人角色接口v1, 返回所有机器人角色和风格

This commit is contained in:
haotian 2025-09-16 14:14:16 +08:00
parent 3f32e563ab
commit 35fe932eda
2 changed files with 44 additions and 5 deletions

View File

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

View File

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