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)
134 lines
4.4 KiB
Python
134 lines
4.4 KiB
Python
"""Kiro provider plugin — unified registration entry.
|
|
|
|
Kiro upstream looks like Claude CLI (Bearer token) from the outside, but uses a
|
|
custom wire protocol:
|
|
- Request: Claude Messages API -> Kiro generateAssistantResponse envelope
|
|
- Response (stream): AWS Event Stream (binary) -> Claude SSE events
|
|
|
|
This plugin registers:
|
|
- Envelope
|
|
- Transport hook (dynamic region base_url)
|
|
- Model fetcher (fixed model catalog — Kiro has no /v1/models endpoint)
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
from urllib.parse import urlencode
|
|
|
|
from src.services.provider.adapters.kiro.constants import DEFAULT_REGION
|
|
from src.services.provider.adapters.kiro.context import get_kiro_request_context
|
|
from src.services.provider.adapters.kiro.models.credentials import KiroAuthConfig
|
|
from src.services.provider.adapters.kiro.request import (
|
|
build_kiro_generate_assistant_url,
|
|
resolve_kiro_base_url,
|
|
)
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Preset model catalog
|
|
# ---------------------------------------------------------------------------
|
|
# Kiro upstream has no /v1/models endpoint. We use the unified preset models
|
|
# registry from preset_models.py.
|
|
from src.services.provider.preset_models import create_preset_models_fetcher
|
|
from src.services.provider.request_context import set_selected_base_url
|
|
|
|
fetch_models_kiro = create_preset_models_fetcher("kiro")
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Transport hook
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def build_kiro_url(
|
|
endpoint: Any,
|
|
*,
|
|
is_stream: bool,
|
|
effective_query_params: dict[str, Any],
|
|
**_kwargs: Any,
|
|
) -> str:
|
|
"""Build Kiro generateAssistantResponse URL.
|
|
|
|
Endpoint base_url may contain a `{region}` placeholder. The actual region is
|
|
resolved from per-request context (set by the envelope).
|
|
"""
|
|
_ = is_stream
|
|
|
|
ctx = get_kiro_request_context()
|
|
region = (ctx.region if ctx else "") or DEFAULT_REGION
|
|
raw_base = str(getattr(endpoint, "base_url", "") or "").rstrip("/")
|
|
cfg = KiroAuthConfig(api_region=region)
|
|
base = resolve_kiro_base_url(raw_base, cfg=cfg)
|
|
set_selected_base_url(base)
|
|
|
|
url = build_kiro_generate_assistant_url(raw_base, cfg=cfg)
|
|
|
|
if effective_query_params:
|
|
query_string = urlencode(effective_query_params, doseq=True)
|
|
if query_string:
|
|
url = f"{url}?{query_string}"
|
|
|
|
return url
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Export builder
|
|
# ---------------------------------------------------------------------------
|
|
|
|
_KIRO_SKIP_KEYS = frozenset(
|
|
{
|
|
"access_token",
|
|
"expires_at",
|
|
"updated_at",
|
|
}
|
|
)
|
|
|
|
|
|
def kiro_export_builder(
|
|
auth_config: dict[str, Any],
|
|
upstream_metadata: dict[str, Any] | None,
|
|
) -> dict[str, Any]:
|
|
"""Kiro 导出：保留 auth_method / refresh_token / machine_id / profile_arn 等，
|
|
IdC 模式额外保留 client_id / client_secret / region。"""
|
|
data = {
|
|
k: v
|
|
for k, v in auth_config.items()
|
|
if k not in _KIRO_SKIP_KEYS and v is not None and v != ""
|
|
}
|
|
# email 可能仅在 upstream_metadata.kiro 中
|
|
if not data.get("email"):
|
|
kiro_meta = (upstream_metadata or {}).get("kiro") or {}
|
|
if kiro_meta.get("email"):
|
|
data["email"] = kiro_meta["email"]
|
|
return data
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Registration
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def register_all() -> None:
|
|
"""Register all Kiro hooks into shared registries."""
|
|
|
|
from src.services.model.upstream_fetcher import UpstreamModelsFetcherRegistry
|
|
from src.services.provider.adapters.kiro.envelope import kiro_envelope
|
|
from src.services.provider.envelope import register_envelope
|
|
from src.services.provider.export import register_export_builder
|
|
from src.services.provider.transport import register_transport_hook
|
|
|
|
register_envelope("kiro", "claude:cli", kiro_envelope)
|
|
register_envelope("kiro", "", kiro_envelope)
|
|
|
|
register_transport_hook("kiro", "claude:cli", build_kiro_url)
|
|
|
|
register_export_builder("kiro", kiro_export_builder)
|
|
|
|
UpstreamModelsFetcherRegistry.register(
|
|
provider_types=["kiro"],
|
|
fetcher=fetch_models_kiro,
|
|
)
|
|
|
|
|
|
__all__ = ["build_kiro_url", "fetch_models_kiro", "kiro_export_builder", "register_all"]
|