BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

优化概率计算方法

pull/266/head
HexatomicRing 2025-03-12 19:07:54 +08:00
parent cf9f67d3f2
commit 73104e9c3c
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/plugins/chat/prompt_builder.py
View File

@ -16,12 +16,11 @@ def choose_personality():
personality = global_config.PROMPT_PERSONALITY personality = global_config.PROMPT_PERSONALITY
probabilities = global_config.PERSONALITIES probabilities = global_config.PERSONALITIES
personality_choice = random.random() personality_choice = random.random()
choice_base = 0.0
i_person = 0 i_person = 0
for i_person, probability in enumerate(probabilities): for i_person, probability in enumerate(probabilities):
choice_base += probability if personality_choice < personality_choice:
if personality_choice < choice_base:
break break
personality_choice -= probability
if i_person < len(personality): if i_person < len(personality):
logger.info(f"正在用人格{i_person + 1}回答") logger.info(f"正在用人格{i_person + 1}回答")
else: else: