🤖 自动格式化代码 [skip ci]
This commit is contained in:
@@ -11,7 +11,7 @@ from src.common.logger import get_logger
|
||||
import math
|
||||
import re
|
||||
import traceback
|
||||
from typing import Optional, Tuple, Dict, Any
|
||||
from typing import Optional, Tuple
|
||||
from maim_message import UserInfo
|
||||
|
||||
from src.person_info.relationship_manager import get_relationship_manager
|
||||
|
||||
@@ -51,7 +51,6 @@ class ChatBot:
|
||||
async def _process_commands_with_new_system(self, message: MessageRecv):
|
||||
"""使用新插件系统处理命令"""
|
||||
try:
|
||||
|
||||
text = message.processed_plain_text
|
||||
|
||||
# 使用新的组件注册中心查找命令
|
||||
|
||||
@@ -182,7 +182,9 @@ class NormalChat:
|
||||
interested_rate=interest_value * self.willing_amplifier,
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"[{self.stream_name}] 处理兴趣消息{msg_id}时出错: {e}\n{traceback.format_exc()}")
|
||||
logger.error(
|
||||
f"[{self.stream_name}] 处理兴趣消息{msg_id}时出错: {e}\n{traceback.format_exc()}"
|
||||
)
|
||||
finally:
|
||||
self.interest_dict.pop(msg_id, None)
|
||||
|
||||
|
||||
@@ -329,7 +329,9 @@ class NormalChatActionModifier:
|
||||
|
||||
should_switch = recent_reply_count > reply_threshold
|
||||
if should_switch:
|
||||
logger.debug(f"{self.log_prefix} 检测到1分钟内回复数量({recent_reply_count})大于{reply_threshold},满足切换到focus模式条件")
|
||||
logger.debug(
|
||||
f"{self.log_prefix} 检测到1分钟内回复数量({recent_reply_count})大于{reply_threshold},满足切换到focus模式条件"
|
||||
)
|
||||
|
||||
return should_switch
|
||||
|
||||
|
||||
@@ -191,7 +191,7 @@ def _build_readable_messages_internal(
|
||||
nonlocal current_pic_counter
|
||||
|
||||
# 匹配 [picid:xxxxx] 格式
|
||||
pic_pattern = r'\[picid:([^\]]+)\]'
|
||||
pic_pattern = r"\[picid:([^\]]+)\]"
|
||||
|
||||
def replace_pic_id(match):
|
||||
nonlocal current_pic_counter
|
||||
@@ -445,7 +445,12 @@ def _build_readable_messages_internal(
|
||||
formatted_string = "".join(output_lines).strip()
|
||||
|
||||
# 返回格式化后的字符串、消息详情列表、图片映射字典和更新后的计数器
|
||||
return formatted_string, [(t, n, c) for t, n, c, is_action in message_details if not is_action], pic_id_mapping, current_pic_counter
|
||||
return (
|
||||
formatted_string,
|
||||
[(t, n, c) for t, n, c, is_action in message_details if not is_action],
|
||||
pic_id_mapping,
|
||||
current_pic_counter,
|
||||
)
|
||||
|
||||
|
||||
def build_pic_mapping_info(pic_id_mapping: Dict[str, str]) -> str:
|
||||
@@ -461,8 +466,6 @@ def build_pic_mapping_info(pic_id_mapping: Dict[str, str]) -> str:
|
||||
if not pic_id_mapping:
|
||||
return ""
|
||||
|
||||
|
||||
|
||||
mapping_lines = []
|
||||
|
||||
# 按图片编号排序
|
||||
@@ -475,7 +478,7 @@ def build_pic_mapping_info(pic_id_mapping: Dict[str, str]) -> str:
|
||||
image = Images.get_or_none(Images.image_id == pic_id)
|
||||
if image and image.description:
|
||||
description = image.description
|
||||
except Exception as e:
|
||||
except Exception:
|
||||
# 如果查询失败,保持默认描述
|
||||
pass
|
||||
|
||||
@@ -593,12 +596,16 @@ def build_readable_messages(
|
||||
|
||||
# 分别格式化,但使用共享的图片映射
|
||||
formatted_before, _, pic_id_mapping, pic_counter = _build_readable_messages_internal(
|
||||
messages_before_mark, replace_bot_name, merge_messages, timestamp_mode, truncate,
|
||||
pic_id_mapping, pic_counter
|
||||
messages_before_mark,
|
||||
replace_bot_name,
|
||||
merge_messages,
|
||||
timestamp_mode,
|
||||
truncate,
|
||||
pic_id_mapping,
|
||||
pic_counter,
|
||||
)
|
||||
formatted_after, _, pic_id_mapping, _ = _build_readable_messages_internal(
|
||||
messages_after_mark, replace_bot_name, merge_messages, timestamp_mode, False,
|
||||
pic_id_mapping, pic_counter
|
||||
messages_after_mark, replace_bot_name, merge_messages, timestamp_mode, False, pic_id_mapping, pic_counter
|
||||
)
|
||||
|
||||
read_mark_line = "\n--- 以上消息是你已经看过---\n--- 请关注以下未读的新消息---\n"
|
||||
@@ -646,7 +653,7 @@ async def build_anonymous_messages(messages: List[Dict[str, Any]]) -> str:
|
||||
nonlocal pic_counter
|
||||
|
||||
# 匹配 [picid:xxxxx] 格式
|
||||
pic_pattern = r'\[picid:([^\]]+)\]'
|
||||
pic_pattern = r"\[picid:([^\]]+)\]"
|
||||
|
||||
def replace_pic_id(match):
|
||||
nonlocal pic_counter
|
||||
|
||||
@@ -411,7 +411,7 @@ class ImageManager:
|
||||
base64=image_base64,
|
||||
type="image",
|
||||
timestamp=current_timestamp,
|
||||
vlm_processed=False
|
||||
vlm_processed=False,
|
||||
)
|
||||
|
||||
# 启动异步VLM处理
|
||||
@@ -453,11 +453,7 @@ class ImageManager:
|
||||
prompt = """请用中文描述这张图片的内容。如果有文字,请把文字描述概括出来,请留意其主题,直观感受,输出为一段平文本,最多50字"""
|
||||
|
||||
# 获取VLM描述
|
||||
description, _ = await self._llm.generate_response_for_image(
|
||||
prompt,
|
||||
image_base64,
|
||||
image_format
|
||||
)
|
||||
description, _ = await self._llm.generate_response_for_image(prompt, image_base64, image_format)
|
||||
|
||||
if description is None:
|
||||
logger.warning("VLM未能生成图片描述")
|
||||
|
||||
@@ -32,11 +32,7 @@ class PluginManager:
|
||||
|
||||
def _ensure_plugin_directories(self):
|
||||
"""确保所有插件目录存在,如果不存在则创建"""
|
||||
default_directories = [
|
||||
"src/plugins/built_in",
|
||||
"src/plugins/examples",
|
||||
"plugins"
|
||||
]
|
||||
default_directories = ["src/plugins/built_in", "src/plugins/examples", "plugins"]
|
||||
|
||||
for directory in default_directories:
|
||||
if not os.path.exists(directory):
|
||||
|
||||
@@ -267,7 +267,7 @@ class ChangeToFocusChatAction(BaseAction):
|
||||
"聊天上下文中自己的回复条数较多(超过3-4条)",
|
||||
"对话进行得非常热烈活跃",
|
||||
"用户表现出深入交流的意图",
|
||||
"话题需要更专注和深入的讨论"
|
||||
"话题需要更专注和深入的讨论",
|
||||
]
|
||||
|
||||
async def execute(self) -> Tuple[bool, str]:
|
||||
|
||||
Reference in New Issue
Block a user