"""Provider→适配器工厂(ARCH §4.2/§4.3)。 `build_gateway_for_tier`(apps/api)据 DB `tier_routing` 的 primary + fallback 逐 provider 建适配器进 `adapters` dict——本工厂按 provider 名分派到对应适配器类,构造其真实客户端。 - `anthropic` → `AnthropicAdapter`(懒构建 `AsyncAnthropic`,不在模块顶层硬 import SDK)。 - `gemini`/`google` → `GeminiAdapter`(懒构建 `google.genai.Client`)。 - `kimi-code` → `KimiCodeAdapter`(OAuth access token 当 api_key + coding base + 伪造头)。 - `kimi-code-key` → `KimiCodeKeyAdapter`(静态 Console Key + coding base,**无伪造头**)。 - 其余(deepseek/kimi/qwen/glm/openai)→ `OpenAICompatAdapter(provider, AsyncOpenAI(...))`。 适配器本身仍接注入客户端(测试零联网);本工厂从 api_key 构造**真实**客户端, 故单测只断分派逻辑(按 provider 名选哪个适配器类),不联网。 """ from __future__ import annotations from typing import cast from openai import AsyncOpenAI from .adapters.anthropic import AnthropicAdapter, AnthropicClient from .adapters.base import ProviderAdapter from .adapters.gemini import GeminiAdapter, GeminiClient from .adapters.kimi_code import KIMI_CODE_PROVIDER, KimiCodeAdapter, build_kimi_code_client from .adapters.kimi_code_key import ( KIMI_CODE_KEY_PROVIDER, KimiCodeKeyAdapter, build_kimi_code_key_client, ) from .adapters.openai_compat import OpenAICompatAdapter # 走专用 SDK 适配器的 provider 名(其余一律走 OpenAI 兼容)。 _ANTHROPIC_PROVIDERS = frozenset({"anthropic"}) _GEMINI_PROVIDERS = frozenset({"gemini", "google"}) def _build_anthropic(provider: str, *, api_key: str, base_url: str | None) -> ProviderAdapter: # 懒 import:SDK 仅在真实接线 Anthropic 时才需要(单测不触发此分支)。 from anthropic import AsyncAnthropic kwargs: dict[str, object] = {"api_key": api_key} if base_url is not None: kwargs["base_url"] = base_url client = AsyncAnthropic(**kwargs) # type: ignore[arg-type] # `AsyncAnthropic` 满足 `AnthropicClient` Protocol(read-only `messages` 不影响鸭子用法); # cast 跨过 Protocol settable-attr 静态约束(适配器仅读取该属性)。 return AnthropicAdapter(provider, cast("AnthropicClient", client)) def _build_gemini(provider: str, *, api_key: str, base_url: str | None) -> ProviderAdapter: # 懒 import:`google-genai` 仅在真实接线 Gemini 时才需要。 from google import genai client = genai.Client(api_key=api_key) # `genai.Client` 满足 `GeminiClient` Protocol(用其 `.aio.models`);cast 跨过 # Protocol settable-attr 静态约束(适配器仅读取 `.aio`)。 return GeminiAdapter(provider, cast("GeminiClient", client)) def build_adapter(provider: str, *, api_key: str, base_url: str | None = None) -> ProviderAdapter: """按 provider 名构造对应适配器,从 api_key(+可选 base_url)建真实客户端。 `build_gateway_for_tier`(apps/api)逐 provider 调用本工厂,组装回退链的 adapters。 """ if provider == KIMI_CODE_PROVIDER: # Kimi 订阅 plan(OAuth):api_key = OAuth access token;coding base + 伪造头(缺头 403)。 return KimiCodeAdapter(build_kimi_code_client(api_key, base_url=base_url)) if provider == KIMI_CODE_KEY_PROVIDER: # Kimi 订阅 plan(静态 Console Key,ToS 合规):coding base + 纯 bearer,**无伪造头**; # 结构化走 JSON 模式(thinking 兼容)。 return KimiCodeKeyAdapter(build_kimi_code_key_client(api_key, base_url=base_url)) if provider in _ANTHROPIC_PROVIDERS: return _build_anthropic(provider, api_key=api_key, base_url=base_url) if provider in _GEMINI_PROVIDERS: return _build_gemini(provider, api_key=api_key, base_url=base_url) return OpenAICompatAdapter(provider, AsyncOpenAI(api_key=api_key, base_url=base_url))