马一丁
66 lines
3.2 KiB
Python
66 lines
3.2 KiB
Python
"""
|
|
Report Engine 配置模块，统一读取环境变量并提供类型安全的访问方式。
|
|
"""
|
|
|
|
import os
|
|
from pydantic_settings import BaseSettings
|
|
from pydantic import Field
|
|
from typing import Optional
|
|
|
|
from loguru import logger
|
|
|
|
class Settings(BaseSettings):
|
|
"""Report Engine 配置，环境变量与字段均为REPORT_ENGINE_前缀一致大写。"""
|
|
REPORT_ENGINE_API_KEY: Optional[str] = Field(None, description="Report Engine LLM API密钥")
|
|
REPORT_ENGINE_BASE_URL: Optional[str] = Field(None, description="Report Engine LLM基础URL")
|
|
REPORT_ENGINE_MODEL_NAME: Optional[str] = Field(None, description="Report Engine LLM模型名称")
|
|
REPORT_ENGINE_PROVIDER: Optional[str] = Field(None, description="模型服务商，仅兼容保留")
|
|
MAX_CONTENT_LENGTH: int = Field(200000, description="最大内容长度")
|
|
OUTPUT_DIR: str = Field("final_reports", description="主输出目录")
|
|
# 章节分块JSON会存储在该目录，便于溯源与断点续传
|
|
CHAPTER_OUTPUT_DIR: str = Field(
|
|
"final_reports/chapters", description="章节JSON缓存目录"
|
|
)
|
|
# 装订后的整本IR/manifest也会持久化，方便调试与审计
|
|
DOCUMENT_IR_OUTPUT_DIR: str = Field(
|
|
"final_reports/ir", description="整本IR/Manifest输出目录"
|
|
)
|
|
TEMPLATE_DIR: str = Field("ReportEngine/report_template", description="多模板目录")
|
|
API_TIMEOUT: float = Field(900.0, description="单API超时时间（秒）")
|
|
MAX_RETRY_DELAY: float = Field(180.0, description="最大重试间隔（秒）")
|
|
MAX_RETRIES: int = Field(8, description="最大重试次数")
|
|
LOG_FILE: str = Field("logs/report.log", description="日志输出文件")
|
|
ENABLE_PDF_EXPORT: bool = Field(True, description="是否允许导出PDF")
|
|
CHART_STYLE: str = Field("modern", description="图表样式：modern/classic/")
|
|
|
|
class Config:
|
|
"""Pydantic配置：允许从.env读取并兼容大小写"""
|
|
env_file = ".env"
|
|
env_prefix = ""
|
|
case_sensitive = False
|
|
extra = "allow"
|
|
|
|
settings = Settings()
|
|
|
|
|
|
def print_config(config: Settings):
|
|
"""将当前配置项按人类可读格式输出到日志，方便排障"""
|
|
message = ""
|
|
message += "\n=== Report Engine 配置 ===\n"
|
|
message += f"LLM 模型: {config.REPORT_ENGINE_MODEL_NAME}\n"
|
|
message += f"LLM Base URL: {config.REPORT_ENGINE_BASE_URL or '(默认)'}\n"
|
|
message += f"最大内容长度: {config.MAX_CONTENT_LENGTH}\n"
|
|
message += f"输出目录: {config.OUTPUT_DIR}\n"
|
|
message += f"章节JSON目录: {config.CHAPTER_OUTPUT_DIR}\n"
|
|
message += f"整本IR目录: {config.DOCUMENT_IR_OUTPUT_DIR}\n"
|
|
message += f"模板目录: {config.TEMPLATE_DIR}\n"
|
|
message += f"API 超时时间: {config.API_TIMEOUT} 秒\n"
|
|
message += f"最大重试间隔: {config.MAX_RETRY_DELAY} 秒\n"
|
|
message += f"最大重试次数: {config.MAX_RETRIES}\n"
|
|
message += f"日志文件: {config.LOG_FILE}\n"
|
|
message += f"PDF 导出: {config.ENABLE_PDF_EXPORT}\n"
|
|
message += f"图表样式: {config.CHART_STYLE}\n"
|
|
message += f"LLM API Key: {'已配置' if config.REPORT_ENGINE_API_KEY else '未配置'}\n"
|
|
message += "=========================\n"
|
|
logger.info(message)
|