Updated package v4.13

Author: UtkarshTheDev

CHANGELOG.md

@@ -2,6 +2,15 @@
 
 All notable changes to LocalLab will be documented in this file.
 
+## 0.4.13 - 2025-03-11
+
+### Fixed
+
+- Fixed critical error with LifespanOn initialization: "LifespanOn.**init**() got an unexpected keyword argument 'logger'"
+- Improved compatibility with different versions of uvicorn by properly handling lifespan initialization
+- Enhanced error handling for different lifespan implementations
+- Added graceful fallbacks when lifespan initialization fails
+
 ## 0.4.12 - 2025-03-11
 
 ### Fixed

locallab/__init__.py

@@ -2,7 +2,7 @@
 LocalLab - A lightweight AI inference server for running LLMs locally
 """
 
-__version__ = "0.4.12"
+__version__ = "0.4.13"
 
 # Only import what's necessary initially, lazy-load the rest
 from .logger import get_logger

locallab/server.py

@@ -209,43 +209,66 @@ async def serve(self, sockets=None):
         try:
             # Try the newer location first (uvicorn >= 0.18.0)
             from uvicorn.lifespan.on import LifespanOn
+            # LifespanOn doesn't accept logger parameter
             self.lifespan = LifespanOn(
                 self.config.app,
-                self.config.lifespan_on if hasattr(self.config, "lifespan_on") else "auto",
-                logger=logger
+                self.config.lifespan_on if hasattr(self.config, "lifespan_on") else "auto"
             )
             logger.info("Using LifespanOn from uvicorn.lifespan.on")
         except (ImportError, AttributeError) as e:
             logger.debug(f"Failed to import LifespanOn: {str(e)}")
             try:
                 # Try the older location (uvicorn < 0.18.0)
                 from uvicorn.lifespan.lifespan import Lifespan
-                self.lifespan = Lifespan(
-                    self.config.app,
-                    "auto",
-                    logger=logger
-                )
+                try:
+                    # Try with logger parameter first
+                    self.lifespan = Lifespan(
+                        self.config.app,
+                        "auto",
+                        logger
+                    )
+                except TypeError:
+                    # If that fails, try without logger parameter
+                    self.lifespan = Lifespan(
+                        self.config.app,
+                        "auto"
+                    )
                 logger.info("Using Lifespan from uvicorn.lifespan.lifespan")
             except (ImportError, AttributeError) as e:
                 logger.debug(f"Failed to import Lifespan from lifespan.lifespan: {str(e)}")
                 try:
                     # Try the oldest location
                     from uvicorn.lifespan import Lifespan
-                    self.lifespan = Lifespan(
-                        self.config.app,
-                        "auto",
-                        logger=logger
-                    )
+                    try:
+                        # Try with logger parameter first
+                        self.lifespan = Lifespan(
+                            self.config.app,
+                            "auto",
+                            logger
+                        )
+                    except TypeError:
+                        # If that fails, try without logger parameter
+                        self.lifespan = Lifespan(
+                            self.config.app,
+                            "auto"
+                        )
                     logger.info("Using Lifespan from uvicorn.lifespan")
                 except (ImportError, AttributeError) as e:
                     logger.debug(f"Failed to import Lifespan from uvicorn.lifespan: {str(e)}")
                     try:
                         # Try the newest location (uvicorn >= 0.21.0)
                         from uvicorn.lifespan.state import LifespanState
-                        self.lifespan = LifespanState(
-                            self.config.app,
-                            logger=logger
-                        )
+                        try:
+                            # Try with logger parameter first
+                            self.lifespan = LifespanState(
+                                self.config.app,
+                                logger=logger
+                            )
+                        except TypeError:
+                            # If that fails, try without logger parameter
+                            self.lifespan = LifespanState(
+                                self.config.app
+                            )
                         logger.info("Using LifespanState from uvicorn.lifespan.state")
                     except (ImportError, AttributeError) as e:
                         logger.debug(f"Failed to import LifespanState: {str(e)}")