修复typing问题，保证类型正确

src/chat/frequency_control/focus_value_control.py

@@ -4,44 +4,43 @@ from src.chat.frequency_control.utils import parse_stream_config_to_chat_id
 
 
 class FocusValueControl:
-    def __init__(self,chat_id:str):
+    def __init__(self, chat_id: str):
         self.chat_id = chat_id
-        self.focus_value_adjust = 1
-        
-        
+        self.focus_value_adjust: float = 1
+
     def get_current_focus_value(self) -> float:
         return get_current_focus_value(self.chat_id) * self.focus_value_adjust
-        
+
 
 class FocusValueControlManager:
     def __init__(self):
-        self.focus_value_controls = {}
-        
-    def get_focus_value_control(self,chat_id:str) -> FocusValueControl:
+        self.focus_value_controls: dict[str, FocusValueControl] = {}
+
+    def get_focus_value_control(self, chat_id: str) -> FocusValueControl:
         if chat_id not in self.focus_value_controls:
             self.focus_value_controls[chat_id] = FocusValueControl(chat_id)
         return self.focus_value_controls[chat_id]
 
 
-
 def get_current_focus_value(chat_id: Optional[str] = None) -> float:
     """
     根据当前时间和聊天流获取对应的 focus_value
     """
     if not global_config.chat.focus_value_adjust:
         return global_config.chat.focus_value
-    
+
     if chat_id:
         stream_focus_value = get_stream_specific_focus_value(chat_id)
         if stream_focus_value is not None:
             return stream_focus_value
-        
+
     global_focus_value = get_global_focus_value()
     if global_focus_value is not None:
         return global_focus_value
-    
+
     return global_config.chat.focus_value
 
+
 def get_stream_specific_focus_value(chat_id: str) -> Optional[float]:
     """
     获取特定聊天流在当前时间的专注度
@@ -140,4 +139,5 @@ def get_global_focus_value() -> Optional[float]:
 
     return None
 
-focus_value_control = FocusValueControlManager()
+
+focus_value_control = FocusValueControlManager()

src/chat/frequency_control/talk_frequency_control.py

@@ -2,20 +2,21 @@ from typing import Optional
 from src.config.config import global_config
 from src.chat.frequency_control.utils import parse_stream_config_to_chat_id
 
+
 class TalkFrequencyControl:
-    def __init__(self,chat_id:str):
+    def __init__(self, chat_id: str):
         self.chat_id = chat_id
-        self.talk_frequency_adjust = 1
-        
+        self.talk_frequency_adjust: float = 1
+
     def get_current_talk_frequency(self) -> float:
         return get_current_talk_frequency(self.chat_id) * self.talk_frequency_adjust
-        
+
 
 class TalkFrequencyControlManager:
     def __init__(self):
         self.talk_frequency_controls = {}
-        
-    def get_talk_frequency_control(self,chat_id:str) -> TalkFrequencyControl:
+
+    def get_talk_frequency_control(self, chat_id: str) -> TalkFrequencyControl:
         if chat_id not in self.talk_frequency_controls:
             self.talk_frequency_controls[chat_id] = TalkFrequencyControl(chat_id)
         return self.talk_frequency_controls[chat_id]
@@ -44,6 +45,7 @@ def get_current_talk_frequency(chat_id: Optional[str] = None) -> float:
     global_frequency = get_global_frequency()
     return global_config.chat.talk_frequency if global_frequency is None else global_frequency
 
+
 def get_time_based_frequency(time_freq_list: list[str]) -> Optional[float]:
     """
     根据时间配置列表获取当前时段的频率
@@ -124,6 +126,7 @@ def get_stream_specific_frequency(chat_stream_id: str):
 
     return None
 
+
 def get_global_frequency() -> Optional[float]:
     """
     获取全局默认频率配置
@@ -141,4 +144,5 @@ def get_global_frequency() -> Optional[float]:
 
     return None
 
-talk_frequency_control = TalkFrequencyControlManager()
+
+talk_frequency_control = TalkFrequencyControlManager()