From d2ad6ea1d8f84a73ef43133e85aefe9c4c69fc1a Mon Sep 17 00:00:00 2001 From: UnCLAS-Prommer Date: Sat, 12 Jul 2025 10:18:16 +0800 Subject: [PATCH] fix typo --- .../{exprssion_learner.py => expression_learner.py} | 0 src/chat/express/expression_selector.py | 4 +++- src/chat/heart_flow/heartflow_message_processor.py | 7 +++++-- src/chat/message_receive/message.py | 2 +- src/main.py | 2 +- 5 files changed, 10 insertions(+), 5 deletions(-) rename src/chat/express/{exprssion_learner.py => expression_learner.py} (100%) diff --git a/src/chat/express/exprssion_learner.py b/src/chat/express/expression_learner.py similarity index 100% rename from src/chat/express/exprssion_learner.py rename to src/chat/express/expression_learner.py diff --git a/src/chat/express/expression_selector.py b/src/chat/express/expression_selector.py index 0b1eaef7a..03456e27e 100644 --- a/src/chat/express/expression_selector.py +++ b/src/chat/express/expression_selector.py @@ -10,7 +10,7 @@ from src.llm_models.utils_model import LLMRequest from src.config.config import global_config from src.common.logger import get_logger from src.chat.utils.prompt_builder import Prompt, global_prompt_manager -from .exprssion_learner import get_expression_learner +from .expression_learner import get_expression_learner logger = get_logger("expression_selector") @@ -84,6 +84,7 @@ class ExpressionSelector: def get_random_expressions( self, chat_id: str, total_num: int, style_percentage: float, grammar_percentage: float ) -> Tuple[List[Dict[str, str]], List[Dict[str, str]]]: + # sourcery skip: extract-duplicate-method, move-assign ( learnt_style_expressions, learnt_grammar_expressions, @@ -174,6 +175,7 @@ class ExpressionSelector: min_num: int = 5, target_message: Optional[str] = None, ) -> List[Dict[str, str]]: + # sourcery skip: inline-variable, list-comprehension """使用LLM选择适合的表达方式""" # 1. 获取35个随机表达方式(现在按权重抽取) diff --git a/src/chat/heart_flow/heartflow_message_processor.py b/src/chat/heart_flow/heartflow_message_processor.py index 4ab29b38e..dd267b079 100644 --- a/src/chat/heart_flow/heartflow_message_processor.py +++ b/src/chat/heart_flow/heartflow_message_processor.py @@ -3,7 +3,7 @@ import re import math import traceback -from typing import Tuple +from typing import Tuple, TYPE_CHECKING from src.config.config import global_config from src.chat.memory_system.Hippocampus import hippocampus_manager @@ -16,6 +16,9 @@ from src.common.logger import get_logger from src.person_info.relationship_manager import get_relationship_manager from src.mood.mood_manager import mood_manager +if TYPE_CHECKING: + from src.chat.heart_flow.sub_heartflow import SubHeartflow + logger = get_logger("chat") @@ -104,7 +107,7 @@ class HeartFCMessageReceiver: await self.storage.store_message(message, chat) - subheartflow = await heartflow.get_or_create_subheartflow(chat.stream_id) + subheartflow: SubHeartflow = await heartflow.get_or_create_subheartflow(chat.stream_id) # type: ignore subheartflow.add_message_to_normal_chat_cache(message, interested_rate, is_mentioned) diff --git a/src/chat/message_receive/message.py b/src/chat/message_receive/message.py index 44a1da26f..f444c768f 100644 --- a/src/chat/message_receive/message.py +++ b/src/chat/message_receive/message.py @@ -112,7 +112,7 @@ class MessageRecv(Message): self.is_mentioned = None self.priority_mode = "interest" self.priority_info = None - self.interest_value = None + self.interest_value: float = None # type: ignore def update_chat_stream(self, chat_stream: "ChatStream"): self.chat_stream = chat_stream diff --git a/src/main.py b/src/main.py index 64129814e..d481c7d03 100644 --- a/src/main.py +++ b/src/main.py @@ -2,7 +2,7 @@ import asyncio import time from maim_message import MessageServer -from src.chat.express.exprssion_learner import get_expression_learner +from src.chat.express.expression_learner import get_expression_learner from src.common.remote import TelemetryHeartBeatTask from src.manager.async_task_manager import async_task_manager from src.chat.utils.statistic import OnlineTimeRecordTask, StatisticOutputTask