first commit
This commit is contained in:
54
src/config/config.py
Normal file
54
src/config/config.py
Normal file
@@ -0,0 +1,54 @@
|
||||
import os
|
||||
from typing import Dict, Any
|
||||
|
||||
class Config:
|
||||
"""系统配置类"""
|
||||
|
||||
# 阿里云千问API配置
|
||||
ALIBABA_API_KEY = "sk-c0dbefa1718d46eaa897199135066f00"
|
||||
ALIBABA_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
|
||||
ALIBABA_MODEL_NAME = "qwen-plus-latest"
|
||||
|
||||
# 数据库配置
|
||||
DATABASE_URL = "mysql+pymysql://root:123456@localhost/tsp_assistant?charset=utf8mb4"
|
||||
|
||||
# 知识库配置
|
||||
KNOWLEDGE_BASE_PATH = "data/knowledge_base"
|
||||
VECTOR_DB_PATH = "data/vector_db"
|
||||
|
||||
# 对话配置
|
||||
MAX_HISTORY_LENGTH = 10
|
||||
RESPONSE_TIMEOUT = 30
|
||||
|
||||
# 分析配置
|
||||
ANALYTICS_UPDATE_INTERVAL = 3600 # 1小时
|
||||
ALERT_THRESHOLD = 0.8 # 预警阈值
|
||||
|
||||
# 日志配置
|
||||
LOG_LEVEL = "INFO"
|
||||
LOG_FILE = "logs/tsp_assistant.log"
|
||||
|
||||
@classmethod
|
||||
def get_api_config(cls) -> Dict[str, Any]:
|
||||
"""获取API配置"""
|
||||
return {
|
||||
"api_key": cls.ALIBABA_API_KEY,
|
||||
"base_url": cls.ALIBABA_BASE_URL,
|
||||
"model_name": cls.ALIBABA_MODEL_NAME
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def get_database_config(cls) -> Dict[str, Any]:
|
||||
"""获取数据库配置"""
|
||||
return {
|
||||
"url": cls.DATABASE_URL,
|
||||
"echo": False
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def get_knowledge_config(cls) -> Dict[str, Any]:
|
||||
"""获取知识库配置"""
|
||||
return {
|
||||
"base_path": cls.KNOWLEDGE_BASE_PATH,
|
||||
"vector_db_path": cls.VECTOR_DB_PATH
|
||||
}
|
||||
Reference in New Issue
Block a user