Aether/_deprecated_py_src/services/provider/adapters/antigravity/constants.py

"""Antigravity 全局常量定义。

注意：这里的 PROVIDER_TYPE 指的是 Provider.provider_type（用于路由与特判），
不是 endpoint signature（family:kind）。
"""

from __future__ import annotations

import platform
import re
import threading
import uuid

# ============== API 端点 ==============
# 唯一定义在 core 层，此处 re-export 保持向后兼容
from src.core.provider_templates.fixed_providers import ANTIGRAVITY_PROD_URL as PROD_BASE_URL

DAILY_BASE_URL = "https://daily-cloudcode-pa.googleapis.com"
SANDBOX_BASE_URL = "https://daily-cloudcode-pa.sandbox.googleapis.com"

# ============== User-Agent ==============
VERSION_FETCH_URL = "https://antigravity-auto-updater-974169037036.us-central1.run.app"
_FALLBACK_VERSION = "1.18.4"
_FALLBACK_CHROME = "132.0.6834.160"
_FALLBACK_ELECTRON = "39.2.3"
_VERSION_RE = re.compile(r"\d+\.\d+\.\d+")


def _detect_platform_info() -> str:
    """检测当前运行平台，格式对齐 AM constants.rs 的 Electron UA。"""
    os_name = platform.system().lower()
    if os_name == "darwin":
        return "Macintosh; Intel Mac OS X 10_15_7"
    elif os_name == "windows":
        return "Windows NT 10.0; Win64; x64"
    else:
        return "X11; Linux x86_64"


_PLATFORM_INFO = _detect_platform_info()


def _build_antigravity_http_user_agent(
    *,
    platform_token: str,
    version: str,
    chrome_version: str,
    electron_version: str,
) -> str:
    return (
        f"Mozilla/5.0 ({platform_token}) AppleWebKit/537.36 (KHTML, like Gecko) "
        f"Antigravity/{version} Chrome/{chrome_version} "
        f"Electron/{electron_version} Safari/537.36"
    )


# HTTP Header User-Agent（对齐 AM constants.rs: 完整 Electron 浏览器格式）
HTTP_USER_AGENT = _build_antigravity_http_user_agent(
    platform_token=_PLATFORM_INFO,
    version=_FALLBACK_VERSION,
    chrome_version=_FALLBACK_CHROME,
    electron_version=_FALLBACK_ELECTRON,
)

# V1InternalRequest.userAgent 字段（固定值）
REQUEST_USER_AGENT = "antigravity"

# --- 动态 User-Agent 支持 ---
_ua_lock = threading.Lock()
_ua_version: str = _FALLBACK_VERSION


def get_http_user_agent() -> str:
    """返回当前 HTTP User-Agent 字符串（对齐 AM Electron UA 格式）。"""
    with _ua_lock:
        return _build_antigravity_http_user_agent(
            platform_token=_PLATFORM_INFO,
            version=_ua_version,
            chrome_version=_FALLBACK_CHROME,
            electron_version=_FALLBACK_ELECTRON,
        )


def update_user_agent_version(version: str) -> None:
    """更新 User-Agent 中的版本号（由 refresh_user_agent 调用）。"""
    global HTTP_USER_AGENT, _ua_version  # noqa: PLW0603
    version = str(version or "").strip()
    if not version:
        return
    with _ua_lock:
        _ua_version = version
        HTTP_USER_AGENT = _build_antigravity_http_user_agent(
            platform_token=_PLATFORM_INFO,
            version=_ua_version,
            chrome_version=_FALLBACK_CHROME,
            electron_version=_FALLBACK_ELECTRON,
        )


def parse_version_string(text: str) -> str | None:
    """从任意文本中提取 X.Y.Z 格式的版本号。"""
    m = _VERSION_RE.search(text)
    return m.group(0) if m else None


# ============== URL 可用性 ==============
URL_UNAVAILABLE_TTL_SECONDS = 300  # 5 分钟

# ============== AM Client Identity Headers ==============
# 对齐 AM upstream/client.rs: 伪装为官方 Antigravity 客户端
# 缺少这些 header 会导致新模型（如 gemini-3.1-pro-preview）返回 404
_SESSION_ID = uuid.uuid4().hex  # 每次进程启动生成一个固定 session ID


def get_v1internal_extra_headers() -> dict[str, str]:
    """构建 v1internal 请求需要的额外 header（对齐 AM upstream/client.rs）。"""
    with _ua_lock:
        version = _ua_version

    return {
        "User-Agent": get_http_user_agent(),
        "x-client-name": "antigravity",
        "x-client-version": version,
        "x-vscode-sessionid": _SESSION_ID,
        "x-goog-api-client": "gl-node/18.18.2 fire/0.8.6 grpc/1.10.x",
    }


# ============== Thinking Signature ==============
# 统一从 core 层导入，避免多处定义
from src.core.api_format.conversion.constants import DUMMY_THOUGHT_SIGNATURE  # noqa: E402
from src.core.api_format.conversion.thinking_cache import MIN_SIGNATURE_LENGTH  # noqa: E402, F401

# ============== Thinking Budget ==============
THINKING_BUDGET_AUTO_CAP = 24576
THINKING_BUDGET_DEFAULT_INJECT = 24576  # 对齐 AM wrapper.rs (was 16000)
# 给输出留的空间（对齐 Antigravity-Manager：普通模型 8192，图像模型 2048）
OUTPUT_OVERHEAD = 8192
OUTPUT_OVERHEAD_IMAGE = 2048
# 模型最大输出限制（防止超限）
MODEL_MAX_OUTPUT_LIMIT = 65536
# 包含这些关键字的模型会自动注入 thinkingConfig（如果缺失）
THINKING_MODELS_AUTO_INJECT_KEYWORDS = (
    "thinking",
    "gemini-2.0-pro",
    "gemini-3-pro",
    "gemini-3.1-pro",
)

