coding-proxy 0.1.0__py3-none-any.whl

coding/proxy/compat/session_store.py

@@ -0,0 +1,137 @@
+"""兼容层会话状态持久化.
+
+数据类型 ``CompatSessionRecord`` 已迁移至 :mod:`coding.proxy.model.compat`。
+本文件保留 ``CompatSessionStore`` 持久化管理器，类型通过 re-export 提供。
+
+.. deprecated::
+    未来版本将移除类型 re-export，请直接从 :mod:`coding.proxy.model.compat` 导入。
+"""
+
+from __future__ import annotations
+
+import json
+import time
+from pathlib import Path
+from typing import Any
+
+import aiosqlite
+
+# noqa: F401
+from ..model.compat import CompatSessionRecord
+
+
+_CREATE_TABLE = """
+CREATE TABLE IF NOT EXISTS compat_session (
+    session_key TEXT PRIMARY KEY,
+    trace_id TEXT NOT NULL DEFAULT '',
+    tool_call_map_json TEXT NOT NULL DEFAULT '{}',
+    thought_signature_map_json TEXT NOT NULL DEFAULT '{}',
+    provider_state_json TEXT NOT NULL DEFAULT '{}',
+    state_version INTEGER NOT NULL DEFAULT 1,
+    updated_at_unix INTEGER NOT NULL DEFAULT 0
+);
+CREATE INDEX IF NOT EXISTS idx_compat_session_updated_at ON compat_session(updated_at_unix);
+"""
+
+
+class CompatSessionStore:
+    """兼容层会话状态 SQLite 持久化存储."""
+
+    def __init__(self, db_path: Path, ttl_seconds: int = 86400) -> None:
+        self._db_path = db_path
+        self._ttl_seconds = ttl_seconds
+        self._db: aiosqlite.Connection | None = None
+
+    async def init(self) -> None:
+        self._db_path.parent.mkdir(parents=True, exist_ok=True)
+        self._db = await aiosqlite.connect(str(self._db_path))
+        self._db.row_factory = aiosqlite.Row
+        await self._db.execute("PRAGMA journal_mode=WAL")
+        await self._db.executescript(_CREATE_TABLE)
+        await self._purge_expired()
+        await self._db.commit()
+
+    async def get(self, session_key: str) -> CompatSessionRecord | None:
+        if not self._db:
+            return None
+        cursor = await self._db.execute(
+            """SELECT session_key, trace_id, tool_call_map_json, thought_signature_map_json,
+                      provider_state_json, state_version, updated_at_unix
+               FROM compat_session WHERE session_key = ?""",
+            (session_key,),
+        )
+        row = await cursor.fetchone()
+        if row is None:
+            return None
+        record = CompatSessionRecord(
+            session_key=row["session_key"],
+            trace_id=row["trace_id"],
+            tool_call_map=_loads_dict(row["tool_call_map_json"]),
+            thought_signature_map=_loads_dict(row["thought_signature_map_json"]),
+            provider_state=_loads_dict(row["provider_state_json"]),
+            state_version=row["state_version"],
+            updated_at_unix=row["updated_at_unix"],
+        )
+        if self._is_expired(record.updated_at_unix):
+            await self.delete(session_key)
+            return None
+        return record
+
+    async def upsert(self, record: CompatSessionRecord) -> None:
+        if not self._db:
+            return
+        updated_at = int(time.time())
+        await self._db.execute(
+            """INSERT INTO compat_session (
+                   session_key, trace_id, tool_call_map_json, thought_signature_map_json,
+                   provider_state_json, state_version, updated_at_unix
+               ) VALUES (?, ?, ?, ?, ?, ?, ?)
+               ON CONFLICT(session_key) DO UPDATE SET
+                   trace_id=excluded.trace_id,
+                   tool_call_map_json=excluded.tool_call_map_json,
+                   thought_signature_map_json=excluded.thought_signature_map_json,
+                   provider_state_json=excluded.provider_state_json,
+                   state_version=excluded.state_version,
+                   updated_at_unix=excluded.updated_at_unix""",
+            (
+                record.session_key,
+                record.trace_id,
+                json.dumps(record.tool_call_map, ensure_ascii=False, sort_keys=True),
+                json.dumps(record.thought_signature_map, ensure_ascii=False, sort_keys=True),
+                json.dumps(record.provider_state, ensure_ascii=False, sort_keys=True),
+                record.state_version,
+                updated_at,
+            ),
+        )
+        await self._db.commit()
+
+    async def delete(self, session_key: str) -> None:
+        if not self._db:
+            return
+        await self._db.execute("DELETE FROM compat_session WHERE session_key = ?", (session_key,))
+        await self._db.commit()
+
+    async def close(self) -> None:
+        if self._db:
+            await self._db.close()
+            self._db = None
+
+    async def _purge_expired(self) -> None:
+        if not self._db:
+            return
+        threshold = int(time.time()) - self._ttl_seconds
+        await self._db.execute(
+            "DELETE FROM compat_session WHERE updated_at_unix > 0 AND updated_at_unix < ?",
+            (threshold,),
+        )
+
+    def _is_expired(self, updated_at_unix: int) -> bool:
+        return updated_at_unix > 0 and (int(time.time()) - updated_at_unix) > self._ttl_seconds
+
+
+def _loads_dict(raw: str) -> dict[str, Any]:
+    try:
+        value = json.loads(raw)
+    except (json.JSONDecodeError, TypeError):
+        return {}
+    return value if isinstance(value, dict) else {}

