修复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