ruff

Author: BBleae

src/chat/actions/base_action.py

@@ -8,19 +8,22 @@
 _ACTION_REGISTRY: Dict[str, Type["BaseAction"]] = {}
 _DEFAULT_ACTIONS: Dict[str, str] = {}
 
+
 # 动作激活类型枚举
 class ActionActivationType:
     ALWAYS = "always"  # 默认参与到planner
-    LLM_JUDGE = "llm_judge"  # LLM判定是否启动该action到planner  
+    LLM_JUDGE = "llm_judge"  # LLM判定是否启动该action到planner
     RANDOM = "random"  # 随机启用action到planner
     KEYWORD = "keyword"  # 关键词触发启用action到planner
 
+
 # 聊天模式枚举
 class ChatMode:
     FOCUS = "focus"  # Focus聊天模式
     NORMAL = "normal"  # Normal聊天模式
     ALL = "all"  # 所有聊天模式
 
+
 def register_action(cls):
     """
     动作注册装饰器
@@ -81,13 +84,13 @@ def __init__(self, action_data: dict, reasoning: str, cycle_timers: dict, thinki
         self.action_description: str = "基础动作"
         self.action_parameters: dict = {}
         self.action_require: list[str] = []
-        
+
         # 动作激活类型设置
         # Focus模式下的激活类型，默认为always
         self.focus_activation_type: str = ActionActivationType.ALWAYS
-        # Normal模式下的激活类型，默认为always  
+        # Normal模式下的激活类型，默认为always
         self.normal_activation_type: str = ActionActivationType.ALWAYS
-        
+
         # 随机激活的概率(0.0-1.0)，用于RANDOM激活类型
         self.random_activation_probability: float = 0.3
         # LLM判定的提示词，用于LLM_JUDGE激活类型

src/chat/actions/default_actions/__init__.py

@@ -4,4 +4,4 @@
 from . import exit_focus_chat_action  # noqa
 from . import emoji_action  # noqa
 
-# 在此处添加更多动作模块导入 
+# 在此处添加更多动作模块导入

src/chat/actions/default_actions/emoji_action.py

@@ -22,29 +22,26 @@ class EmojiAction(BaseAction):
     action_parameters: dict[str:str] = {
         "description": "文字描述你想要发送的表情包内容",
     }
-    action_require: list[str] = [
-        "表达情绪时可以选择使用",
-        "重点：不要连续发，如果你已经发过[表情包]，就不要选择此动作"]
+    action_require: list[str] = ["表达情绪时可以选择使用", "重点：不要连续发，如果你已经发过[表情包]，就不要选择此动作"]
 
     associated_types: list[str] = ["emoji"]
 
     enable_plugin = True
-    
+
     focus_activation_type = ActionActivationType.LLM_JUDGE
     normal_activation_type = ActionActivationType.RANDOM
-    
+
     random_activation_probability = global_config.normal_chat.emoji_chance
-    
+
     parallel_action = True
-    
-    
+
     llm_judge_prompt = """
     判定是否需要使用表情动作的条件：
     1. 用户明确要求使用表情包
     2. 这是一个适合表达强烈情绪的场合
     3. 不要发送太多表情包，如果你已经发送过多个表情包
     """
-    
+
     # 模式启用设置 - 表情动作只在Focus模式下使用
     mode_enable = ChatMode.ALL
 
@@ -147,4 +144,4 @@ async def _handle_reply(
                 elif type == "emoji":
                     reply_text += data
 
-        return success, reply_text
+        return success, reply_text

src/chat/actions/default_actions/exit_focus_chat_action.py

@@ -27,7 +27,7 @@ class ExitFocusChatAction(BaseAction):
     ]
     # 退出专注聊天是系统核心功能，不是插件，但默认不启用（需要特定条件触发）
     enable_plugin = False
-    
+
     # 模式启用设置 - 退出专注聊天动作只在Focus模式下使用
     mode_enable = ChatMode.FOCUS
 

src/chat/actions/default_actions/no_reply_action.py

@@ -29,10 +29,10 @@ class NoReplyAction(BaseAction):
         "想要休息一下",
     ]
     enable_plugin = True
-    
+
     # 激活类型设置
     focus_activation_type = ActionActivationType.ALWAYS
-    
+
     # 模式启用设置 - no_reply动作只在Focus模式下使用
     mode_enable = ChatMode.FOCUS
 

src/chat/actions/default_actions/reply_action.py

@@ -31,16 +31,16 @@ class ReplyAction(BaseAction):
     action_require: list[str] = [
         "你想要闲聊或者随便附和",
         "有人提到你",
-        "如果你刚刚进行了回复，不要对同一个话题重复回应"
+        "如果你刚刚进行了回复，不要对同一个话题重复回应",
     ]
 
     associated_types: list[str] = ["text"]
 
     enable_plugin = True
-    
+
     # 激活类型设置
     focus_activation_type = ActionActivationType.ALWAYS
-    
+
     # 模式启用设置 - 回复动作只在Focus模式下使用
     mode_enable = ChatMode.FOCUS
 
@@ -89,12 +89,12 @@ async def handle_action(self) -> Tuple[bool, str]:
             cycle_timers=self.cycle_timers,
             thinking_id=self.thinking_id,
         )
-        
+
         await self.store_action_info(
             action_build_into_prompt=False,
             action_prompt_display=f"{reply_text}",
         )
-        
+
         return success, reply_text
 
     async def _handle_reply(
@@ -115,30 +115,29 @@ async def _handle_reply(
         chatting_observation: ChattingObservation = next(
             obs for obs in self.observations if isinstance(obs, ChattingObservation)
         )
-        
+
         reply_to = reply_data.get("reply_to", "none")
-            
+
         # sender = ""
         target = ""
         if ":" in reply_to or "：" in reply_to:
             # 使用正则表达式匹配中文或英文冒号
-            parts = re.split(pattern=r'[:：]', string=reply_to, maxsplit=1)
+            parts = re.split(pattern=r"[:：]", string=reply_to, maxsplit=1)
             if len(parts) == 2:
                 # sender = parts[0].strip()
                 target = parts[1].strip()
                 anchor_message = chatting_observation.search_message_by_text(target)
         else:
             anchor_message = None
-        
-        if anchor_message:  
+
+        if anchor_message:
             anchor_message.update_chat_stream(self.chat_stream)
         else:
             logger.info(f"{self.log_prefix} 未找到锚点消息，创建占位符")
             anchor_message = await create_empty_anchor_message(
                 self.chat_stream.platform, self.chat_stream.group_info, self.chat_stream
             )
 
-
         success, reply_set = await self.replyer.deal_reply(
             cycle_timers=cycle_timers,
             action_data=reply_data,
@@ -158,8 +157,9 @@ async def _handle_reply(
 
         return success, reply_text
 
-
-    async def store_action_info(self, action_build_into_prompt: bool = False, action_prompt_display: str = "", action_done: bool = True) -> None:
+    async def store_action_info(
+        self, action_build_into_prompt: bool = False, action_prompt_display: str = "", action_done: bool = True
+    ) -> None:
         """存储action执行信息到数据库
 
         Args:
@@ -188,9 +188,9 @@ async def store_action_info(self, action_build_into_prompt: bool = False, action
                 chat_info_platform=chat_stream.platform,
                 user_id=chat_stream.user_info.user_id if chat_stream.user_info else "",
                 user_nickname=chat_stream.user_info.user_nickname if chat_stream.user_info else "",
-                user_cardname=chat_stream.user_info.user_cardname if chat_stream.user_info else ""
+                user_cardname=chat_stream.user_info.user_cardname if chat_stream.user_info else "",
             )
             logger.debug(f"{self.log_prefix} 已存储action信息: {action_prompt_display}")
         except Exception as e:
             logger.error(f"{self.log_prefix} 存储action信息时出错: {e}")
-            traceback.print_exc()
+            traceback.print_exc()

src/chat/actions/plugin_action.py

@@ -1,10 +1,6 @@
-import traceback
-from typing import Tuple, Dict, List, Any, Optional, Union, Type
+from typing import Tuple, Dict, Any, Optional
 from src.chat.actions.base_action import BaseAction, register_action, ActionActivationType, ChatMode  # noqa F401
-from src.chat.heart_flow.observation.chatting_observation import ChattingObservation
-from src.chat.focus_chat.hfc_utils import create_empty_anchor_message
 from src.common.logger_manager import get_logger
