From 347547da00f81fc80ad6350c2402a173ffbc090c Mon Sep 17 00:00:00 2001 From: zhh Date: Wed, 26 Nov 2025 16:20:59 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=20=E9=9A=8F=E6=9C=BA?= =?UTF-8?q?=E6=B1=A0=E7=A7=8D=E7=B1=BB=E9=80=92=E5=87=8F=EF=BC=88=E6=8A=8A?= =?UTF-8?q?=E6=96=B0=E5=A2=9E=E7=9A=84=E7=B1=BB=E5=9E=8B=E4=BB=8E=E9=9A=8F?= =?UTF-8?q?=E6=9C=BA=E6=B1=A0=E4=B8=AD=E5=89=94=E9=99=A4=EF=BC=89=20?= =?UTF-8?q?=E9=9A=8F=E6=9C=BA=E6=B1=A0=E4=B8=AD=E5=89=94=E9=99=A4Socks?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/server/ChatbotAgent/core/stylist_agent_server.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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']}})