commit
This commit is contained in:
17
config.py
17
config.py
@@ -48,6 +48,7 @@ class BaseConfig:
|
||||
CLICK_INTERVAL_MINUTES = int(os.getenv('CLICK_INTERVAL_MINUTES', 30)) # 点击间隔(分钟)
|
||||
MIN_TASK_INTERVAL_MINUTES = int(os.getenv('MIN_TASK_INTERVAL_MINUTES', 3)) # 任务间最小间隔(分钟)
|
||||
MAX_TASK_INTERVAL_MINUTES = int(os.getenv('MAX_TASK_INTERVAL_MINUTES', 5)) # 任务间最大间隔(分钟)
|
||||
MAX_CONCURRENT_WORKERS = int(os.getenv('MAX_CONCURRENT_WORKERS', 2)) # 最大并发任务数(默认2)
|
||||
WORK_START_HOUR = int(os.getenv('WORK_START_HOUR', 9)) # 工作开始时间
|
||||
WORK_END_HOUR = int(os.getenv('WORK_END_HOUR', 21)) # 工作结束时间
|
||||
REPLY_WAIT_TIMEOUT = int(os.getenv('REPLY_WAIT_TIMEOUT', 30)) # 回复等待超时(秒)
|
||||
@@ -60,6 +61,7 @@ class BaseConfig:
|
||||
# 数据存储路径
|
||||
DATA_DIR = os.getenv('DATA_DIR', './data')
|
||||
LOG_DIR = os.getenv('LOG_DIR', './logs')
|
||||
QUERY_UPLOAD_DIR = os.getenv('QUERY_UPLOAD_DIR', './query_upload') # Query挖掘上传目录
|
||||
|
||||
# 调试模式
|
||||
DEBUG = os.getenv('DEBUG', 'False').lower() == 'true'
|
||||
@@ -74,11 +76,24 @@ class BaseConfig:
|
||||
MYSQL_PASSWORD = os.getenv('MYSQL_PASSWORD', '')
|
||||
MYSQL_DATABASE = os.getenv('MYSQL_DATABASE', 'ai_article')
|
||||
|
||||
# 远程点击服务配置(分布式部署时使用)
|
||||
# 为空表示本地模式,调度器在Web服务内运行
|
||||
# 设置URL表示远程模式,Web服务转发请求到远程点击服务
|
||||
CLICK_SERVICE_URL = os.getenv('CLICK_SERVICE_URL', '') # 例如: http://192.168.1.100:8888
|
||||
|
||||
# 服务模式:web=提供前端界面, click=仅提供调度API
|
||||
SERVICE_MODE = os.getenv('SERVICE_MODE', 'web')
|
||||
|
||||
# 千问大模型API配置
|
||||
QWEN_API_KEY = os.getenv('QWEN_API_KEY', 'sk-6d22dd845a624d9c92a821d24a50e2e8')
|
||||
QWEN_API_URL = os.getenv('QWEN_API_URL', 'https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions')
|
||||
|
||||
@classmethod
|
||||
def ensure_dirs(cls):
|
||||
"""确保必要的目录存在"""
|
||||
os.makedirs(cls.DATA_DIR, exist_ok=True)
|
||||
os.makedirs(cls.LOG_DIR, exist_ok=True)
|
||||
os.makedirs(cls.QUERY_UPLOAD_DIR, exist_ok=True)
|
||||
|
||||
|
||||
class DevelopmentConfig(BaseConfig):
|
||||
@@ -88,7 +103,7 @@ class DevelopmentConfig(BaseConfig):
|
||||
|
||||
class ProductionConfig(BaseConfig):
|
||||
"""生产环境配置"""
|
||||
DEBUG = False
|
||||
DEBUG = True
|
||||
|
||||
|
||||
# 根据环境选择配置
|
||||
|
||||
Reference in New Issue
Block a user