fix: 主题提取bug修复,了吗?
This commit is contained in:
@@ -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"]
|
||||
|
||||
Reference in New Issue
Block a user