115 lines
3.3 KiB
Python
115 lines
3.3 KiB
Python
# 文件位置: config.py
|
||
# 文件名: config.py
|
||
|
||
import os
|
||
from datetime import timedelta
|
||
from dotenv import load_dotenv
|
||
|
||
# 加载环境变量
|
||
load_dotenv()
|
||
|
||
class Config:
|
||
"""基础配置"""
|
||
# Flask配置
|
||
SECRET_KEY = os.environ.get('SECRET_KEY', 'dev-secret-key-change-in-production')
|
||
DEBUG = False
|
||
TESTING = False
|
||
|
||
# API配置
|
||
DEEPSEEK_API_KEY = os.environ.get('DEEPSEEK_API_KEY')
|
||
DEEPSEEK_BASE_URL = os.environ.get('DEEPSEEK_BASE_URL', 'https://api.deepseek.com/v1')
|
||
TAVILY_API_KEY = os.environ.get('TAVILY_API_KEY')
|
||
|
||
# 模型配置
|
||
R1_MODEL = "deepseek-reasoner" # R1-0528
|
||
V3_MODEL = "deepseek-chat" # V3-0324
|
||
|
||
# 研究配置
|
||
MAX_CONCURRENT_SUBTOPICS = 10
|
||
MAX_SEARCHES_HIGH_PRIORITY = 15
|
||
MAX_SEARCHES_MEDIUM_PRIORITY = 10
|
||
MAX_SEARCHES_LOW_PRIORITY = 5
|
||
|
||
# 搜索配置
|
||
TAVILY_MAX_RESULTS = 10
|
||
TAVILY_SEARCH_DEPTH = "advanced"
|
||
TAVILY_INCLUDE_ANSWER = True
|
||
TAVILY_INCLUDE_RAW_CONTENT = False
|
||
|
||
# 任务管理配置（替代Celery）
|
||
TASK_POOL_SIZE = 10 # 线程池大小
|
||
TASK_TIMEOUT = {
|
||
'question_analysis': 60,
|
||
'outline_creation': 120,
|
||
'search': 30,
|
||
'report_generation': 180
|
||
}
|
||
|
||
# 重试配置
|
||
MAX_RETRIES = 3
|
||
RETRY_DELAY = 5 # 秒
|
||
|
||
# 存储配置
|
||
DATA_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data')
|
||
SESSIONS_DIR = os.path.join(DATA_DIR, 'sessions')
|
||
REPORTS_DIR = os.path.join(DATA_DIR, 'reports')
|
||
CACHE_DIR = os.path.join(DATA_DIR, 'cache')
|
||
|
||
# 日志配置
|
||
LOG_LEVEL = os.environ.get('LOG_LEVEL', 'INFO')
|
||
LOG_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'logs')
|
||
|
||
# MongoDB配置（可选）
|
||
MONGODB_URI = os.environ.get('MONGODB_URI', 'mongodb://localhost:27017/deepresearch')
|
||
|
||
# WebSocket配置
|
||
SOCKETIO_ASYNC_MODE = 'eventlet'
|
||
|
||
@staticmethod
|
||
def init_app(app):
|
||
"""初始化应用配置"""
|
||
# 确保必要的目录存在
|
||
for dir_path in [Config.DATA_DIR, Config.SESSIONS_DIR,
|
||
Config.REPORTS_DIR, Config.CACHE_DIR, Config.LOG_DIR]:
|
||
os.makedirs(dir_path, exist_ok=True)
|
||
|
||
class DevelopmentConfig(Config):
|
||
"""开发环境配置"""
|
||
DEBUG = True
|
||
|
||
class ProductionConfig(Config):
|
||
"""生产环境配置"""
|
||
DEBUG = False
|
||
|
||
@classmethod
|
||
def init_app(cls, app):
|
||
Config.init_app(app)
|
||
|
||
# 生产环境特定的初始化
|
||
import logging
|
||
from logging.handlers import RotatingFileHandler
|
||
|
||
if not app.debug and not app.testing:
|
||
file_handler = RotatingFileHandler(
|
||
os.path.join(cls.LOG_DIR, 'deepresearch.log'),
|
||
maxBytes=10485760,
|
||
backupCount=10
|
||
)
|
||
file_handler.setFormatter(logging.Formatter(
|
||
'%(asctime)s %(levelname)s: %(message)s [in %(pathname)s:%(lineno)d]'
|
||
))
|
||
file_handler.setLevel(logging.INFO)
|
||
app.logger.addHandler(file_handler)
|
||
app.logger.setLevel(logging.INFO)
|
||
app.logger.info('DeepResearch startup')
|
||
|
||
class TestingConfig(Config):
|
||
"""测试环境配置"""
|
||
TESTING = True
|
||
|
||
config = {
|
||
'development': DevelopmentConfig,
|
||
'production': ProductionConfig,
|
||
'testing': TestingConfig,
|
||
'default': DevelopmentConfig
|
||
} |