修复interest_value为None的情况,确保在计算兴趣值时不会出现错误。将interest_value的默认值设置为0.0,以提高代码的健壮性。

This commit is contained in:
minecraft1024a
2025-08-13 13:44:12 +08:00
parent 7ee9b8f39f
commit 6030b9e7c2
2 changed files with 7 additions and 1 deletions

View File

@@ -291,6 +291,9 @@ class HeartFChatting:
total_interest = 0.0
for msg_dict in new_message:
interest_value = msg_dict.get("interest_value", 0.0)
# 确保 interest_value 不为 None
if interest_value is None:
interest_value = 0.0
if msg_dict.get("processed_plain_text", ""):
total_interest += interest_value
@@ -308,6 +311,9 @@ class HeartFChatting:
for msg_dict in new_message:
text = msg_dict.get("processed_plain_text", "")
interest_value = msg_dict.get("interest_value", 0.0)
# 确保 interest_value 不为 None
if interest_value is None:
interest_value = 0.0
if text:
accumulated_interest += interest_value

View File

@@ -117,7 +117,7 @@ class MessageRecv(Message):
self.priority_mode = "interest"
self.priority_info = None
self.interest_value: float = None # type: ignore
self.interest_value: float = 0.0
def update_chat_stream(self, chat_stream: "ChatStream"):
self.chat_stream = chat_stream