-from src.config.config import global_config
 import os
 import inspect
 import toml  # 导入 toml 库
@@ -20,10 +16,10 @@
 from src.chat.actions.plugin_api.hearflow_api import HearflowAPI
 
 # 以下为类型注解需要
-from src.chat.message_receive.chat_stream import ChatStream # noqa
-from src.chat.focus_chat.expressors.default_expressor import DefaultExpressor # noqa
-from src.chat.focus_chat.replyer.default_replyer import DefaultReplyer # noqa
-from src.chat.focus_chat.info.obs_info import ObsInfo # noqa
+from src.chat.message_receive.chat_stream import ChatStream  # noqa
+from src.chat.focus_chat.expressors.default_expressor import DefaultExpressor  # noqa
+from src.chat.focus_chat.replyer.default_replyer import DefaultReplyer  # noqa
+from src.chat.focus_chat.info.obs_info import ObsInfo  # noqa
 
 logger = get_logger("plugin_action")
 
@@ -35,15 +31,15 @@ class PluginAction(BaseAction, MessageAPI, LLMAPI, DatabaseAPI, ConfigAPI, Utils
     """
 
     action_config_file_name: Optional[str] = None  # 插件可以覆盖此属性来指定配置文件名
-    
+
     # 默认激活类型设置，插件可以覆盖
     focus_activation_type = ActionActivationType.ALWAYS
     normal_activation_type = ActionActivationType.ALWAYS
     random_activation_probability: float = 0.3
     llm_judge_prompt: str = ""
     activation_keywords: list[str] = []
     keyword_case_sensitive: bool = False
-    
+
     # 默认模式启用设置 - 插件动作默认在所有模式下可用，插件可以覆盖
     mode_enable = ChatMode.ALL
 

src/chat/actions/plugin_api/__init__.py

@@ -7,11 +7,11 @@
 from src.chat.actions.plugin_api.hearflow_api import HearflowAPI
 
 __all__ = [
-    'MessageAPI',
-    'LLMAPI',
-    'DatabaseAPI',
-    'ConfigAPI',
-    'UtilsAPI',
-    'StreamAPI',
-    'HearflowAPI',
-] 
+    "MessageAPI",
+    "LLMAPI",
+    "DatabaseAPI",
+    "ConfigAPI",
+    "UtilsAPI",
+    "StreamAPI",
+    "HearflowAPI",
+]

src/chat/actions/plugin_api/config_api.py

@@ -5,49 +5,50 @@
 
 logger = get_logger("config_api")
 
+
 class ConfigAPI:
     """配置API模块
-    
+
     提供了配置读取和用户信息获取等功能
     """
-    
+
     def get_global_config(self, key: str, default: Any = None) -> Any:
         """
         安全地从全局配置中获取一个值。
         插件应使用此方法读取全局配置，以保证只读和隔离性。
-        
+
         Args:
             key: 配置键名
             default: 如果配置不存在时返回的默认值
-            
+
         Returns:
             Any: 配置值或默认值
         """
         return global_config.get(key, default)
-        
+
     async def get_user_id_by_person_name(self, person_name: str) -> tuple[str, str]:
         """根据用户名获取用户ID
-        
+
         Args:
             person_name: 用户名
-            
+
         Returns:
             tuple[str, str]: (平台, 用户ID)
         """
         person_id = person_info_manager.get_person_id_by_person_name(person_name)
         user_id = await person_info_manager.get_value(person_id, "user_id")
         platform = await person_info_manager.get_value(person_id, "platform")
         return platform, user_id
-        
+
     async def get_person_info(self, person_id: str, key: str, default: Any = None) -> Any:
         """获取用户信息
-        
+
         Args:
             person_id: 用户ID
             key: 信息键名
             default: 默认值
-            
+
         Returns:
             Any: 用户信息值或默认值
         """
-        return await person_info_manager.get_value(person_id, key, default) 
+        return await person_info_manager.get_value(person_id, key, default)