Spaces:
Running
Running
File size: 991 Bytes
ca56e6a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
from __future__ import annotations
from typing import Any, cast, Dict, Type
import pydantic
# --------------- Pydantic v2 compatibility ---------------
PYDANTIC_V2 = pydantic.VERSION.startswith("2.")
def model_json(model: pydantic.BaseModel, **kwargs) -> str:
if PYDANTIC_V2:
return model.model_dump_json(**kwargs)
return model.json(**kwargs) # type: ignore
def model_dump(model: pydantic.BaseModel, **kwargs) -> Dict[str, Any]:
if PYDANTIC_V2:
return model.model_dump(**kwargs)
return cast(
"dict[str, Any]",
model.dict(**kwargs),
)
def model_parse(model: Type[pydantic.BaseModel], data: Any) -> pydantic.BaseModel:
if PYDANTIC_V2:
return model.model_validate(data)
return model.parse_obj(data) # pyright: ignore[reportDeprecated]
def disable_warnings(model: Type[pydantic.BaseModel]):
# Disable warning for model_name settings
if PYDANTIC_V2:
model.model_config["protected_namespaces"] = ()
|