WIP need to fix encoder

This commit is contained in:
Lorenze Jay
2024-07-08 14:37:19 -07:00
parent 1a44a34c17
commit 5c04c63127

View File

@@ -0,0 +1,22 @@
import json
from datetime import datetime
from uuid import UUID
from openai import BaseModel
class CrewJSONEncoder(json.JSONEncoder):
"""
Custom JSON Encoder for Crew related objects.
"""
def default(self, obj):
if isinstance(obj, datetime):
return obj.isoformat()
if isinstance(obj, UUID):
return str(obj)
if isinstance(obj, BaseModel):
return obj.model_dump()
if isinstance(obj, set):
return list(obj)
return super().default(obj)