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

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()
Model Runtime (#1858) Co-authored-by: StyleZhang <jasonapring2015@outlook.com> Co-authored-by: Garfield Dai <dai.hai@foxmail.com> Co-authored-by: chenhe <guchenhe@gmail.com> Co-authored-by: jyong <jyong@dify.ai> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Yeuoly <admin@srmxy.cn> 2024-01-02 23:42:00 +08:00			`import pydantic`
			`from pydantic import BaseModel`


			`def dump_model(model: BaseModel) -> dict:`
chore(api/core): apply ruff reformatting (#7624) 2024-09-10 17:00:20 +08:00			`if hasattr(pydantic, "model_dump"):`
Model Runtime (#1858) Co-authored-by: StyleZhang <jasonapring2015@outlook.com> Co-authored-by: Garfield Dai <dai.hai@foxmail.com> Co-authored-by: chenhe <guchenhe@gmail.com> Co-authored-by: jyong <jyong@dify.ai> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Yeuoly <admin@srmxy.cn> 2024-01-02 23:42:00 +08:00			`return pydantic.model_dump(model)`
			`else:`
improve: mordernizing validation by migrating pydantic from 1.x to 2.x (#4592) 2024-06-14 01:05:37 +08:00			`return model.model_dump()`