coding/proxy/config/__init__.py

@@ -0,0 +1,6 @@
+"""配置模块."""
+
+from .loader import load_config
+from .schema import ProxyConfig
+
+__all__ = ["load_config", "ProxyConfig"]

coding/proxy/config/auth_schema.py

@@ -0,0 +1,24 @@
+"""OAuth 认证配置模型."""
+
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class AuthConfig(BaseModel):
+    """OAuth 登录配置.
+
+    .. note::
+        各 Provider 的硬编码默认值（google.py / github.py）应与此配置保持同步。
+        此配置作为运行时注入值的权威来源（Single Source of Truth）。
+    """
+
+    github_client_id: str = "Iv1.b507a08c87ecfe98"
+    google_client_id: str = (
+        "1071006060591-tmhssin2h21lcre235vtolojh4g403ep"
+        ".apps.googleusercontent.com"
+    )
+    google_client_secret: str = "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"
+    token_store_path: str = "~/.coding-proxy/tokens.json"
+
+__all__ = ["AuthConfig"]

coding/proxy/config/loader.py

@@ -0,0 +1,139 @@
+"""YAML 配置加载 + 环境变量展开 + 示例配置深度合并."""
+
+from __future__ import annotations
+
+import logging
+import os
+import re
+from pathlib import Path
+
+import yaml
+
+from .schema import ProxyConfig
+
+logger = logging.getLogger(__name__)
+
+_ENV_VAR_RE = re.compile(r"\$\{([^}]+)\}")
+
+# ── Legacy flat 格式字段集合（用于检测旧配置，避免与 example vendors 冲突） ──
+_LEGACY_FLAT_KEYS: frozenset[str] = frozenset({
+    "primary", "copilot", "antigravity", "fallback",
+    "circuit_breaker", "copilot_circuit_breaker", "antigravity_circuit_breaker",
+    "quota_guard", "copilot_quota_guard", "antigravity_quota_guard",
+})
+
+
+def _expand_env(value: str) -> str:
+    """将 ${VAR} 替换为环境变量值."""
+    def _replacer(match: re.Match) -> str:
+        var_name = match.group(1)
+        return os.environ.get(var_name, "")
+    return _ENV_VAR_RE.sub(_replacer, value)
+
+
+def _expand_env_recursive(obj):
+    """递归展开字典中的环境变量."""
+    if isinstance(obj, dict):
+        return {k: _expand_env_recursive(v) for k, v in obj.items()}
+    if isinstance(obj, list):
+        return [_expand_env_recursive(v) for v in obj]
+    if isinstance(obj, str):
+        return _expand_env(obj)
+    return obj
+
+
+def _deep_merge(defaults: dict, override: dict) -> dict:
+    """深度合并两个字典.
+
+    合并策略：
+    - dict + dict → 递归合并子键（支持部分覆盖嵌套配置）
+    - list         → override 完整替换 default（有序集合，顺序敏感）
+    - 标量         → override 替换 default
+    - override 中不存在于 defaults 的新键直接添加
+
+    Args:
+        defaults: 基础字典（通常来自 config.example.yaml）
+        override: 覆盖字典（来自用户配置文件）
+
+    Returns:
+        合并后的新字典
+    """
+    result = dict(defaults)
+    for key, ov in override.items():
+        if key not in result:
+            result[key] = ov
+        elif isinstance(result.get(key), dict) and isinstance(ov, dict):
+            result[key] = _deep_merge(result[key], ov)
+        else:
+            result[key] = ov
+    return result
+
+
+def _get_example_config_path() -> Path | None:
+    """定位 config.example.yaml 文件路径.
+
+    搜索策略：从 loader.py 所在目录向上回溯到项目根目录查找。
+    路径链：config/ → proxy/ → coding/ → src/ → project_root
+
+    Returns:
+        文件路径对象，未找到时返回 None（触发降级至 Pydantic 默认值）
+    """
+    current = Path(__file__).resolve().parent
+    # 先检查当前层，再逐级向上（共检查 5 层：config/ ~ project_root/）
+    for _ in range(5):
+        candidate = current / "config.example.yaml"
+        if candidate.is_file():
+            return candidate
+        current = current.parent
+    logger.debug("未找到 config.example.yaml，将使用 Pydantic 默认值")
+    return None
+
+
+def load_config(path: Path | None = None) -> ProxyConfig:
+    """加载配置文件，以 config.example.yaml 为基础进行深度合并.
+
+    加载优先级（低→高）：
+    1. config.example.yaml 内置完整默认值
+    2. 用户配置文件（CWD/config.yaml > ~/.coding-proxy/config.yaml > -c 指定路径）
+
+    环境变量展开（${VAR}）在深度合并之后执行，确保用户可通过环境变量覆盖任意字段。
+    """
+    # ── 第 1 步：确定并加载用户配置 ─────────────────────────────
+    user_raw: dict = {}
+    if path is None:
+        candidates = [
+            Path("config.yaml"),
+            Path("~/.coding-proxy/config.yaml").expanduser(),
+        ]
+        for candidate in candidates:
+            if candidate.exists():
+                path = candidate
+                break
+
+    if path and path.exists():
+        with open(path) as f:
+            user_raw = yaml.safe_load(f) or {}
+
+    # ── 第 2 步：加载示例默认配置 ─────────────────────────────
+    example_path = _get_example_config_path()
+    if example_path is None:
+        # 降级：无示例文件时使用纯 Pydantic 默认值（向后兼容）
+        expanded = _expand_env_recursive(user_raw)
+        return ProxyConfig(**expanded)
+
+    with open(example_path) as f:
+        defaults = yaml.safe_load(f) or {}
+
+    # ── Legacy 兼容：旧 flat 格式用户配置不应继承 example 的 vendors ──
+    # 当用户使用 legacy 字段时，移除 defaults 中的 vendors，
+    # 让 ProxyConfig._migrate_legacy_fields 迁移器正常接管 vendors 构建
+    if any(k in user_raw for k in _LEGACY_FLAT_KEYS):
+        defaults.pop("vendors", None)
+
+    # ── 第 3 步：深度合并 ─────────────────────────────────────
+    merged = _deep_merge(defaults, user_raw)
+
+    # ── 第 4 步：环境变量展开（必须在合并之后） ────────────────
+    expanded = _expand_env_recursive(merged)
+
+    return ProxyConfig(**expanded)

