diff --git a/app/server/ChatbotAgent/core/stylist_agent_server.py b/app/server/ChatbotAgent/core/stylist_agent_server.py index 41bea62..88901ca 100644 --- a/app/server/ChatbotAgent/core/stylist_agent_server.py +++ b/app/server/ChatbotAgent/core/stylist_agent_server.py @@ -424,11 +424,14 @@ class AsyncStylistAgent: # 3. 检查终止条件 if gemini_data.get('action') == 'stop': - response_data['path'] = minio_path - response_data['items'].append({"item_id": item_id, "category": item_category}) - response_data['status'] = "ok" - response = post_request(url=url, data=json.dumps(response_data), headers=headers) - logger.info(f"request data :{response_data} | JAVA callback info -> status:{response.status_code} | message:{response.text}") + if is_duplicate_by_key(response_data['items'], {"item_id": item_id, "category": item_category}): + print("重复(按item_id判断),不插入") + else: + response_data['path'] = minio_path + response_data['items'].append({"item_id": item_id, "category": item_category}) + response_data['status'] = "ok" + response = post_request(url=url, data=json.dumps(response_data), headers=headers) + 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, len(self.outfit_items)) @@ -540,3 +543,11 @@ class AsyncStylistAgent: gcs_uri = f"gs://{bucket_name}/{blob_name}" return gcs_uri + + +def is_duplicate_by_key(data, target_item): + """基于item_id快速判断重复""" + # 提取所有item_id到集合 + existing_ids = {item['item_id'] for item in data} + # 判断目标item_id是否在集合中 + return target_item['item_id'] in existing_ids