fix: 修改embedding
This commit is contained in:
@@ -22,7 +22,7 @@ check_interval = 120 # 检查表情包的时间间隔
|
|||||||
register_interval = 10 # 注册表情包的时间间隔
|
register_interval = 10 # 注册表情包的时间间隔
|
||||||
auto_save = true # 自动偷表情包
|
auto_save = true # 自动偷表情包
|
||||||
enable_check = false # 是否启用表情包过滤
|
enable_check = false # 是否启用表情包过滤
|
||||||
check_prompt = "不要包含违反公序良俗的内容" # 表情包过滤要求
|
check_prompt = "符合公序良俗" # 表情包过滤要求
|
||||||
|
|
||||||
[cq_code]
|
[cq_code]
|
||||||
enable_pic_translate = false
|
enable_pic_translate = false
|
||||||
|
|||||||
@@ -32,7 +32,7 @@ class BotConfig:
|
|||||||
EMOJI_REGISTER_INTERVAL: int = 10 # 表情包注册间隔(分钟)
|
EMOJI_REGISTER_INTERVAL: int = 10 # 表情包注册间隔(分钟)
|
||||||
EMOJI_SAVE: bool = True # 偷表情包
|
EMOJI_SAVE: bool = True # 偷表情包
|
||||||
EMOJI_CHECK: bool = False #是否开启过滤
|
EMOJI_CHECK: bool = False #是否开启过滤
|
||||||
EMOJI_CHECK_PROMPT: str = "不要包含违反公序良俗的内容" # 表情包过滤要求
|
EMOJI_CHECK_PROMPT: str = "符合公序良俗" # 表情包过滤要求
|
||||||
|
|
||||||
ban_words = set()
|
ban_words = set()
|
||||||
|
|
||||||
|
|||||||
@@ -98,7 +98,7 @@ class EmojiManager:
|
|||||||
|
|
||||||
# 获取文本的embedding
|
# 获取文本的embedding
|
||||||
text_for_search= await self._get_kimoji_for_text(text)
|
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:
|
if not text_embedding:
|
||||||
logger.error("无法获取文本的embedding")
|
logger.error("无法获取文本的embedding")
|
||||||
return None
|
return None
|
||||||
@@ -299,7 +299,7 @@ class EmojiManager:
|
|||||||
logger.info(f"其不满足过滤规则,被剔除 {check}")
|
logger.info(f"其不满足过滤规则,被剔除 {check}")
|
||||||
continue
|
continue
|
||||||
logger.info(f"check通过 {check}")
|
logger.info(f"check通过 {check}")
|
||||||
embedding = get_embedding(discription)
|
embedding = await get_embedding(discription)
|
||||||
if discription is not None:
|
if discription is not None:
|
||||||
# 准备数据库记录
|
# 准备数据库记录
|
||||||
emoji_record = {
|
emoji_record = {
|
||||||
|
|||||||
@@ -673,7 +673,7 @@ class Hippocampus:
|
|||||||
if first_layer:
|
if first_layer:
|
||||||
# 如果记忆条数超过限制,随机选择指定数量的记忆
|
# 如果记忆条数超过限制,随机选择指定数量的记忆
|
||||||
if len(first_layer) > max_memory_num/2:
|
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:
|
for memory in first_layer:
|
||||||
relevant_memories.append({
|
relevant_memories.append({
|
||||||
|
|||||||
Reference in New Issue
Block a user