修复interest_value为None的情况,确保在计算兴趣值时不会出现错误。将interest_value的默认值设置为0.0,以提高代码的健壮性。
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user