Added serialize function for save

aixplain/utils/asset_cache.py

@@ -112,7 +112,7 @@ def save(self):
                 data_dict = {}
                 for asset_id, asset in self.store.data.items():
                     try:
-                        data_dict[asset_id] = asset.to_dict()
+                        data_dict[asset_id] = serialize(asset)
                     except Exception as e:
                         logger.error(f"Error serializing {asset_id}: {e}")
                 serializable_store = {
@@ -138,3 +138,18 @@ def get_all(self) -> List[T]:
 
     def has_valid_cache(self) -> bool:
         return self.store.expiry >= time.time() and bool(self.store.data)
+
+def serialize(obj):
+    if isinstance(obj, (str, int, float, bool, type(None))):
+        return obj
+    elif isinstance(obj, (list, tuple, set)):
+        return [serialize(o) for o in obj]
+    elif isinstance(obj, dict):
+        return {str(k): serialize(v) for k, v in obj.items()}
+    elif hasattr(obj, "to_dict"):
+        return serialize(obj.to_dict())
+    elif hasattr(obj, "__dict__"):
+        return serialize(vars(obj))
+    else:
+        return str(obj)
+

tests/functional/model/run_model_test.py

@@ -217,7 +217,7 @@ def test_aixplain_model_cache_creation():
 
     assert "data" in cache_data, "Cache file structure invalid - missing 'data' key."
     assert any(m.get("id") == model_id for m in cache_data["data"]["items"]), "Instantiated model not found in cache."
-=======
+
 def test_index_model_air_with_image():
     from aixplain.factories import IndexFactory
     from aixplain.modules.model.record import Record