fix: 修改embedding
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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()
|
||||
|
||||
|
||||
@@ -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 = {
|
||||
|
||||
@@ -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({
|
||||
|
||||
Reference in New Issue
Block a user