From 146ca4867168570d7273aac7106e82ba2da84e58 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=A2=A8=E6=A2=93=E6=9F=92?= <1787882683@qq.com>
Date: Wed, 28 May 2025 23:29:11 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E6=9B=B4=E6=96=B0MongoToSQLiteMigrator?=
 =?UTF-8?q?=E7=9A=84=E5=94=AF=E4=B8=80=E5=AD=97=E6=AE=B5=E5=92=8C=E9=AA=8C?=
 =?UTF-8?q?=E8=AF=81=E8=AE=BE=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 scripts/mongodb_to_sqlite.py             | 4 ++--
 src/experimental/only_message_process.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/scripts/mongodb_to_sqlite.py b/scripts/mongodb_to_sqlite.py
index 0f292fef6..e3522c6aa 100644
--- a/scripts/mongodb_to_sqlite.py
+++ b/scripts/mongodb_to_sqlite.py
@@ -198,8 +198,8 @@ class MongoToSQLiteMigrator:
                     "status": "status",
                     "timestamp": "timestamp",
                 },
-                enable_validation=False,  # 禁用数据验证
-                unique_fields=["user_id", "timestamp"],  # 组合唯一性
+                enable_validation=True,  # 禁用数据验证"
+                unique_fields=["user_id", "prompt_tokens","completion_tokens","total_tokens","cost"],  # 组合唯一性
             ),
             # 消息迁移配置
             MigrationConfig(
diff --git a/src/experimental/only_message_process.py b/src/experimental/only_message_process.py
index 62f73c700..8fb1e3bf2 100644
--- a/src/experimental/only_message_process.py
+++ b/src/experimental/only_message_process.py
@@ -16,7 +16,7 @@ class MessageProcessor:
     @staticmethod
     def _check_ban_words(text: str, chat, userinfo) -> bool:
         """检查消息中是否包含过滤词"""
-        for word in global_config.chat.ban_words:
+        for word in global_config.message_receive.ban_words:
             if word in text:
                 logger.info(
                     f"[{chat.group_info.group_name if chat.group_info else '私聊'}]{userinfo.user_nickname}:{text}"