修复interest_value为None的情况,确保在计算兴趣值时不会出现错误。将interest_value的默认值设置为0.0,以提高代码的健壮性。
This commit is contained in:
@@ -291,6 +291,9 @@ class HeartFChatting:
|
|||||||
total_interest = 0.0
|
total_interest = 0.0
|
||||||
for msg_dict in new_message:
|
for msg_dict in new_message:
|
||||||
interest_value = msg_dict.get("interest_value", 0.0)
|
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", ""):
|
if msg_dict.get("processed_plain_text", ""):
|
||||||
total_interest += interest_value
|
total_interest += interest_value
|
||||||
|
|
||||||
@@ -308,6 +311,9 @@ class HeartFChatting:
|
|||||||
for msg_dict in new_message:
|
for msg_dict in new_message:
|
||||||
text = msg_dict.get("processed_plain_text", "")
|
text = msg_dict.get("processed_plain_text", "")
|
||||||
interest_value = msg_dict.get("interest_value", 0.0)
|
interest_value = msg_dict.get("interest_value", 0.0)
|
||||||
|
# 确保 interest_value 不为 None
|
||||||
|
if interest_value is None:
|
||||||
|
interest_value = 0.0
|
||||||
if text:
|
if text:
|
||||||
accumulated_interest += interest_value
|
accumulated_interest += interest_value
|
||||||
|
|
||||||
|
|||||||
@@ -117,7 +117,7 @@ class MessageRecv(Message):
|
|||||||
|
|
||||||
self.priority_mode = "interest"
|
self.priority_mode = "interest"
|
||||||
self.priority_info = None
|
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"):
|
def update_chat_stream(self, chat_stream: "ChatStream"):
|
||||||
self.chat_stream = chat_stream
|
self.chat_stream = chat_stream
|
||||||
|
|||||||
Reference in New Issue
Block a user