good version for 算法注册

This commit is contained in:
2026-02-15 21:23:28 +08:00
parent 3c03777b97
commit 62ea5d36a5
115 changed files with 9566 additions and 1576 deletions

View File

@@ -1,5 +1,6 @@
from pydantic_settings import BaseSettings
from typing import Optional
from typing import Optional, Dict, Any
from sqlalchemy.orm import Session
class Settings(BaseSettings):
@@ -46,11 +47,56 @@ class Settings(BaseSettings):
GITEA_DEFAULT_OWNER: str = ""
GITEA_REPO_PREFIX: str = "AI"
# 服务管理配置
SERVICE_MANAGEMENT: Dict[str, Any] = {
"mode": "supervisor", # 服务管理模式local, docker, supervisor
"service_root_dir": "/opt/ai-services",
"supervisor_config_dir": "/etc/supervisor/conf.d",
}
class Config:
env_file = ".env"
case_sensitive = True
extra = "allow" # 允许额外的环境变量
def get_config(self, config_key: str, default: Any = None) -> Any:
"""获取配置,优先级:环境变量 > 数据库 > 文件默认值
Args:
config_key: 配置键
default: 默认值
Returns:
配置值
"""
# 1. 先从环境变量获取
env_key = config_key.upper().replace('.', '_')
env_value = getattr(self, env_key, None)
if env_value is not None:
return env_value
# 2. 从数据库获取
try:
from app.models.database import SessionLocal
from app.models.models import ServiceConfig
db: Session = SessionLocal()
try:
config = db.query(ServiceConfig).filter_by(
config_key=config_key,
status="active"
).first()
if config:
return config.config_value
finally:
db.close()
except Exception as e:
# 数据库连接失败时,返回默认值
print(f"Failed to load config from database: {str(e)}")
# 3. 返回默认值
return default
# 创建全局配置实例
settings = Settings()