From 73104e9c3ce5b58ae6c6d5221a079c389efb5e61 Mon Sep 17 00:00:00 2001
From: HexatomicRing <54496918+HexatomicRing@users.noreply.github.com>
Date: Wed, 12 Mar 2025 19:07:54 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=A6=82=E7=8E=87=E8=AE=A1?=
 =?UTF-8?q?=E7=AE=97=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/plugins/chat/prompt_builder.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/plugins/chat/prompt_builder.py b/src/plugins/chat/prompt_builder.py
index 7930dbf5..accef522 100644
--- a/src/plugins/chat/prompt_builder.py
+++ b/src/plugins/chat/prompt_builder.py
@@ -16,12 +16,11 @@ def choose_personality():
     personality = global_config.PROMPT_PERSONALITY
     probabilities = global_config.PERSONALITIES
     personality_choice = random.random()
-    choice_base = 0.0
     i_person = 0
     for i_person, probability in enumerate(probabilities):
-        choice_base += probability
-        if personality_choice < choice_base:
+        if personality_choice < personality_choice:
             break
+        personality_choice -= probability
     if i_person < len(personality):
         logger.info(f"正在用人格{i_person + 1}回答")
     else: