diff --git a/src/chat/emoji_system/emoji_manager.py b/src/chat/emoji_system/emoji_manager.py index d2d2e2c07..a228e7479 100644 --- a/src/chat/emoji_system/emoji_manager.py +++ b/src/chat/emoji_system/emoji_manager.py @@ -389,7 +389,7 @@ class EmojiManager: self._scan_task = None - self.vlm = LLMRequest(model_set=model_config.model_task_config.vlm, request_type="emoji") + self.vlm = LLMRequest(model_set=model_config.model_task_config.emoji_vlm, request_type="emoji") self.llm_emotion_judge = LLMRequest( model_set=model_config.model_task_config.utils, request_type="emoji" ) # 更高的温度,更少的token(后续可以根据情绪来调整温度) @@ -945,11 +945,11 @@ class EmojiManager: with get_db_session() as session: # from src.common.database.database_model_compat import Images - stmt = select(Images).where((Images.emoji_hash == image_hash) & (Images.type == "emoji")) - existing_image = session.execute(stmt).scalar_one_or_none() - if existing_image and existing_image.description: - existing_description = existing_image.description - logger.info(f"[复用描述] 找到已有详细描述: {existing_description[:50]}...") + stmt = select(Images).where((Images.emoji_hash == image_hash) & (Images.type == "emoji")) + existing_image = session.query(Images).filter((Images.emoji_hash == image_hash) & (Images.type == "emoji")).one_or_none() + if existing_image and existing_image.description: + existing_description = existing_image.description + logger.info(f"[复用描述] 找到已有详细描述: {existing_description[:50]}...") except Exception as e: logger.debug(f"查询已有描述时出错: {e}") diff --git a/src/config/api_ada_configs.py b/src/config/api_ada_configs.py index 839f747da..ccdb2762e 100644 --- a/src/config/api_ada_configs.py +++ b/src/config/api_ada_configs.py @@ -148,6 +148,12 @@ class ModelTaskConfig(ConfigBase): )) """视频分析模型配置""" + emoji_vlm: TaskConfig = field(default_factory=lambda: TaskConfig( + model_list=["qwen2.5-vl-72b"], + max_tokens=800 + )) + """表情包识别模型配置""" + def get_task(self, task_name: str) -> TaskConfig: """获取指定任务的配置""" if hasattr(self, task_name): diff --git a/template/model_config_template.toml b/template/model_config_template.toml index 208d5bb60..649c42c8d 100644 --- a/template/model_config_template.toml +++ b/template/model_config_template.toml @@ -1,5 +1,5 @@ [inner] -version = "1.2.2" +version = "1.2.3" # 配置文件版本号迭代规则同bot_config.toml @@ -144,6 +144,11 @@ max_tokens = 800 model_list = ["qwen2.5-vl-72b"] max_tokens = 800 +[model_task_config.emoji_vlm] # 专用表情包识别模型 +model_list = ["qwen2.5-vl-72b"] +max_tokens = 800 + + [model_task_config.utils_video] # 专用视频分析模型 model_list = ["qwen2.5-vl-72b"] temperature = 0.3