coding/proxy/config/resiliency.py

@@ -0,0 +1,46 @@
+"""弹性设施配置模型（熔断、重试、故障转移、配额守卫）."""
+
+from __future__ import annotations
+
+from pydantic import BaseModel, Field
+
+
+class CircuitBreakerConfig(BaseModel):
+    failure_threshold: int = 3
+    recovery_timeout_seconds: int = 300
+    success_threshold: int = 2
+    max_recovery_seconds: int = 3600
+
+
+class RetryConfig(BaseModel):
+    """传输层重试配置."""
+
+    max_retries: int = 2
+    initial_delay_ms: int = 500
+    max_delay_ms: int = 5000
+    backoff_multiplier: float = 2.0
+    jitter: bool = True
+
+
+class FailoverConfig(BaseModel):
+    status_codes: list[int] = Field(
+        default=[429, 403, 503, 500],
+    )
+    error_types: list[str] = Field(
+        default=["rate_limit_error", "overloaded_error", "api_error"],
+    )
+    error_message_patterns: list[str] = Field(
+        default=["quota", "limit exceeded", "usage cap", "capacity"],
+    )
+
+
+class QuotaGuardConfig(BaseModel):
+    enabled: bool = False
+    token_budget: int = 0
+    window_hours: float = 5.0
+    threshold_percent: float = 99.0
+    probe_interval_seconds: int = 300
+
+__all__ = [
+    "CircuitBreakerConfig", "RetryConfig", "FailoverConfig", "QuotaGuardConfig",
+]

