fix: 增加设置项
This commit is contained in:
@@ -30,6 +30,7 @@ class BotConfig:
|
|||||||
forget_memory_interval: int = 300 # 记忆遗忘间隔(秒)
|
forget_memory_interval: int = 300 # 记忆遗忘间隔(秒)
|
||||||
EMOJI_CHECK_INTERVAL: int = 120 # 表情包检查间隔(分钟)
|
EMOJI_CHECK_INTERVAL: int = 120 # 表情包检查间隔(分钟)
|
||||||
EMOJI_REGISTER_INTERVAL: int = 10 # 表情包注册间隔(分钟)
|
EMOJI_REGISTER_INTERVAL: int = 10 # 表情包注册间隔(分钟)
|
||||||
|
EMOJI_CHECK_PROMPT: str = "不要包含违反公序良俗的内容" # 表情包过滤要求
|
||||||
|
|
||||||
ban_words = set()
|
ban_words = set()
|
||||||
|
|
||||||
@@ -94,6 +95,7 @@ class BotConfig:
|
|||||||
emoji_config = toml_dict["emoji"]
|
emoji_config = toml_dict["emoji"]
|
||||||
config.EMOJI_CHECK_INTERVAL = emoji_config.get("check_interval", config.EMOJI_CHECK_INTERVAL)
|
config.EMOJI_CHECK_INTERVAL = emoji_config.get("check_interval", config.EMOJI_CHECK_INTERVAL)
|
||||||
config.EMOJI_REGISTER_INTERVAL = emoji_config.get("register_interval", config.EMOJI_REGISTER_INTERVAL)
|
config.EMOJI_REGISTER_INTERVAL = emoji_config.get("register_interval", config.EMOJI_REGISTER_INTERVAL)
|
||||||
|
config.EMOJI_CHECK_PROMPT = emoji_config.get('check_prompt',config.EMOJI_CHECK_PROMPT)
|
||||||
|
|
||||||
if "cq_code" in toml_dict:
|
if "cq_code" in toml_dict:
|
||||||
cq_code_config = toml_dict["cq_code"]
|
cq_code_config = toml_dict["cq_code"]
|
||||||
|
|||||||
@@ -206,7 +206,7 @@ class EmojiManager:
|
|||||||
|
|
||||||
async def _check_emoji(self, image_base64: str) -> str:
|
async def _check_emoji(self, image_base64: str) -> str:
|
||||||
try:
|
try:
|
||||||
prompt = '这是一个表情包,请回答这个表情包是否满足\"动漫风格,画风可爱\"的要求,是则回答是,否则回答否,不要出现任何其他内容'
|
prompt = f'这是一个表情包,请回答这个表情包是否满足\"{global_config.EMOJI_CHECK_PROMPT}\"的要求,是则回答是,否则回答否,不要出现任何其他内容'
|
||||||
|
|
||||||
content, _ = await self.llm.generate_response_for_image(prompt, image_base64)
|
content, _ = await self.llm.generate_response_for_image(prompt, image_base64)
|
||||||
logger.debug(f"输出描述: {content}")
|
logger.debug(f"输出描述: {content}")
|
||||||
|
|||||||
Reference in New Issue
Block a user