Enforce function definition limits

Author: psachdeva-ss

singlestoredb/apps/_python_udfs.py

@@ -13,6 +13,8 @@
 # Keep track of currently running server
 _running_server: 'typing.Optional[AwaitableUvicornServer]' = None
 
+# Maximum number of UDFs allowed
+MAX_UDFS_LIMIT = 10
 
 async def run_udf_app(
     log_level: str = 'error',
@@ -46,6 +48,11 @@ async def run_udf_app(
         udf_suffix = '_test'
     app = Application(url=base_url, app_mode='managed', name_suffix=udf_suffix)
 
+    if not app.endpoints:
+        raise ValueError("You must define at least one function.")
+    if len(app.endpoints) > MAX_UDFS_LIMIT:
+        raise ValueError(f"You can only define a maximum of {MAX_UDFS_LIMIT} functions.")
+
     config = uvicorn.Config(
         app,
         host='0.0.0.0',