fix: 主题提取bug修复,了吗?

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

View File

@@ -41,8 +41,10 @@ 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
llm_topic_extract: Dict[str, str] = field(default_factory=lambda: {})
API_USING: str = "siliconflow" # 使用的API
API_PAID: bool = False # 是否使用付费API
@@ -124,6 +126,7 @@ class BotConfig:
if "llm_normal" in model_config:
config.llm_normal = model_config["llm_normal"]
config.llm_topic_extract = config.llm_normal
if "llm_normal_minor" in model_config:
config.llm_normal_minor = model_config["llm_normal_minor"]