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

避免重复item加入回调参数中

Browse Source
This commit is contained in:
zhh
2025-11-27 11:09:19 +08:00
parent 9451e95667
commit ee695e7511
1 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -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