fix: 调整topic提取的设置

This commit is contained in:
tcmofashi
2025-03-04 19:40:00 +08:00
parent f1a2ccfccf
commit 5a46ee7c70
4 changed files with 45 additions and 8 deletions

View File

@@ -41,6 +41,8 @@ class BotConfig:
llm_normal_minor: Dict[str, str] = field(default_factory=lambda: {})
embedding: Dict[str, str] = field(default_factory=lambda: {})
vlm: Dict[str, str] = field(default_factory=lambda: {})
topic_extract: str = 'snownlp' # 只支持jieba,snownlp,llm
llm_topic_extract=llm_normal_minor
API_USING: str = "siliconflow" # 使用的API
API_PAID: bool = False # 是否使用付费API
@@ -132,6 +134,15 @@ class BotConfig:
if "embedding" in model_config:
config.embedding = model_config["embedding"]
if 'topic' in toml_dict:
topic_config=toml_dict['topic']
if 'topic_extract' in topic_config:
config.topic_extract=topic_config.get('topic_extract',config.topic_extract)
print(f"载入自定义主题提取为{config.topic_extract}")
if config.topic_extract=='llm' and 'llm_topic' in topic_config:
config.llm_topic_extract=topic_config['llm_topic']
print(f"载入自定义主题提取模型为{config.llm_topic_extract['name']}")
# 消息配置
if "message" in toml_dict:
msg_config = toml_dict["message"]