Updated LocalLab v0.2.3

Author: UtkarshTheDev

CHANGELOG.md

@@ -2,6 +2,15 @@
 
 All notable changes for version updates.
 
+## [0.2.3] - 2025-03-03
+
+### Fixed
+
+- Fixed critical server startup error in Google Colab environment with uvicorn callback configuration
+- Resolved "'list' object is not callable" error by properly implementing the callback_notify as an async function
+- Enhanced server startup sequence for better compatibility with both local and Colab environments
+- Improved custom server implementation to handle callbacks more robustly
+
 ## [0.2.2] - 2025-03-03
 
 ### Fixed

locallab/__init__.py

@@ -2,7 +2,7 @@
 LocalLab - A lightweight AI inference server
 """
 
-__version__ = "0.2.2" 
+__version__ = "0.2.3" 
 
 from typing import Dict, Any, Optional
 

locallab/server.py

@@ -253,14 +253,19 @@ def on_startup():
             import nest_asyncio
             nest_asyncio.apply()
             logger.info(f"Starting server on port {port} (Colab mode)")
+            
+            # Define the callback for Colab
+            async def on_startup_async():
+                on_startup()
+            
             config = uvicorn.Config(
                 app, 
                 host="0.0.0.0", 
                 port=port, 
                 reload=False, 
                 log_level="info",
-                # Add callback to print the RUNNING banner when server starts
-                callback_notify=[on_startup]
+                # Use an async callback function, not a list
+                callback_notify=on_startup_async
             )
             server = uvicorn.Server(config)
             asyncio.get_event_loop().run_until_complete(server.serve())
@@ -278,8 +283,13 @@ async def serve(self, sockets=None):
                     self.config.setup_event_loop()
                     await self.startup(sockets=sockets)
                     # Call our callback before processing requests
-                    for callback in self.config.callback_notify:
-                        callback()
+                    if callable(self.config.callback_notify):
+                        await self.config.callback_notify()
+                    elif isinstance(self.config.callback_notify, list):
+                        for callback in self.config.callback_notify:
+                            if callable(callback):
+                                callback()
+                    on_startup()  # Always ensure our startup function is called
                     await self.main_loop()
                     await self.shutdown()
 
@@ -290,7 +300,8 @@ async def serve(self, sockets=None):
                 reload=False, 
                 workers=1, 
                 log_level="info",
-                callback_notify=[on_startup]
+                # This won't be used directly, as we call on_startup in the ServerWithCallback class
+                callback_notify=None
             )
             server = ServerWithCallback(config)
             asyncio.run(server.serve())

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="locallab",
-    version="0.2.2",
+    version="0.2.3",
     packages=find_packages(include=["locallab", "locallab.*"]),
     install_requires=[
         "fastapi>=0.68.0,<1.0.0",