refactor: 移除 Python 后端源码,全面迁移至 Rust gateway 架构

- 删除全部 Python 源码 (src/) 及 Alembic 迁移脚本,归档至 _deprecated_py_src/
- 重构 Rust gateway ai_pipeline: 拆分 planner/finalize 模块,新增 contracts/adaptation 层
- 重组 handlers 模块为 admin/public/proxy/internal/shared 子模块结构
- 新增 executor 模块,引入 Rust 原生数据库迁移 (aether-data/migrations)
- 简化 CI/Docker 构建流程,移除 base image 二级构建,统一为单一 app image
- 移除 Python 相关基础设施文件 (entrypoint.sh, gunicorn_conf.py, Dockerfile.base)
This commit is contained in:
fawney19
2026-04-03 16:26:16 +08:00
parent 8f26e1a31f
commit 1d9c77522a
868 changed files with 1735 additions and 2433 deletions

View File

@@ -0,0 +1,82 @@
"""
访问限制检查
从 CacheAwareScheduler 提取的 ApiKey + User 访问限制合并逻辑。
"""
from __future__ import annotations
from typing import Any
from src.core.logger import logger
from src.core.model_permissions import merge_allowed_models
from src.models.database import ApiKey
def get_effective_restrictions(user_api_key: ApiKey | None) -> dict[str, Any]:
"""
获取有效的访问限制(合并 ApiKey 和 User 的限制)
逻辑:
- 如果 ApiKey 和 User 都有限制,取交集
- 如果只有一方有限制,使用该方的限制
- 如果都没有限制,返回 None表示不限制
Args:
user_api_key: 用户 API Key 对象(可能包含 user relationship
Returns:
包含 allowed_providers, allowed_models, allowed_api_formats 的字典
"""
result: dict[str, Any] = {
"allowed_providers": None,
"allowed_models": None,
"allowed_api_formats": None,
}
if not user_api_key:
return result
# 获取 User 的限制
# 注意:这里可能触发 lazy loading需要确保 session 仍然有效
try:
user = user_api_key.user if hasattr(user_api_key, "user") else None
except Exception as e:
logger.warning("无法加载 ApiKey 关联的 User: {},仅使用 ApiKey 级别的限制", e)
user = None
# 调试日志
logger.debug(
"[_get_effective_restrictions] ApiKey={}..., User={}..., "
"ApiKey.allowed_models={}, User.allowed_models={}",
user_api_key.id[:8],
user.id[:8] if user else "None",
user_api_key.allowed_models,
user.allowed_models if user else "N/A",
)
# 合并 allowed_providers
result["allowed_providers"] = merge_restriction_sets(
user_api_key.allowed_providers, user.allowed_providers if user else None
)
# 合并 allowed_models取交集
result["allowed_models"] = merge_allowed_models(
user_api_key.allowed_models, user.allowed_models if user else None
)
# 合并 allowed_api_formats
result["allowed_api_formats"] = merge_restriction_sets(
user_api_key.allowed_api_formats, user.allowed_api_formats if user else None
)
return result
def merge_restriction_sets(key_restriction: Any, user_restriction: Any) -> set[Any] | None:
"""合并两个限制列表,取交集;任一方为空则使用另一方;均空返回 None"""
key_set = set(key_restriction) if key_restriction else None
user_set = set(user_restriction) if user_restriction else None
if key_set and user_set:
return key_set & user_set
return key_set or user_set