mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 01:10:23 +08:00
1d9c77522a
- 删除全部 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)
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
"""配额冷却判定工具。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Any
|
||
|
||
from src.core.logger import logger
|
||
from src.services.scheduling.quota_skipper import is_key_quota_exhausted
|
||
|
||
|
||
def resolve_effective_cooldown_reason(
|
||
*,
|
||
provider_type: str | None,
|
||
key: Any,
|
||
redis_reason: str | None,
|
||
) -> str | None:
|
||
"""返回 Key 的有效冷却原因。
|
||
|
||
规则：
|
||
- Redis 冷却存在时，优先返回 Redis 原因（429/403/quota_exhausted 等）。
|
||
- Redis 冷却不存在时，回退到 upstream_metadata 配额判断：
|
||
若账号级配额耗尽（Codex/Kiro），返回 ``quota_exhausted``。
|
||
"""
|
||
if redis_reason:
|
||
return redis_reason
|
||
|
||
try:
|
||
exhausted, _ = is_key_quota_exhausted(provider_type, key, model_name="")
|
||
except Exception:
|
||
logger.opt(exception=True).debug(
|
||
"quota_cooldown: is_key_quota_exhausted failed for key={}",
|
||
getattr(key, "id", "?"),
|
||
)
|
||
return None
|
||
return "quota_exhausted" if exhausted else None
|