Function wrap_cli_request

src/api/antigravity.py:196–239

将 Gemini 格式请求包装成 Antigravity CLI 格式。 返回 (payload, request_id)。

(
    gemini_request: Dict[str, Any],
    model: str,
    project_id: str,
)

async def wrap_cli_request(
    gemini_request: Dict[str, Any],
    model: str,
    project_id: str,
) -> Tuple[Dict[str, Any], str]:
    """
    将 Gemini 格式请求包装成 Antigravity CLI 格式。
    返回 (payload, request_id)。
    """
    inner = dict(gemini_request)

    # 移除 safetySettings（CLI 不发送）
    inner.pop("safetySettings", None)

    # 获取/更新会话状态
    state = await _get_session_state(inner, model)

    # 注入 sessionId
    if not inner.get("sessionId"):
        inner["sessionId"] = state.session_id

    # 注入 labels
    inner["labels"] = _build_labels(model, state.trajectory_id, state.step_index)

    # toolConfig 默认 VALIDATED
    tool_config = inner.get("toolConfig") or {}
    func_config = tool_config.get("functionCallingConfig") or {}
    if "mode" not in func_config:
        func_config["mode"] = "VALIDATED"
    tool_config["functionCallingConfig"] = func_config
    inner["toolConfig"] = tool_config

    request_id = _generate_request_id(state.conversation_id, state.trajectory_id, state.step_index)

    payload = {
        "project": project_id,
        "requestId": request_id,
        "request": inner,
        "model": model,
        "userAgent": "antigravity",
        "requestType": "agent",
        "enabledCreditTypes": ["GOOGLE_ONE_AI"],
    }
    return payload, request_id


# ==================== 辅助函数 ====================