From 2aed8495286ec3ea96fd30c23cd36e28cb23c450 Mon Sep 17 00:00:00 2001
From: Batuhan Taskaya <batuhan@python.org>
Date: Wed, 24 Jan 2024 15:32:56 +0300
Subject: [PATCH] fix: add serialization listeners for classes (#56)

---
 projects/fal/src/fal/app.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/projects/fal/src/fal/app.py b/projects/fal/src/fal/app.py
index ba7d5782..3ab296eb 100644
--- a/projects/fal/src/fal/app.py
+++ b/projects/fal/src/fal/app.py
@@ -3,6 +3,7 @@
 import inspect
 import os
 import fal.api
+from fal._serialization import add_serialization_listeners_for
 from contextlib import asynccontextmanager
 from fal.toolkit import mainify
 from fastapi import FastAPI
@@ -14,6 +15,8 @@
 
 
 def wrap_app(cls: type[App], **kwargs) -> fal.api.IsolatedFunction:
+    add_serialization_listeners_for(cls)
+
     def initialize_and_serve():
         app = cls()
         app.serve()