Function _system_message_for_mode

python/copilot/_mode.py:142–175

Apply empty-mode environment_context stripping to a system message dict. The caller passes the already-normalized wire payload (a ``dict`` with ``mode`` / ``content`` / ``sections``) or ``None``. The caller's value wins if it already specifies an ``environment_context`` override.

(
    mode: CopilotClientMode | None,
    supplied: Any,
)

def _system_message_for_mode(
    mode: CopilotClientMode | None,
    supplied: Any,
) -> Any:
    """Apply empty-mode environment_context stripping to a system message dict.

    The caller passes the already-normalized wire payload (a ``dict`` with
    ``mode`` / ``content`` / ``sections``) or ``None``. The caller's value
    wins if it already specifies an ``environment_context`` override.
    """
    if mode != "empty":
        return supplied
    remove_action = {"action": "remove"}
    if supplied is None:
        return {"mode": "customize", "sections": {"environment_context": remove_action}}
    supplied_mode = supplied.get("mode", "")
    if supplied_mode == "replace":
        return supplied
    if supplied_mode == "customize":
        sections = supplied.get("sections") or {}
        if "environment_context" in sections:
            return supplied
        merged = {**supplied, "sections": {**sections, "environment_context": remove_action}}
        return merged
    # append (or unspecified): promote to customize so we can also strip
    # environment_context. The runtime appends ``content`` in both modes, so
    # the caller's text is preserved verbatim.
    out: dict[str, Any] = {
        "mode": "customize",
        "sections": {"environment_context": remove_action},
    }
    if "content" in supplied and supplied["content"] is not None:
        out["content"] = supplied["content"]
    return out


def _empty_mode_bool_default(