aidlab/lc_stylist_agent
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增 取消agent配饰（保留鞋子）推荐，改为默认随机配饰搭配 使用json文件补充stylist删除掉的必要配饰

Browse Source
This commit is contained in:
zhh
2025-11-21 11:19:27 +08:00
parent 2d38e3fc0b
commit 38af75077d
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
app/server/ChatbotAgent/core/stylist_agent_server.py
View File

@@ -293,7 +293,7 @@ class AsyncStylistAgent:
print(f"An error occurred during item retrieval: {e}")
return None
async def _get_random_accessories(self, stylist):
async def _get_random_accessories(self, stylist, item_count):
stylist_item = []
stylist_item_ids = []
for i in stylist:
@@ -303,12 +303,16 @@ class AsyncStylistAgent:
stylist_item += random.choices(stylist_results['metadatas'][0], k=i['count'])
stylist_item_ids += [item_id['item_id'] for item_id in stylist_item]
accessories_count = 2
accessories_count = 9 - item_count - len(stylist_item)
# 2. 在配饰池中过滤掉已经选中的item ，然后抽两个item
random_single_ids = random.choices(list(set(self.local_db.cache_filtered_ids) - set([f"{i}_img" for i in stylist_item_ids])), k=accessories_count)
random_items = self.local_db.random_get_accessories(random_single_ids)['metadatas']
all_items = stylist_item + random_items
if accessories_count > 0:
accessories_count = random.randint(1, accessories_count)
# 2. 在配饰池中过滤掉已经选中的item ，然后抽两个item
random_single_ids = random.choices(list(set(self.local_db.cache_filtered_ids) - set([f"{i}_img" for i in stylist_item_ids])), k=accessories_count)
random_items = self.local_db.random_get_accessories(random_single_ids)['metadatas']
all_items = stylist_item + random_items
else:
all_items = stylist_item
items_data = []
@@ -393,7 +397,7 @@ class AsyncStylistAgent:
logger.info(f"request data ：{response_data} | JAVA callback info -> status:{response.status_code} | message:{response.text}")
# 根据stylist要求随机增加配饰 3-4个配饰
new_item = await self._get_random_accessories(self.style_accessories_guide)
new_item = await self._get_random_accessories(self.style_accessories_guide, len(self.outfit_items))
for item in new_item:
self.outfit_items.append(item)
response_data['items'].append({"item_id": item.get('item_id'), "category": item.get('category')})
@@ -476,7 +480,7 @@ class AsyncStylistAgent:
logger.info(f"request data ：{response_data} | JAVA callback info -> status:{response.status_code} | message:{response.text}")
# 根据stylist要求随机增加配饰 3-4个配饰
new_item = await self._get_random_accessories(self.style_accessories_guide)
new_item = await self._get_random_accessories(self.style_accessories_guide, len(self.outfit_items))
for item in new_item:
self.outfit_items.append(item)
response_data['items'].append({"item_id": item.get('item_id'), "category": item.get('category')})