json_safe.py

Go to the documentation of this file.

# Copyright (c) 2025-present Aquin Labs Private Limited. All Rights Reserved.
# This file is part of the Aquin Engine. Unauthorized copying, modification,
# distribution, or use of this file, via any medium, is strictly prohibited.
# Proprietary and confidential. See LICENSE for terms.
 
"""JSON helpers that never emit NaN/Inf (invalid for JS JSON.parse)."""
from __future__ import annotations
 
import json
import math
from typing import Any
 
 
def sanitize_jsonable(obj: Any) -> Any:
    """Recursively replace non-finite floats with None; pass through the rest."""
    if isinstance(obj, float):
        return None if (math.isnan(obj) or math.isinf(obj)) else obj
    if isinstance(obj, dict):
        return {k: sanitize_jsonable(v) for k, v in obj.items()}
    if isinstance(obj, (list, tuple)):
        return [sanitize_jsonable(v) for v in obj]
    # numpy / torch scalars often duck-type to float via .item()
    item = getattr(obj, "item", None)
    if callable(item):
        try:
            return sanitize_jsonable(item())
        except Exception:
            return str(obj)
    return obj
 
 
def dumps_json_safe(obj: Any, **kwargs: Any) -> str:
    """json.dumps with NaN/Inf stripped so receivers always get strict JSON."""
    kwargs.setdefault("ensure_ascii", False)
    kwargs["allow_nan"] = False
    return json.dumps(sanitize_jsonable(obj), **kwargs)