feat: 扩展 Rust gateway 全功能模块,新增 billing/crypto/wallet crate 及完整数据层

- 新增 aether-billing、aether-crypto、aether-wallet 独立 crate
- aether-data 扩展 repository 层:announcements、auth_modules、billing、
  candidate_selection、gemini_file_mappings、global_models、management_tokens、
  oauth_providers、proxy_nodes、quota、users、wallet 等模块
- aether-gateway 新增 api/auth/billing/control/middleware/scheduler/usage/
  video_tasks/hooks/maintenance/model_fetch/provider_transport 等功能模块
- 重构 executor decision 和 gateway state 为模块目录结构
- 新增 gateway router、frontdoor 路由层及对应测试
- Python 侧 API 路由重构,新增 compat/support 模块
- 前端 Logo 组件更新及 Provider 管理页面调整
This commit is contained in:
fawney19
2026-03-31 19:19:04 +08:00
parent b5a0070023
commit ddf18fed9a
690 changed files with 235085 additions and 16299 deletions
+96
View File
@@ -58,6 +58,8 @@ def _build_context(raw_body: bytes, headers: dict[str, str] | None = None) -> Ap
api_key=None,
request_id="req_test",
start_time=0.0,
request_method="POST",
request_path="/v1/messages",
client_ip="127.0.0.1",
user_agent="pytest",
original_headers=headers or {},
@@ -67,6 +69,74 @@ def _build_context(raw_body: bytes, headers: dict[str, str] | None = None) -> Ap
class TestApiRequestContextEnsureJsonBody:
def test_build_prefers_request_state_request_id_over_trace_header(self) -> None:
request = _build_request(headers={"x-trace-id": "trace-frontdoor-123"})
request.state.request_id = "state-rid-001"
context = ApiRequestContext.build(
request=request,
db=None, # type: ignore[arg-type]
user=None,
api_key=None,
raw_body=b"{}",
)
assert context.request_id == "state-rid-001"
assert request.state.request_id == "state-rid-001"
def test_build_prefers_trace_header_for_request_id(self) -> None:
request = _build_request(headers={"x-trace-id": "trace-frontdoor-123"})
context = ApiRequestContext.build(
request=request,
db=None, # type: ignore[arg-type]
user=None,
api_key=None,
raw_body=b"{}",
)
assert context.request_id == "trace-frontdoor-123"
assert request.state.request_id == "trace-frontdoor-123"
def test_build_snapshots_request_method_path_and_path_params(self) -> None:
request = _build_request(headers={"x-trace-id": "trace-frontdoor-123"})
request.scope["method"] = "GET"
request.scope["path"] = "/v1beta/models/gemini-2.5-pro:generateContent"
request.scope["raw_path"] = b"/v1beta/models/gemini-2.5-pro:generateContent"
request.scope["path_params"] = {"model": "gemini-2.5-pro"}
context = ApiRequestContext.build(
request=request,
db=None, # type: ignore[arg-type]
user=None,
api_key=None,
raw_body=b"{}",
)
assert context.request_method == "GET"
assert context.request_path == "/v1beta/models/gemini-2.5-pro:generateContent"
assert context.path_params == {"model": "gemini-2.5-pro"}
def test_build_snapshots_request_runtime_state(self) -> None:
request = _build_request(headers={"x-trace-id": "trace-frontdoor-123"})
request.state.prefetched_balance_remaining = "12.5"
request.state.gateway_execution_path = "public_proxy_after_executor_miss"
request.state.rate_limit_scope = "user"
request.state.tx_committed_by_route = True
context = ApiRequestContext.build(
request=request,
db=None, # type: ignore[arg-type]
user=None,
api_key=None,
raw_body=b"{}",
)
assert context.prefetched_balance_remaining == 12.5
assert context.gateway_execution_path == "public_proxy_after_executor_miss"
assert context.rate_limit_scope == "user"
assert context.tx_committed_by_route is True
def test_decompresses_gzip_body(self) -> None:
payload = {"message": "hello", "count": 2}
raw_body = gzip.compress(json.dumps(payload).encode("utf-8"))
@@ -88,6 +158,7 @@ class TestApiRequestContextEnsureJsonBody:
def test_build_records_client_encoding_preferences(self) -> None:
request = _build_request(
headers={
"content-type": "application/json",
"content-encoding": "gzip",
"accept-encoding": "gzip, deflate",
}
@@ -102,6 +173,31 @@ class TestApiRequestContextEnsureJsonBody:
assert context.client_content_encoding == "gzip"
assert context.client_accept_encoding == "gzip, deflate"
assert context.request_content_type == "application/json"
def test_build_records_perf_only_when_payload_not_empty(self) -> None:
request = _build_request(headers={"x-trace-id": "trace-frontdoor-123"})
request.state.perf_metrics = {}
context = ApiRequestContext.build(
request=request,
db=None, # type: ignore[arg-type]
user=None,
api_key=None,
raw_body=b"{}",
)
assert "perf" not in context.extra
request_with_perf = _build_request(headers={"x-trace-id": "trace-frontdoor-456"})
request_with_perf.state.perf_metrics = {"pipeline": {"auth_ms": 3}}
context_with_perf = ApiRequestContext.build(
request=request_with_perf,
db=None, # type: ignore[arg-type]
user=None,
api_key=None,
raw_body=b"{}",
)
assert context_with_perf.extra["perf"] == {"pipeline": {"auth_ms": 3}}
assert context_with_perf.perf_metrics == {"pipeline": {"auth_ms": 3}}
@pytest.mark.asyncio
async def test_ensure_json_body_async_loads_body_lazily(self) -> None: