修改旧的DatabaseMessages引用
This commit is contained in:
@@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user