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