Function _pydantic_schema_str

quantmind/magic.py:162–201

Render a JSON-schema-ish description for resolver instructions. Cases handled: 1. ``Annotated[X, ...]`` — peel via ``__metadata__`` and recurse on X. 2. Plain ``BaseModel`` subclass — use ``model_json_schema()``. 3. ``Union[...]`` / ``T | U`` — recurse on each variant; emit

(t: Any)

def _strip_optional(anno: Any) -> Any:
    """Peel ``T | None`` / ``Optional[T]`` to return the inner T."""
    origin = get_origin(anno)
    if origin in (Union, types.UnionType):
        non_none = [a for a in get_args(anno) if a is not type(None)]
        if len(non_none) == 1:
            return non_none[0]
    return anno


def _pydantic_schema_str(t: Any) -> str:
    """Render a JSON-schema-ish description for resolver instructions.

    Cases handled:

    1. ``Annotated[X, ...]`` — peel via ``__metadata__`` and recurse on X.
    2. Plain ``BaseModel`` subclass — use ``model_json_schema()``.
    3. ``Union[...]`` / ``T | U`` — recurse on each variant; emit
       ``{"oneOf": [...]}``.
    4. Anything else — fall back to ``repr`` so the resolver still gets
       *some* hint. Should not happen for the supported flows.
    """
    if hasattr(t, "__metadata__"):
        inner = get_args(t)[0]
        return _pydantic_schema_str(inner)

    if isinstance(t, type) and hasattr(t, "model_json_schema"):
        try:
            return json.dumps(t.model_json_schema(), indent=2)
        except Exception:
            # Some Pydantic models (e.g. those holding callable fields
            # like ``ModelSettings`` from the agents SDK) cannot render
            # a full JSON schema. Fall back to a name+fields summary
            # so the resolver still has something to work with.
            fields = {
                name: repr(field.annotation)
                for name, field in t.model_fields.items()
            }
            return json.dumps({"title": t.__name__, "fields": fields}, indent=2)

    origin = get_origin(t)
    if origin in (Union, types.UnionType):
        variants = get_args(t)
        schemas = [