fix: 修改embedding

This commit is contained in:
tcmofashi
2025-03-07 03:40:14 +08:00
parent a3b8a545af
commit 0ced4939ec
4 changed files with 5 additions and 5 deletions

View File

@@ -22,7 +22,7 @@ check_interval = 120 # 检查表情包的时间间隔
register_interval = 10 # 注册表情包的时间间隔
auto_save = true # 自动偷表情包
enable_check = false # 是否启用表情包过滤
check_prompt = "不要包含违反公序良俗的内容" # 表情包过滤要求
check_prompt = "符合公序良俗" # 表情包过滤要求
[cq_code]
enable_pic_translate = false

View File

@@ -32,7 +32,7 @@ class BotConfig:
EMOJI_REGISTER_INTERVAL: int = 10 # 表情包注册间隔(分钟)
EMOJI_SAVE: bool = True # 偷表情包
EMOJI_CHECK: bool = False #是否开启过滤
EMOJI_CHECK_PROMPT: str = "不要包含违反公序良俗的内容" # 表情包过滤要求
EMOJI_CHECK_PROMPT: str = "符合公序良俗" # 表情包过滤要求
ban_words = set()

View File

@@ -98,7 +98,7 @@ class EmojiManager:
# 获取文本的embedding
text_for_search= await self._get_kimoji_for_text(text)
text_embedding = get_embedding(text_for_search)
text_embedding = await get_embedding(text_for_search)
if not text_embedding:
logger.error("无法获取文本的embedding")
return None
@@ -299,7 +299,7 @@ class EmojiManager:
logger.info(f"其不满足过滤规则,被剔除 {check}")
continue
logger.info(f"check通过 {check}")
embedding = get_embedding(discription)
embedding = await get_embedding(discription)
if discription is not None:
# 准备数据库记录
emoji_record = {

View File

@@ -673,7 +673,7 @@ class Hippocampus:
if first_layer:
# 如果记忆条数超过限制,随机选择指定数量的记忆
if len(first_layer) > max_memory_num/2:
first_layer = random.sample(first_layer, max_memory_num)
first_layer = random.sample(first_layer, max_memory_num//2)
# 为每条记忆添加来源主题和相似度信息
for memory in first_layer:
relevant_memories.append({