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

新增 随机池种类递减（把新增的类型从随机池中剔除） 随机池中剔除Socks

Browse Source
This commit is contained in:
zhh
2025-11-26 16:20:59 +08:00
parent ff5bdecb69
commit 347547da00
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -298,9 +298,11 @@ class AsyncStylistAgent:
stylist_item = [] stylist_item = []
stylist_item_ids = [] stylist_item_ids = []
# 初始过滤类别
filter_items = [ filter_items = [
{"item_group_id": {"$ne": "Clothing"}}, {"item_group_id": {"$ne": "Clothing"}},
{"item_group_id": {"$ne": "Shoes"}}, {"item_group_id": {"$ne": "Shoes"}},
{"category": {"$ne": "Socks"}},
{"modality": "image"} {"modality": "image"}
] ]
random_items = [] random_items = []
@@ -324,12 +326,12 @@ class AsyncStylistAgent:
logger.info(f"random_poll 数量: {len(random_poll)}") logger.info(f"random_poll 数量: {len(random_poll)}")
item = self.local_db.random_get_accessories(random.choice(random_poll)) item = self.local_db.random_get_accessories(random.choice(random_poll))
# 如果随机选中了包类 则所有包类别都过滤掉
if item['metadatas'][0]['category'] in ['Shopping Totes', 'Handbags', 'Backpacks', 'Briefcases']: if item['metadatas'][0]['category'] in ['Shopping Totes', 'Handbags', 'Backpacks', 'Briefcases']:
filter_items.append({"category": {"$ne": "Shopping Totes"}}) filter_items.append({"category": {"$ne": "Shopping Totes"}})
filter_items.append({"category": {"$ne": "Handbags"}}) filter_items.append({"category": {"$ne": "Handbags"}})
filter_items.append({"category": {"$ne": "Backpacks"}}) filter_items.append({"category": {"$ne": "Backpacks"}})
filter_items.append({"category": {"$ne": "Briefcases"}}) filter_items.append({"category": {"$ne": "Briefcases"}})
filter_items.append({"category": {"$ne": "Socks"}})
else: else:
filter_items.append({"category": {"$ne": item['metadatas'][0]['category']}}) filter_items.append({"category": {"$ne": item['metadatas'][0]['category']}})