ALT from pydantic import BaseModel from typing import Type, TypeVar, Union, Callable from langchain_core.language_models.chat_models import BaseChatModel InputT = TypeVar("InputT", bound=Union[str, BaseModel]) OutputT = TypeVar("OutputT", bound=BaseModel) def add_llm(llm: BaseChatModel): def decorator(cls: Type[BaseModel]): class Meta(type(cls)): def __ror__(cls, other: InputT): if isinstance(other, BaseModel): other = other.model_dump_json() return llm.with_structured_output(cls).invoke(other) return type(cls.__name__, (cls,), {}, metaclass=Meta) return decorator