修改旧的DatabaseMessages引用

This commit is contained in:
DrSmoothl
2026-03-12 20:12:05 +08:00
parent 16f115bfac
commit df39fa7584

View File

@@ -10,7 +10,6 @@ from datetime import datetime
from typing import Optional, Tuple, List, TYPE_CHECKING from typing import Optional, Tuple, List, TYPE_CHECKING
from src.common.logger import get_logger from src.common.logger import get_logger
from src.common.data_models.database_data_model import DatabaseMessages
from src.config.config import global_config, model_config from src.config.config import global_config, model_config
from src.chat.message_receive.message import SessionMessage from src.chat.message_receive.message import SessionMessage
from src.chat.message_receive.chat_manager import chat_manager as _chat_manager from src.chat.message_receive.chat_manager import chat_manager as _chat_manager
@@ -785,7 +784,7 @@ def record_replyer_action_temp(chat_id: str, reason: str, think_level: int) -> N
logger.warning(f"记录replyer动作选择失败: {e}") logger.warning(f"记录replyer动作选择失败: {e}")
def assign_message_ids(messages: List[DatabaseMessages]) -> List[Tuple[str, DatabaseMessages]]: def assign_message_ids(messages: List[SessionMessage]) -> List[Tuple[str, SessionMessage]]:
""" """
为消息列表中的每个消息分配唯一的简短随机ID 为消息列表中的每个消息分配唯一的简短随机ID
@@ -793,9 +792,9 @@ def assign_message_ids(messages: List[DatabaseMessages]) -> List[Tuple[str, Data
messages: 消息列表 messages: 消息列表
Returns: Returns:
List[DatabaseMessages]: 分配了唯一ID的消息列表(写入message_id属性) List[SessionMessage]: 分配了唯一ID的消息列表
""" """
result: List[Tuple[str, DatabaseMessages]] = [] # 复制原始消息列表 result: List[Tuple[str, SessionMessage]] = [] # 复制原始消息列表
used_ids = set() used_ids = set()
len_i = len(messages) len_i = len(messages)
if len_i > 100: if len_i > 100: