将20命名为常量

This commit is contained in:
Bakadax
2025-04-29 19:58:16 +08:00
parent 429656ffb7
commit 815f616dfd

View File

@@ -22,7 +22,7 @@ logger = get_logger("emoji")
BASE_DIR = os.path.join("data")
EMOJI_DIR = os.path.join(BASE_DIR, "emoji") # 表情包存储目录
EMOJI_REGISTED_DIR = os.path.join(BASE_DIR, "emoji_registed") # 已注册的表情包注册目录
MAX_EMOJI_FOR_PROMPT = 20 # 最大表情包描述于图片替换的 prompt 中
"""
还没经过测试,有些地方数据库和内存数据同步可能不完全
@@ -634,7 +634,7 @@ class EmojiManager:
# 使用概率分布选择最多20个表情包
selected_emojis = random.choices(
emoji_objects, weights=normalized_probabilities, k=min(20, len(emoji_objects))
emoji_objects, weights=normalized_probabilities, k=min(MAX_EMOJI_FOR_PROMPT, len(emoji_objects))
)
# 将表情包信息转换为可读的字符串