mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-10-15 16:18:10 +08:00
21 lines
580 B
Python
21 lines
580 B
Python
import json
|
|
from typing import TYPE_CHECKING, Any, Dict
|
|
|
|
|
|
if TYPE_CHECKING:
|
|
from pydantic import BaseModel
|
|
|
|
|
|
def dictify(data: "BaseModel") -> Dict[str, Any]:
|
|
try: # pydantic v2
|
|
return data.model_dump(exclude_unset=True)
|
|
except AttributeError: # pydantic v1
|
|
return data.dict(exclude_unset=True)
|
|
|
|
|
|
def jsonify(data: "BaseModel") -> str:
|
|
try: # pydantic v2
|
|
return json.dumps(data.model_dump(exclude_unset=True), ensure_ascii=False)
|
|
except AttributeError: # pydantic v1
|
|
return data.json(exclude_unset=True, ensure_ascii=False)
|