typing change, use enum instead of string, fix typo

This commit is contained in:
UnCLASPrommer
2025-07-14 23:40:09 +08:00
parent 8a32ae8ab9
commit eae399fb95
7 changed files with 28 additions and 31 deletions

View File

@@ -2,7 +2,7 @@ import traceback
import os
import re
from typing import Dict, Any
from typing import Dict, Any, Optional
from maim_message import UserInfo
from src.common.logger import get_logger
@@ -210,7 +210,7 @@ class ChatBot:
# 确认从接口发来的message是否有自定义的prompt模板信息
if message.message_info.template_info and not message.message_info.template_info.template_default:
template_group_name = message.message_info.template_info.template_name
template_group_name: Optional[str] = message.message_info.template_info.template_name # type: ignore
template_items = message.message_info.template_info.template_items
async with global_prompt_manager.async_message_scope(template_group_name):
if isinstance(template_items, dict):

View File

@@ -479,7 +479,7 @@ def message_from_db_dict(db_dict: dict) -> MessageRecv:
msg = MessageRecv(recv_dict)
# 从数据库字典中填充其他可选字段
msg.interest_value = db_dict.get("interest_value")
msg.interest_value = db_dict.get("interest_value", 0.0)
msg.is_mentioned = db_dict.get("is_mentioned")
msg.priority_mode = db_dict.get("priority_mode", "interest")
msg.priority_info = db_dict.get("priority_info")