diff --git a/aurora/utilities/json.py b/aurora/utilities/json.py index bac1668..c3e79a3 100644 --- a/aurora/utilities/json.py +++ b/aurora/utilities/json.py @@ -1,7 +1,7 @@ import json from datetime import datetime, timedelta -from aurora.models import Moderation +from pydantic import BaseModel class JSONEncoder(json.JSONEncoder): @@ -10,7 +10,7 @@ class JSONEncoder(json.JSONEncoder): return int(o.timestamp()) if isinstance(o, timedelta): return str(o) - if isinstance(o, Moderation): + if isinstance(o, BaseModel): return o.model_dump() return super().default(o)