From f91e065d274686cbca08982ad29661e3c4f23ad5 Mon Sep 17 00:00:00 2001
From: zzx <zzx@suq.cn>
Date: Sat, 20 Sep 2025 22:25:02 +0800
Subject: [PATCH] =?UTF-8?q?feat(ai):=20=E6=9B=B4=E6=96=B0=E5=93=81?=
 =?UTF-8?q?=E7=89=8C=E8=AF=8D=E6=8F=90=E5=8F=96=E4=BB=BB=E5=8A=A1=E8=A6=81?=
 =?UTF-8?q?=E6=B1=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 不再对提取的品牌词进行去重处理
- 提取文本中提到的所有品牌，包括文本总结中的品牌- 调整提示词以适应新的提取逻辑
- 修改测试入口函数名
---
 main.py     | 2 +-
 utils/ai.py | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index bb785d8..d0a7964 100644
--- a/main.py
+++ b/main.py
@@ -102,4 +102,4 @@ async def check_session(platform_ids=None):
 
 
 if __name__ == '__main__':
-    asyncio.get_event_loop().run_until_complete(check_session_by_platform_id(6))
+    asyncio.get_event_loop().run_until_complete(test())
diff --git a/utils/ai.py b/utils/ai.py
index 4e47976..e8d83b5 100644
--- a/utils/ai.py
+++ b/utils/ai.py
@@ -20,8 +20,9 @@ async def main():
             logger.info(f"[{result['id']}] 已读取过排名")
             continue
         prompt = f"""
-        任务: 请在以下文本中, 按出现的顺序提取出品牌词, 多次出现的品牌词仅提取一次, 返回json数组
-        返回格式: json中包含brands字段, 字段的值为数组, 数组内容是按顺序提取的品牌词
+            任务: 根据用户输入的文本，按照提取出品牌名称以及提取的依据，并且不要去重；
+            只要文本中提到了的品牌都需要提取出来，文本总结里面的品牌也需要。
+            以json格式返回，json中包含brands字段, 字段的值为数组
 
         文本正文:
         {result['content']}