82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
应用配置中心 - 集中管理所有配置项
|
||
"""
|
||
|
||
import os
|
||
from dataclasses import dataclass, field
|
||
from typing import List, Optional
|
||
|
||
|
||
@dataclass
|
||
class AppConfig:
|
||
"""应用配置中心"""
|
||
|
||
# 分析配置
|
||
max_rounds: int = field(default=20)
|
||
force_max_rounds: bool = field(default=False)
|
||
default_output_dir: str = field(default="outputs")
|
||
|
||
# 数据处理配置
|
||
max_file_size_mb: int = field(default=500) # 最大文件大小（MB）
|
||
chunk_size: int = field(default=100000) # 分块读取大小
|
||
data_cache_enabled: bool = field(default=True)
|
||
cache_dir: str = field(default=".cache/data")
|
||
|
||
# LLM配置
|
||
llm_cache_enabled: bool = field(default=True)
|
||
llm_cache_dir: str = field(default=".cache/llm")
|
||
llm_stream_enabled: bool = field(default=False)
|
||
|
||
# 代码执行配置
|
||
code_timeout: int = field(default=300) # 代码执行超时（秒）
|
||
allowed_imports: List[str] = field(default_factory=lambda: [
|
||
'pandas', 'numpy', 'matplotlib', 'seaborn', 'plotly',
|
||
'scipy', 'sklearn', 'duckdb', 'datetime', 'json',
|
||
'os', 're', 'pathlib', 'glob', 'typing', 'collections',
|
||
'itertools', 'functools', 'warnings'
|
||
])
|
||
|
||
# Web配置
|
||
web_host: str = field(default="0.0.0.0")
|
||
web_port: int = field(default=8000)
|
||
upload_dir: str = field(default="uploads")
|
||
|
||
# 日志配置
|
||
log_filename: str = field(default="log.txt")
|
||
enable_code_logging: bool = field(default=False) # 是否记录生成的代码
|
||
|
||
@classmethod
|
||
def from_env(cls) -> 'AppConfig':
|
||
"""从环境变量创建配置"""
|
||
config = cls()
|
||
|
||
# 从环境变量覆盖配置
|
||
if max_rounds := os.getenv("APP_MAX_ROUNDS"):
|
||
config.max_rounds = int(max_rounds)
|
||
|
||
if chunk_size := os.getenv("APP_CHUNK_SIZE"):
|
||
config.chunk_size = int(chunk_size)
|
||
|
||
if cache_enabled := os.getenv("APP_CACHE_ENABLED"):
|
||
config.data_cache_enabled = cache_enabled.lower() == "true"
|
||
|
||
return config
|
||
|
||
def validate(self) -> bool:
|
||
"""验证配置"""
|
||
if self.max_rounds <= 0:
|
||
raise ValueError("max_rounds must be positive")
|
||
|
||
if self.chunk_size <= 0:
|
||
raise ValueError("chunk_size must be positive")
|
||
|
||
if self.code_timeout <= 0:
|
||
raise ValueError("code_timeout must be positive")
|
||
|
||
return True
|
||
|
||
|
||
# 全局配置实例
|
||
app_config = AppConfig.from_env()
|