241 lines
11 KiB
Python
241 lines
11 KiB
Python
import json
|
||
import os
|
||
from pathlib import Path
|
||
from typing import Any, Dict, Optional
|
||
|
||
from dotenv import load_dotenv
|
||
|
||
load_dotenv()
|
||
|
||
|
||
class Settings:
|
||
# ========================================================================
|
||
# Deployment Mode Configuration
|
||
# ========================================================================
|
||
# community: 社区版（开源，功能受限）
|
||
# cloud: SaaS 云服务版（全功能，按量计费）
|
||
# enterprise: 企业私有化版（License 控制）
|
||
DEPLOYMENT_MODE: str = os.getenv("DEPLOYMENT_MODE", "community")
|
||
|
||
# License 配置（企业版）
|
||
LICENSE_FILE: str = os.getenv("LICENSE_FILE", "/etc/app/license.json")
|
||
LICENSE_SERVER_URL: str = os.getenv("LICENSE_SERVER_URL", "https://license.yourcompany.com")
|
||
|
||
# 计费服务配置（SaaS 版）
|
||
BILLING_SERVICE_URL: str = os.getenv("BILLING_SERVICE_URL", "")
|
||
|
||
# 基础 URL（用于 SSO 回调等）
|
||
BASE_URL: str = os.getenv("BASE_URL", "http://localhost:8000")
|
||
FRONTEND_URL: str = os.getenv("FRONTEND_URL", "http://localhost:3000")
|
||
|
||
ENABLE_SINGLE_WORKSPACE: bool = os.getenv("ENABLE_SINGLE_WORKSPACE", "true").lower() == "true"
|
||
# API Keys Configuration
|
||
OPENAI_API_KEY: str = os.getenv("OPENAI_API_KEY", "")
|
||
DASHSCOPE_API_KEY: str = os.getenv("DASHSCOPE_API_KEY", "")
|
||
|
||
# Neo4j Configuration (记忆系统数据库)
|
||
NEO4J_URI: str = os.getenv("NEO4J_URI", "bolt://1.94.111.67:7687")
|
||
NEO4J_USERNAME: str = os.getenv("NEO4J_USERNAME", "neo4j")
|
||
NEO4J_PASSWORD: str = os.getenv("NEO4J_PASSWORD", "")
|
||
|
||
# Database configuration (Postgres)
|
||
DB_HOST: str = os.getenv("DB_HOST", "127.0.0.1")
|
||
DB_PORT: int = int(os.getenv("DB_PORT", "5432"))
|
||
DB_USER: str = os.getenv("DB_USER", "postgres")
|
||
DB_PASSWORD: str = os.getenv("DB_PASSWORD", "password")
|
||
DB_NAME: str = os.getenv("DB_NAME", "redbear-mem")
|
||
DB_POOL_SIZE: int = int(os.getenv("DB_POOL_SIZE", "50"))
|
||
DB_MAX_OVERFLOW: int = int(os.getenv("DB_MAX_OVERFLOW", "20"))
|
||
DB_POOL_RECYCLE: int = int(os.getenv("DB_POOL_RECYCLE", "1800"))
|
||
DB_POOL_TIMEOUT: int = int(os.getenv("DB_POOL_TIMEOUT", "30"))
|
||
DB_POOL_PRE_PING: bool = os.getenv("DB_POOL_PRE_PING", "true").lower() == "true"
|
||
|
||
DB_AUTO_UPGRADE = os.getenv("DB_AUTO_UPGRADE", "false").lower() == "true"
|
||
|
||
# Redis configuration
|
||
REDIS_HOST: str = os.getenv("REDIS_HOST", "127.0.0.1")
|
||
REDIS_PORT: int = int(os.getenv("REDIS_PORT", "6379"))
|
||
REDIS_DB: int = int(os.getenv("REDIS_DB", "1"))
|
||
REDIS_PASSWORD: str = os.getenv("REDIS_PASSWORD", "")
|
||
|
||
|
||
# ElasticSearch configuration
|
||
ELASTICSEARCH_HOST: str = os.getenv("ELASTICSEARCH_HOST", "https://127.0.0.1")
|
||
ELASTICSEARCH_PORT: int = int(os.getenv("ELASTICSEARCH_PORT", "9200"))
|
||
ELASTICSEARCH_USERNAME: str = os.getenv("ELASTICSEARCH_USERNAME", "elastic")
|
||
ELASTICSEARCH_PASSWORD: str = os.getenv("ELASTICSEARCH_PASSWORD", "")
|
||
ELASTICSEARCH_VERIFY_CERTS: bool = os.getenv("ELASTICSEARCH_VERIFY_CERTS", "False").lower() == "true"
|
||
ELASTICSEARCH_CA_CERTS: str = os.getenv("ELASTICSEARCH_CA_CERTS", "")
|
||
ELASTICSEARCH_REQUEST_TIMEOUT: int = int(os.getenv("ELASTICSEARCH_REQUEST_TIMEOUT", "100000"))
|
||
ELASTICSEARCH_RETRY_ON_TIMEOUT: bool = os.getenv("ELASTICSEARCH_RETRY_ON_TIMEOUT", "True").lower() == "true"
|
||
ELASTICSEARCH_MAX_RETRIES: int = int(os.getenv("ELASTICSEARCH_MAX_RETRIES", "10"))
|
||
|
||
# Xinference configuration
|
||
XINFERENCE_URL: str = os.getenv("XINFERENCE_URL", "http://127.0.0.1")
|
||
|
||
# LangSmith configuration
|
||
LANGCHAIN_TRACING_V2: bool = os.getenv("LANGCHAIN_TRACING_V2", "false").lower() == "true"
|
||
LANGCHAIN_TRACING: bool = os.getenv("LANGCHAIN_TRACING", "false").lower() == "true"
|
||
LANGCHAIN_API_KEY: str = os.getenv("LANGCHAIN_API_KEY", "")
|
||
LANGCHAIN_ENDPOINT: str = os.getenv("LANGCHAIN_ENDPOINT", "")
|
||
|
||
# LLM Request Configuration
|
||
LLM_TIMEOUT: float = float(os.getenv("LLM_TIMEOUT", "120.0"))
|
||
LLM_MAX_RETRIES: int = int(os.getenv("LLM_MAX_RETRIES", "2"))
|
||
|
||
# JWT Token Configuration
|
||
SECRET_KEY: str = os.getenv("SECRET_KEY", "a_default_secret_key_that_is_long_and_random")
|
||
ALGORITHM: str = "HS256"
|
||
ACCESS_TOKEN_EXPIRE_MINUTES: int = int(os.getenv("ACCESS_TOKEN_EXPIRE_MINUTES", "30"))
|
||
REFRESH_TOKEN_EXPIRE_DAYS: int = int(os.getenv("REFRESH_TOKEN_EXPIRE_DAYS", "7"))
|
||
|
||
# Single Sign-On configuration
|
||
ENABLE_SINGLE_SESSION: bool = os.getenv("ENABLE_SINGLE_SESSION", "false").lower() == "true"
|
||
|
||
# SSO 免登配置
|
||
SSO_TOKEN_EXPIRE_SECONDS: int = int(os.getenv("SSO_TOKEN_EXPIRE_SECONDS", "300"))
|
||
SSO_TRUSTED_SOURCES_CONFIG: str = os.getenv("SSO_TRUSTED_SOURCES_CONFIG", "{}")
|
||
|
||
# File Upload
|
||
MAX_FILE_SIZE: int = int(os.getenv("MAX_FILE_SIZE", "52428800"))
|
||
FILE_PATH: str = os.getenv("FILE_PATH", "/files")
|
||
FILE_URL_EXPIRES: int = int(os.getenv("FILE_URL_EXPIRES", "3600"))
|
||
|
||
# Storage Configuration
|
||
STORAGE_TYPE: str = os.getenv("STORAGE_TYPE", "local")
|
||
|
||
# Aliyun OSS Configuration
|
||
OSS_ENDPOINT: str = os.getenv("OSS_ENDPOINT", "")
|
||
OSS_ACCESS_KEY_ID: str = os.getenv("OSS_ACCESS_KEY_ID", "")
|
||
OSS_ACCESS_KEY_SECRET: str = os.getenv("OSS_ACCESS_KEY_SECRET", "")
|
||
OSS_BUCKET_NAME: str = os.getenv("OSS_BUCKET_NAME", "")
|
||
|
||
# AWS S3 Configuration
|
||
S3_REGION: str = os.getenv("S3_REGION", "")
|
||
S3_ACCESS_KEY_ID: str = os.getenv("S3_ACCESS_KEY_ID", "")
|
||
S3_SECRET_ACCESS_KEY: str = os.getenv("S3_SECRET_ACCESS_KEY", "")
|
||
S3_BUCKET_NAME: str = os.getenv("S3_BUCKET_NAME", "")
|
||
|
||
# VOLC ASR settings
|
||
VOLC_APP_KEY: str = os.getenv("VOLC_APP_KEY", "")
|
||
VOLC_ACCESS_KEY: str = os.getenv("VOLC_ACCESS_KEY", "")
|
||
VOLC_SUBMIT_URL: str = os.getenv("VOLC_SUBMIT_URL", "https://openspeech.bytedance.com/api/v3/auc/bigmodel/submit")
|
||
VOLC_QUERY_URL: str = os.getenv("VOLC_QUERY_URL", "https://openspeech.bytedance.com/api/v3/auc/bigmodel/query")
|
||
|
||
# Langfuse configuration
|
||
LANGFUSE_ENABLED: bool = os.getenv("LANGFUSE_ENABLED", "false").lower() == "true"
|
||
LANGFUSE_PUBLIC_KEY: str = os.getenv("LANGFUSE_PUBLIC_KEY", "")
|
||
LANGFUSE_SECRET_KEY: str = os.getenv("LANGFUSE_SECRET_KEY", "")
|
||
LANGFUSE_HOST: str = os.getenv("LANGFUSE_HOST", "")
|
||
|
||
# Server Configuration
|
||
SERVER_IP: str = os.getenv("SERVER_IP", "127.0.0.1")
|
||
FILE_LOCAL_SERVER_URL : str = os.getenv("FILE_LOCAL_SERVER_URL", "http://localhost:8000/api")
|
||
|
||
# ========================================================================
|
||
# Internal Configuration (not in .env, used by application code)
|
||
# ========================================================================
|
||
|
||
# Superuser settings (internal defaults)
|
||
FIRST_SUPERUSER_EMAIL: str = os.getenv("FIRST_SUPERUSER_EMAIL", "admin@example.com")
|
||
FIRST_SUPERUSER_USERNAME: str = os.getenv("FIRST_SUPERUSER_USERNAME", "admin")
|
||
FIRST_SUPERUSER_PASSWORD: str = os.getenv("FIRST_SUPERUSER_PASSWORD", "admin_password")
|
||
|
||
# Generic File Upload (internal)
|
||
GENERIC_FILE_PATH: str = os.getenv("GENERIC_FILE_PATH", "/uploads")
|
||
ENABLE_FILE_COMPRESSION: bool = os.getenv("ENABLE_FILE_COMPRESSION", "false").lower() == "true"
|
||
ENABLE_VIRUS_SCAN: bool = os.getenv("ENABLE_VIRUS_SCAN", "false").lower() == "true"
|
||
FILE_ACCESS_URL_PREFIX: str = os.getenv("FILE_ACCESS_URL_PREFIX", "http://localhost:8000/api/files")
|
||
|
||
# Frontend URL for workspace invitations (internal)
|
||
WEB_URL: str = os.getenv("WEB_URL", "http://localhost:3000")
|
||
|
||
# CORS configuration (internal)
|
||
CORS_ORIGINS: list[str] = [
|
||
origin.strip()
|
||
for origin in os.getenv("CORS_ORIGINS", "").split(",")
|
||
if origin.strip()
|
||
]
|
||
|
||
# Logging settings
|
||
LOG_LEVEL: str = os.getenv("LOG_LEVEL", "INFO")
|
||
LOG_FORMAT: str = os.getenv("LOG_FORMAT", "%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
||
LOG_FILE_PATH: str = os.getenv("LOG_FILE_PATH", "logs/app.log")
|
||
LOG_MAX_SIZE: int = int(os.getenv("LOG_MAX_SIZE", "10485760")) # 10MB
|
||
LOG_BACKUP_COUNT: int = int(os.getenv("LOG_BACKUP_COUNT", "5"))
|
||
LOG_TO_CONSOLE: bool = os.getenv("LOG_TO_CONSOLE", "true").lower() == "true"
|
||
LOG_TO_FILE: bool = os.getenv("LOG_TO_FILE", "true").lower() == "true"
|
||
|
||
# Sensitive Data Filtering
|
||
ENABLE_SENSITIVE_DATA_FILTER: bool = os.getenv("ENABLE_SENSITIVE_DATA_FILTER", "true").lower() == "true"
|
||
|
||
# Memory Module Logging
|
||
PROMPT_LOG_LEVEL: str = os.getenv("PROMPT_LOG_LEVEL", "INFO")
|
||
ENABLE_TEMPLATE_LOGGING: bool = os.getenv("ENABLE_TEMPLATE_LOGGING", "false").lower() == "true"
|
||
TIMING_LOG_FILE: str = os.getenv("TIMING_LOG_FILE", "logs/time.log")
|
||
TIMING_LOG_TO_CONSOLE: bool = os.getenv("TIMING_LOG_TO_CONSOLE", "true").lower() == "true"
|
||
AGENT_LOG_FILE: str = os.getenv("AGENT_LOG_FILE", "logs/agent_service.log")
|
||
AGENT_LOG_MAX_SIZE: int = int(os.getenv("AGENT_LOG_MAX_SIZE", "5242880")) # 5MB
|
||
AGENT_LOG_BACKUP_COUNT: int = int(os.getenv("AGENT_LOG_BACKUP_COUNT", "20"))
|
||
|
||
# Log Streaming Configuration
|
||
LOG_STREAM_KEEPALIVE_INTERVAL: int = int(os.getenv("LOG_STREAM_KEEPALIVE_INTERVAL", "300")) # 5 minutes
|
||
LOG_STREAM_MAX_CONNECTIONS: int = int(os.getenv("LOG_STREAM_MAX_CONNECTIONS", "10"))
|
||
LOG_STREAM_BUFFER_SIZE: int = int(os.getenv("LOG_STREAM_BUFFER_SIZE", "8192")) # 8KB
|
||
LOG_FILE_MAX_SIZE_MB: int = int(os.getenv("LOG_FILE_MAX_SIZE_MB", "10")) # 10MB
|
||
|
||
# Celery configuration (internal)
|
||
CELERY_BROKER: int = int(os.getenv("CELERY_BROKER", "1"))
|
||
CELERY_BACKEND: int = int(os.getenv("CELERY_BACKEND", "2"))
|
||
|
||
REFLECTION_INTERVAL_SECONDS: float = float(os.getenv("REFLECTION_INTERVAL_SECONDS", "300"))
|
||
HEALTH_CHECK_SECONDS: float = float(os.getenv("HEALTH_CHECK_SECONDS", "600"))
|
||
MEMORY_INCREMENT_INTERVAL_HOURS: float = float(os.getenv("MEMORY_INCREMENT_INTERVAL_HOURS", "24"))
|
||
DEFAULT_WORKSPACE_ID: Optional[str] = os.getenv("DEFAULT_WORKSPACE_ID", None)
|
||
REFLECTION_INTERVAL_TIME: Optional[str] = int(os.getenv("REFLECTION_INTERVAL_TIME", 30))
|
||
|
||
# Memory Cache Regeneration Configuration
|
||
MEMORY_CACHE_REGENERATION_HOURS: int = int(os.getenv("MEMORY_CACHE_REGENERATION_HOURS", "24"))
|
||
|
||
# Memory Module Configuration (internal)
|
||
MEMORY_OUTPUT_DIR: str = os.getenv("MEMORY_OUTPUT_DIR", "logs/memory-output")
|
||
MEMORY_CONFIG_DIR: str = os.getenv("MEMORY_CONFIG_DIR", "app/core/memory")
|
||
|
||
# Tool Management Configuration
|
||
TOOL_CONFIG_DIR: str = os.getenv("TOOL_CONFIG_DIR", "app/core/tools")
|
||
TOOL_EXECUTION_TIMEOUT: int = int(os.getenv("TOOL_EXECUTION_TIMEOUT", "60"))
|
||
TOOL_MAX_CONCURRENCY: int = int(os.getenv("TOOL_MAX_CONCURRENCY", "10"))
|
||
ENABLE_TOOL_MANAGEMENT: bool = os.getenv("ENABLE_TOOL_MANAGEMENT", "true").lower() == "true"
|
||
|
||
# official environment system version
|
||
SYSTEM_VERSION: str = os.getenv("SYSTEM_VERSION", "v0.2.1")
|
||
|
||
# workflow config
|
||
WORKFLOW_NODE_TIMEOUT: int = int(os.getenv("WORKFLOW_NODE_TIMEOUT", 600))
|
||
|
||
def get_memory_output_path(self, filename: str = "") -> str:
|
||
"""
|
||
Get the full path for memory module output files.
|
||
|
||
Args:
|
||
filename: Optional filename to append to the output directory
|
||
|
||
Returns:
|
||
Full path to the output file or directory
|
||
"""
|
||
base_path = Path(self.MEMORY_OUTPUT_DIR)
|
||
if filename:
|
||
return str(base_path / filename)
|
||
return str(base_path)
|
||
|
||
def ensure_memory_output_dir(self) -> None:
|
||
"""
|
||
Ensure the memory output directory exists.
|
||
Creates the directory if it doesn't exist.
|
||
"""
|
||
output_dir = Path(self.MEMORY_OUTPUT_DIR)
|
||
output_dir.mkdir(parents=True, exist_ok=True)
|
||
|
||
|
||
settings = Settings()
|