diff --git a/anansyTypes/schemas.py b/anansyTypes/schemas.py index b288816..3f7b021 100644 --- a/anansyTypes/schemas.py +++ b/anansyTypes/schemas.py @@ -25,7 +25,7 @@ class ModifiedBaseModel(PydanticBaseModel): return { 'id': v.id, 'label': v.label, - 'image': v.image.tolist() + 'image': v.image.tolist() if hasattr(v.image, 'tolist') else v.image } if isinstance(v, Embedding): return {'data': v.data.tolist()}