coding/proxy/config/routing.py

@@ -0,0 +1,279 @@
+"""路由层配置模型（供应商类型、Vendor、模型映射、定价）."""
+
+from __future__ import annotations
+
+import logging
+import re
+from typing import Annotated, Any, Literal
+
+from pydantic import BaseModel, BeforeValidator, Field, PrivateAttr, model_validator
+
+from .resiliency import CircuitBreakerConfig, QuotaGuardConfig, RetryConfig
+
+# ── 价格字段解析（$ / ¥ 前缀支持） ──────────────────────────
+
+_PRICE_RE = re.compile(r"^([$\u00a5])\s*(.+)$")
+
+
+def _price_to_float(v: Any) -> float:
+    """Pydantic BeforeValidator: 将 $/¥ 前缀字符串 / 纯数字统一转为 float."""
+    if isinstance(v, (int, float)):
+        return float(v)
+    if isinstance(v, str):
+        m = _PRICE_RE.match(v.strip())
+        if m:
+            return float(m.group(2))
+        return float(v)
+    return float(v)
+
+
+def _detect_currency(v: Any) -> str | None:
+    """从原始值中检测币种前缀. 返回 ``'USD'``/``'CNY'``/``None``."""
+    if isinstance(v, str):
+        m = _PRICE_RE.match(v.strip())
+        if m:
+            return "USD" if m.group(1) == "$" else "CNY"
+    return None
+
+
+PriceField = Annotated[float, BeforeValidator(_price_to_float)]
+
+logger = logging.getLogger(__name__)
+
+# ── 供应商专属字段分组映射 ──────────────────────────────────────
+# 每个 vendor 类型对应其专属字段集合，用于 VendorConfig 的语义标注与校验
+
+_COPILOT_FIELDS: frozenset[str] = frozenset({
+    "github_token", "account_type", "token_url", "models_cache_ttl_seconds",
+})
+_ANTIGRAVITY_FIELDS: frozenset[str] = frozenset({
+    "client_id", "client_secret", "refresh_token", "model_endpoint",
+})
+_ZHIPU_FIELDS: frozenset[str] = frozenset({"api_key",})
+
+_VENDOR_EXCLUSIVE_FIELDS: dict[str, frozenset[str]] = {
+    "copilot": _COPILOT_FIELDS,
+    "antigravity": _ANTIGRAVITY_FIELDS,
+    "zhipu": _ZHIPU_FIELDS,
+}
+
+VendorType = Literal["anthropic", "copilot", "antigravity", "zhipu"]
+
+
+class ModelMappingRule(BaseModel):
+    pattern: str
+    target: str
+    is_regex: bool = False
+    vendors: list[str] = Field(default_factory=list)
+
+
+class ModelPricingEntry(BaseModel):
+    """单个模型的定价配置（支持 $ / ¥ 前缀指定币种）.
+
+    用法示例::
+
+        input_cost_per_mtok: $3.0       # USD
+        output_cost_per_mtok: \\u00a53.2  # CNY (¥)
+        cache_read_cost_per_mtok: 0.5   # 无前缀，默认 USD
+
+    向后兼容：不带前缀的纯数字默认视为 USD。
+    """
+
+    model_config = {"extra": "allow"}
+
+    vendor: str                            # 供应商名称（对应 usage 表"供应商"列）
+    model: str                              # 实际模型名（对应 usage 表"实际模型"列）
+    input_cost_per_mtok: PriceField = 0.0   # 输入 Token 单价
+    output_cost_per_mtok: PriceField = 0.0  # 输出 Token 单价
+    cache_write_cost_per_mtok: PriceField = 0.0  # 缓存创建 Token 单价
+    cache_read_cost_per_mtok: PriceField = 0.0   # 缓存读取 Token 单价
+
+    # ── 内部状态：币种信息（不参与序列化） ───────────────────
+    _currency: str = PrivateAttr(default="USD")
+
+    # ── 币种一致性校验与提取 ──────────────────────────────
+
+    @model_validator(mode="before")
+    @classmethod
+    def _check_currency_consistency(cls, data: Any) -> Any:
+        """校验同一 entry 内所有非零价格的币种一致性，并提取币种."""
+        if not isinstance(data, dict):
+            return data
+
+        price_field_names = [
+            "input_cost_per_mtok", "output_cost_per_mtok",
+            "cache_write_cost_per_mtok", "cache_read_cost_per_mtok",
+        ]
+        currencies: set[str] = set()
+
+        for name in price_field_names:
+            raw = data.get(name)
+            if raw is None or raw == 0 or raw == 0.0:
+                continue
+            detected = _detect_currency(raw)
+            if detected:
+                currencies.add(detected)
+
+        if len(currencies) > 1:
+            vendor = data.get("vendor", "?")
+            model = data.get("model", "?")
+            raise ValueError(
+                f"PricingEntry(vendor={vendor!r}, model={model!r}): "
+                f"检测到混合币种 {sorted(currencies)}，"
+                "同一模型的所有单价必须使用相同币种 ($ 或 ¥)"
+            )
+
+        # 将检测到的币种暂存到临时键（mode=after 中消费后清理）
+        if currencies:
+            data["__detected_currency__"] = currencies.pop()
+
+        return data
+
+    @model_validator(mode="after")
+    def _capture_currency(self) -> "ModelPricingEntry":
+        """将 mode=before 检测到的币种转移到 PrivateAttr，清理临时键."""
+        detected = getattr(self, "__detected_currency__", None)
+        if detected:
+            self._currency = detected
+            # 从 __pydantic_extra__ 中移除临时键，避免序列化泄露
+            if hasattr(self, "__pydantic_extra__") and "__detected_currency__" in self.__pydantic_extra__:
+                del self.__pydantic_extra__["__detected_currency__"]
+            else:
+                # 回退：直接删除实例属性
+                try:
+                    object.__delattr__(self, "__detected_currency__")
+                except AttributeError:
+                    pass
+        return self
+
+    @model_validator(mode="after")
+    def _validate_non_negative(self) -> "ModelPricingEntry":
+        """校验所有价格字段非负."""
+        for name in (
+            "input_cost_per_mtok", "output_cost_per_mtok",
+            "cache_write_cost_per_mtok", "cache_read_cost_per_mtok",
+        ):
+            val = getattr(self, name)
+            if val < 0:
+                raise ValueError(
+                    f"PricingEntry(vendor={self.vendor!r}, model={self.model!r}): "
+                    f"{name} 不能为负数（当前值: {val}）"
+                )
+        return self
+
+    @property
+    def currency(self) -> str:
+        """本条目的币种代码 (``'USD'`` / ``'CNY'``).
+
+        从 PrivateAttr 读取，若无显式币种前缀则默认 ``'USD'``。
+        """
+        return self._currency
+
+
+class VendorConfig(BaseModel):
+    """单个 Vendor 的统一配置（支持所有供应商类型）.
+
+    .. note::
+        当 ``tiers`` 未配置时，vendors 列表顺序即为优先级；
+        配置了 ``tiers`` 后，优先级由其显式指定。
+
+    无 circuit_breaker 的 Vendor 为终端层（不触发故障转移）。
+
+    各供应商类型的专属字段已通过 ``Field(description=...)`` 标注适用范围，
+    非当前 vendor 类型的专属字段在验证阶段会发出 warning 日志。
+    """
+
+    vendor: VendorType
+
+    # ── 通用字段（所有供应商共用） ──────────────────────────────
+    enabled: bool = True
+    base_url: str = Field(
+        default="",
+        description="供应商 API 基础 URL；留空时使用各供应商默认值",
+    )
+    timeout_ms: int = Field(
+        default=300000,
+        description="请求超时时间（毫秒），适用于所有供应商",
+    )
+
+    # ── Copilot 专属字段 ─────────────────────────────────────────────
+    github_token: str = Field(
+        default="",
+        description="[copilot] GitHub Personal Access Token 或 OAuth Token",
+    )
+    account_type: str = Field(
+        default="individual",
+        description="[copilot] Copilot 账户类型：individual / business / enterprise",
+    )
+    token_url: str = Field(
+        default="https://api.github.com/copilot_internal/v2/token",
+        description="[copilot] Copilot Token 交换端点 URL",
+    )
+    models_cache_ttl_seconds: int = Field(
+        default=300,
+        description="[copilot] 模型列表缓存 TTL（秒）",
+    )
+
+    # ── Antigravity 专属字段 ────────────────────────────────────────
+    client_id: str = Field(
+        default="",
+        description="[antigravity] Google OAuth2 Client ID",
+    )
+    client_secret: str = Field(
+        default="",
+        description="[antigravity] Google OAuth2 Client Secret",
+    )
+    refresh_token: str = Field(
+        default="",
+        description="[antigravity] Google OAuth2 Refresh Token",
+    )
+    model_endpoint: str = Field(
+        default="models/claude-sonnet-4-20250514",
+        description="[antigravity] Antigravity 模型端点路径",
+    )
+
+    # ── Zhipu 专属字段 ────────────────────────────────────────────
+    api_key: str = Field(
+        default="",
+        description="[zhipu] 智谱 GLM API Key",
+    )
+
+    # ── 弹性配置 ──────────────────────────────────────────────
+    circuit_breaker: CircuitBreakerConfig | None = Field(
+        default=None,
+        description="熔断器配置；None 表示终端层（不触发故障转移）",
+    )
+    retry: RetryConfig = Field(default_factory=RetryConfig)
+    quota_guard: QuotaGuardConfig = Field(default_factory=QuotaGuardConfig)
+    weekly_quota_guard: QuotaGuardConfig = Field(default_factory=QuotaGuardConfig)
+
+    @model_validator(mode="after")
+    def _warn_irrelevant_fields(self) -> "VendorConfig":
+        """对非当前 vendor 类型的非空专属字段发出 warning."""
+        exclusive = _VENDOR_EXCLUSIVE_FIELDS.get(self.vendor)
+        if not exclusive:
+            return self
+        for vendor_type, fields in _VENDOR_EXCLUSIVE_FIELDS.items():
+            if vendor_type == self.vendor:
+                continue
+            for field_name in fields:
+                value = getattr(self, field_name, None)
+                if value and value != getattr(VendorConfig.model_fields[field_name], "default", None):
+                    logger.warning(
+                        "VendorConfig(vendor=%s): 字段 %s 属于 %s 供应商，当前值将被忽略",
+                        self.vendor, field_name, vendor_type,
+                    )
+        return self
+
+# ── 向后兼容别名（v2 移除）────────────────────────────────────
+
+TierConfig = VendorConfig
+BackendType = VendorType
+_BACKEND_EXCLUSIVE_FIELDS = _VENDOR_EXCLUSIVE_FIELDS
+
+__all__ = [
+    "VendorType", "VendorConfig", "ModelMappingRule", "ModelPricingEntry",
+    "TierConfig", "BackendType",  # 向后兼容别名
+    "_COPILOT_FIELDS", "_ANTIGRAVITY_FIELDS", "_ZHIPU_FIELDS",
+    "_VENDOR_EXCLUSIVE_FIELDS", "_BACKEND_EXCLUSIVE_FIELDS",
+]