# ============== Google Search (Grounding) ==============
# 对齐 AM common_utils.rs: 仅 gemini-2.5-flash 支持 googleSearch tool
WEB_SEARCH_MODEL = "gemini-2.5-flash"
# 联网工具检测关键字（对齐 AM detects_networking_tool）
NETWORKING_TOOL_KEYWORDS = frozenset(
    {
        "web_search",
        "google_search",
        "web_search_20250305",
        "google_search_retrieval",
    }
)

# ============== Image Generation ==============
# 上游图像生成模型的固定名称
IMAGE_GEN_UPSTREAM_MODEL = "gemini-3-pro-image"
# 模型后缀 → 宽高比映射
IMAGE_ASPECT_RATIO_SUFFIXES: dict[str, str] = {
    "-21x9": "21:9",
    "-21-9": "21:9",
    "-16x9": "16:9",
    "-16-9": "16:9",
    "-9x16": "9:16",
    "-9-16": "9:16",
    "-4x3": "4:3",
    "-4-3": "4:3",
    "-3x4": "3:4",
    "-3-4": "3:4",
    "-3x2": "3:2",
    "-3-2": "3:2",
    "-2x3": "2:3",
    "-2-3": "2:3",
    "-5x4": "5:4",
    "-5-4": "5:4",
    "-4x5": "4:5",
    "-4-5": "4:5",
    "-1x1": "1:1",
    "-1-1": "1:1",
}
# 标准宽高比字符串（用于直接匹配 size 参数）
STANDARD_ASPECT_RATIOS = frozenset(
    {
        "21:9",
        "16:9",
        "9:16",
        "4:3",
        "3:4",
        "3:2",
        "2:3",
        "5:4",
        "4:5",
        "1:1",
    }
)
# 宽高比容差匹配表：(ratio, label)
ASPECT_RATIO_TABLE: tuple[tuple[float, str], ...] = (
    (21.0 / 9.0, "21:9"),
    (16.0 / 9.0, "16:9"),
    (4.0 / 3.0, "4:3"),
    (3.0 / 4.0, "3:4"),
    (9.0 / 16.0, "9:16"),
    (3.0 / 2.0, "3:2"),
    (2.0 / 3.0, "2:3"),
    (5.0 / 4.0, "5:4"),
    (4.0 / 5.0, "4:5"),
    (1.0, "1:1"),
)

# ============== Retry ==============
RETRY_429_BASE_SECONDS = 5.0
RETRY_503_BASE_SECONDS = 10.0
RETRY_503_MAX_SECONDS = 60.0
RETRY_500_BASE_SECONDS = 3.0

# ============== v1internal 路径 ==============
V1INTERNAL_PATH_TEMPLATE = "/v1internal:{action}"

# ============== Signature 错误关键字（用于 400 错误检测） ==============
SIGNATURE_ERROR_KEYWORDS = (
    "Invalid `signature`",
    "thinking.signature",
    "thinking.thinking",
    "Corrupted thought signature",
)

# ============== Antigravity System Instruction ==============
ANTIGRAVITY_SYSTEM_INSTRUCTION = (
    "You are Antigravity, a powerful agentic AI coding assistant designed by the "
    "Google Deepmind team working on Advanced Agentic Coding.\n"
    "You are pair programming with a USER to solve their coding task. The task may "
    "require creating a new codebase, modifying or debugging an existing codebase, "
    "or simply answering a question.\n"
    "**Absolute paths only**\n"
    "**Proactiveness**"
)

__all__ = [
    "ANTIGRAVITY_SYSTEM_INSTRUCTION",
    "ASPECT_RATIO_TABLE",
    "DAILY_BASE_URL",
    "DUMMY_THOUGHT_SIGNATURE",
    "HTTP_USER_AGENT",
    "IMAGE_ASPECT_RATIO_SUFFIXES",
    "IMAGE_GEN_UPSTREAM_MODEL",
    "MIN_SIGNATURE_LENGTH",
    "MODEL_MAX_OUTPUT_LIMIT",
    "NETWORKING_TOOL_KEYWORDS",
    "OUTPUT_OVERHEAD",
    "OUTPUT_OVERHEAD_IMAGE",
    "PROD_BASE_URL",
    "REQUEST_USER_AGENT",
    "RETRY_429_BASE_SECONDS",
    "RETRY_500_BASE_SECONDS",
    "RETRY_503_BASE_SECONDS",
    "RETRY_503_MAX_SECONDS",
    "SANDBOX_BASE_URL",
    "SIGNATURE_ERROR_KEYWORDS",
    "STANDARD_ASPECT_RATIOS",
    "THINKING_BUDGET_AUTO_CAP",
    "THINKING_BUDGET_DEFAULT_INJECT",
    "THINKING_MODELS_AUTO_INJECT_KEYWORDS",
    "URL_UNAVAILABLE_TTL_SECONDS",
    "V1INTERNAL_PATH_TEMPLATE",
    "VERSION_FETCH_URL",
    "WEB_SEARCH_MODEL",
    "get_http_user_agent",
    "get_v1internal_extra_headers",
    "parse_version_string",
    "update_user_agent_version",
]