dify/api/core/model_runtime/utils/helper.py

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()