BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

类型标注

pull/937/head
Bakadax 2025-05-07 15:54:03 +08:00
parent 1905e02587
commit b1893e9514
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/plugins/PFC/pfc_emotion_updater.py
View File

@ -2,6 +2,7 @@
from typing import List, Dict, Any, Optional
from MaiMBot.src.plugins.PFC.chat_observer import ChatObserver
from src.common.logger_manager import get_logger
from src.plugins.models.utils_model import LLMRequest
from src.plugins.moods.moods import MoodManager # MoodManager 本身是单例
@ -48,14 +49,14 @@ class PfcEmotionUpdater:
self,
conversation_info: ConversationInfo,
observation_info: ObservationInfo,
chat_observer_for_history, # ChatObserver 实例
chat_observer_for_history: ChatObserver, # ChatObserver 实例
event_description: str
) -> None:
if not self.llm:
logger.error(f"[私聊][{self.private_name}] LLM未初始化，无法进行情绪更新。")
# 即使LLM失败，也应该更新conversation_info中的情绪文本为MoodManager的当前状态
if conversation_info and self.mood_mng:
conversation_info.current_emotion_text = self.mood_mng.get_prompt()
conversation_info.current_emotion_text = self.mood_mng.get_prompt()
return
if not self.mood_mng or not conversation_info or not observation_info:

5
src/plugins/PFC/pfc_relationship.py
View File

@ -1,4 +1,5 @@
from typing import List, Dict, Any
from MaiMBot.src.plugins.PFC.chat_observer import ChatObserver
from src.common.logger_manager import get_logger
from src.plugins.models.utils_model import LLMRequest
from src.plugins.person_info.person_info import person_info_manager
@ -65,7 +66,7 @@ class PfcRelationshipUpdater:
self,
conversation_info: ConversationInfo,
observation_info: ObservationInfo,
chat_observer_for_history # ChatObserver 实例
chat_observer_for_history: ChatObserver # ChatObserver 实例
) -> None:
if not self.llm:
logger.error(f"[私聊][{self.private_name}] LLM未初始化，无法进行增量关系更新。")
@ -141,7 +142,7 @@ class PfcRelationshipUpdater:
self,
conversation_info: ConversationInfo,
observation_info: ObservationInfo,
chat_observer_for_history
chat_observer_for_history: ChatObserver
) -> None:
if not self.llm:
logger.error(f"[私聊][{self.private_name}] LLM未初始化，无法进行最终关系更新。")