dify/api/core/model_runtime/utils/helper.py
2024-09-10 17:00:20 +08:00

10 lines
217 B
Python

import pydantic
from pydantic import BaseModel
def dump_model(model: BaseModel) -> dict:
if hasattr(pydantic, "model_dump"):
return pydantic.model_dump(model)
else:
return model.model_dump()