gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 主题提取bug修复，了吗？

Browse Source
This commit is contained in:
tcmofashi
2025-03-04 19:56:04 +08:00
parent 5a46ee7c70
commit 0b16001ae7
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
config/bot_config_template.toml
View File

@@ -89,7 +89,7 @@ key = "SILICONFLOW_KEY"
# 主题提取，jieba和snownlp不用api，llm需要api # 主题提取，jieba和snownlp不用api，llm需要api
[topic] [topic]
topic='llm' # 只支持jieba,snownlp,llm三种选项 topic_extract='snownlp' # 只支持jieba,snownlp,llm三种选项
[topic.llm_topic] [topic.llm_topic]
name = "Pro/deepseek-ai/DeepSeek-V3" name = "Pro/deepseek-ai/DeepSeek-V3"

5
src/plugins/chat/config.py
View File

@@ -41,8 +41,10 @@ class BotConfig:
llm_normal_minor: Dict[str, str] = field(default_factory=lambda: {}) llm_normal_minor: Dict[str, str] = field(default_factory=lambda: {})
embedding: Dict[str, str] = field(default_factory=lambda: {}) embedding: Dict[str, str] = field(default_factory=lambda: {})
vlm: Dict[str, str] = field(default_factory=lambda: {}) vlm: Dict[str, str] = field(default_factory=lambda: {})
# 主题提取配置
topic_extract: str = 'snownlp' # 只支持jieba,snownlp,llm topic_extract: str = 'snownlp' # 只支持jieba,snownlp,llm
llm_topic_extract=llm_normal_minor llm_topic_extract: Dict[str, str] = field(default_factory=lambda: {})
API_USING: str = "siliconflow" # 使用的API API_USING: str = "siliconflow" # 使用的API
API_PAID: bool = False # 是否使用付费API API_PAID: bool = False # 是否使用付费API
@@ -124,6 +126,7 @@ class BotConfig:
if "llm_normal" in model_config: if "llm_normal" in model_config:
config.llm_normal = model_config["llm_normal"] config.llm_normal = model_config["llm_normal"]
config.llm_topic_extract = config.llm_normal
if "llm_normal_minor" in model_config: if "llm_normal_minor" in model_config:
config.llm_normal_minor = model_config["llm_normal_minor"] config.llm_normal_minor = model_config["llm_normal_minor"]