Function parse_value_for_schema

src/zenml/utils/json_utils.py:191–244

Parses a raw string input into a value compatible with a JSON schema. This does not perform a full JSON Schema validation. It only checks if the parsed **top-level type** matches the schema. If the parsed object does not match the schema and the schema accepts strings, the raw stri

(
    raw_value: str,
    schema: Dict[str, Any],
)

def parse_value_for_schema(
    raw_value: str,
    schema: Dict[str, Any],
) -> Any:
    """Parses a raw string input into a value compatible with a JSON schema.

    This does not perform a full JSON Schema validation. It only checks if the
    parsed **top-level type** matches the schema.

    If the parsed object does not match the schema and the schema accepts
    strings, the raw string is returned.

    Args:
        raw_value: The raw user input.
        schema: JSON schema.

    Raises:
        ValueError: If the input cannot be coerced to a value matching the
            schema's top-level types.

    Returns:
        The coerced value.
    """  # noqa: DOC503
    parse_error: Optional[json.JSONDecodeError] = None
    parsed: Any = None
    try:
        parsed = json.loads(raw_value)
    except json.JSONDecodeError as e:
        parse_error = e

    allowed = _schema_allowed_json_types(schema)

    if parse_error is None:
        if not allowed:
            return parsed
        parsed_type = _json_type_of(parsed)
        candidates = {parsed_type} if parsed_type else set()
        # JSON Schema: integers are also valid numbers.
        if parsed_type == "integer":
            candidates.add("number")
        if candidates & allowed:
            return parsed
        if "string" in allowed:
            return raw_value

        raise ValueError(
            f"Input does not match expected schema: got "
            f"{parsed_type}, expected one of {sorted(allowed)}."
        )

    if "string" in allowed or not allowed:
        return raw_value

    raise parse_error