config.py

REGEX_RULE = {
    "FilmTele-Play": ["播放", "电视剧"], # 句子是不是包含特定的单词，做出分类
    "HomeAppliance-Control": ["空调", "广播"]
}




CATEGORY_NAME = [
    'Travel-Query', 'Music-Play', 'FilmTele-Play', 'Video-Play',
    'Radio-Listen', 'HomeAppliance-Control', 'Weather-Query',
    'Alarm-Update', 'Calendar-Query', 'TVProgram-Play', 'Audio-Play',
    'Other'
]

TFIDF_MODEL_PKL_PATH = "assets/weights/tfidf_ml.pkl"

BERT_MODEL_PKL_PATH = "assets/weights/bert.pt"  # 微调完成后的权重
BERT_MODEL_PERTRAINED_PATH = "assets/models/bert-base-chinese/"

LLM_OPENAI_SERVER_URL = f"https://llm-1u0cr1ya4sew3j07.cn-beijing.maas.aliyuncs.com/compatible-mode/v1" # ollama
LLM_OPENAI_API_KEY = "sk-ws-H.RYPPHRP.8vGC.MEQCIHOnIbG3Xgji7P4S5aEUrJJcNrI9cuQ9m_MZnaHRxpqlAiAdmVloCGH9CvaUwy_6IvibdA-QEzPJxi_5hosJnnLM2g"
LLM_MODEL_NAME = "qwen-turbo"