HFC对应修改（部分）

src/common/toml_utils.py

@@ -1,89 +0,0 @@
-"""
-TOML文件工具函数 - 保留格式和注释
-"""
-
-import os
-import tomlkit
-from typing import Any
-
-
-def save_toml_with_format(data: dict[str, Any], file_path: str) -> None:
-    """
-    保存TOML数据到文件，保留现有格式（如果文件存在）
-
-    Args:
-        data: 要保存的数据字典
-        file_path: 文件路径
-    """
-    # 如果文件不存在，直接创建
-    if not os.path.exists(file_path):
-        with open(file_path, "w", encoding="utf-8") as f:
-            tomlkit.dump(data, f)
-        return
-
-    # 如果文件存在，尝试读取现有文件以保留格式
-    try:
-        with open(file_path, "r", encoding="utf-8") as f:
-            existing_doc = tomlkit.load(f)
-    except Exception:
-        # 如果读取失败，直接覆盖
-        with open(file_path, "w", encoding="utf-8") as f:
-            tomlkit.dump(data, f)
-        return
-
-    # 递归更新，保留现有格式
-    _merge_toml_preserving_format(existing_doc, data)
-
-    # 保存
-    with open(file_path, "w", encoding="utf-8") as f:
-        tomlkit.dump(existing_doc, f)
-
-
-def _merge_toml_preserving_format(target: dict[str, Any], source: dict[str, Any]) -> None:
-    """
-    递归合并source到target，保留target中的格式和注释
-
-    Args:
-        target: 目标文档（保留格式）
-        source: 源数据（新数据）
-    """
-    for key, value in source.items():
-        if key in target:
-            # 如果两个都是字典且都是表格，递归合并
-            if isinstance(value, dict) and isinstance(target[key], dict):
-                if hasattr(target[key], "items"):  # 确实是字典/表格
-                    _merge_toml_preserving_format(target[key], value)
-                else:
-                    target[key] = value
-            else:
-                # 其他情况直接替换
-                target[key] = value
-        else:
-            # 新键直接添加
-            target[key] = value
-
-
-def _update_toml_doc(target: dict[str, Any], source: dict[str, Any]) -> None:
-    """
-    更新TOML文档中的字段，保留现有的格式和注释
-
-    这是一个递归函数，用于在部分更新配置时保留现有的格式和注释。
-
-    Args:
-        target: 目标表格（会被修改）
-        source: 源数据（新数据）
-    """
-    for key, value in source.items():
-        if key in target:
-            # 如果两个都是字典，递归更新
-            if isinstance(value, dict) and isinstance(target[key], dict):
-                if hasattr(target[key], "items"):  # 确实是表格
-                    _update_toml_doc(target[key], value)
-                else:
-                    target[key] = value
-            else:
-                # 直接更新值，保留注释
-                target[key] = value
-        else:
-            # 新键直接添加
-            target[key] = value

src/common/utils/utils_message.py

@@ -1,5 +1,5 @@
 from maim_message import MessageBase, Seg
-from typing import List, Tuple, Optional, Sequence
+from typing import List, Tuple, Optional, Sequence, TYPE_CHECKING
 
 import base64
 import hashlib
@@ -19,6 +19,9 @@ from src.common.data_models.message_component_data_model import (
 )
 from src.config.config import global_config
 
+if TYPE_CHECKING:
+    from src.chat.message_receive.message import SessionMessage
+
 
 class MessageUtils:
     @staticmethod
@@ -135,3 +138,12 @@ class MessageUtils:
         else:
             components = [platform, user_id, "private"]
         return hashlib.md5("_".join(components).encode()).hexdigest()
+
+    @staticmethod
+    def store_message_to_db(message: "SessionMessage"):
+        """存储消息到数据库"""
+        from src.common.database.database import get_db_session
+
+        with get_db_session() as session:
+            db_message = message.to_db_instance()
+            session.add(db_message)