From 0ced4939ec20f4185106f83d221273f7678d4f94 Mon Sep 17 00:00:00 2001
From: tcmofashi <mofashiforzbx@qq.com>
Date: Fri, 7 Mar 2025 03:40:14 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E6=94=B9embedding?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 config/bot_config_template.toml     | 2 +-
 src/plugins/chat/config.py          | 2 +-
 src/plugins/chat/emoji_manager.py   | 4 ++--
 src/plugins/memory_system/memory.py | 2 +-
 4 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/config/bot_config_template.toml b/config/bot_config_template.toml
index 3287b3d20..507c6d2d6 100644
--- a/config/bot_config_template.toml
+++ b/config/bot_config_template.toml
@@ -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
diff --git a/src/plugins/chat/config.py b/src/plugins/chat/config.py
index 6cb8b9fee..a2adc9e30 100644
--- a/src/plugins/chat/config.py
+++ b/src/plugins/chat/config.py
@@ -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()
 
diff --git a/src/plugins/chat/emoji_manager.py b/src/plugins/chat/emoji_manager.py
index 3592bd09b..1cdb62c07 100644
--- a/src/plugins/chat/emoji_manager.py
+++ b/src/plugins/chat/emoji_manager.py
@@ -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 = {
diff --git a/src/plugins/memory_system/memory.py b/src/plugins/memory_system/memory.py
index 49d19c253..a25e15bdf 100644
--- a/src/plugins/memory_system/memory.py
+++ b/src/plugins/memory_system/memory.py
@@ -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({