mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-07 15:18:29 +00:00
bfe2c44f55
* feat: add docstring * feat: add new docstring * fix: linting --------- Co-authored-by: João Moura <joaomdmoura@gmail.com>
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
"""JSON encoder for handling CrewAI specific types."""
|
|
|
|
import json
|
|
from datetime import date, datetime
|
|
from decimal import Decimal
|
|
from enum import Enum
|
|
from uuid import UUID
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class CrewJSONEncoder(json.JSONEncoder):
|
|
"""Custom JSON encoder for CrewAI objects and special types."""
|
|
def default(self, obj):
|
|
if isinstance(obj, BaseModel):
|
|
return self._handle_pydantic_model(obj)
|
|
elif isinstance(obj, UUID) or isinstance(obj, Decimal) or isinstance(obj, Enum):
|
|
return str(obj)
|
|
|
|
elif isinstance(obj, datetime) or isinstance(obj, date):
|
|
return obj.isoformat()
|
|
|
|
return super().default(obj)
|
|
|
|
def _handle_pydantic_model(self, obj):
|
|
try:
|
|
data = obj.model_dump()
|
|
# Remove circular references
|
|
for key, value in data.items():
|
|
if isinstance(value, BaseModel):
|
|
data[key] = str(
|
|
value
|
|
) # Convert nested models to string representation
|
|
return data
|
|
except RecursionError:
|
|
return str(
|
|
obj
|
|
) # Fall back to string representation if circular reference is detected
|