sunholo 0.144.0__py3-none-any.whl → 0.144.1__py3-none-any.whl

sunholo/agents/fastapi/vac_routes.py

@@ -22,6 +22,7 @@ import inspect
 import asyncio
 from typing import Dict, List, Optional, Callable, Any, TYPE_CHECKING
 from functools import partial
+from contextlib import asynccontextmanager
 
 if TYPE_CHECKING:
     from fastapi import FastAPI, Request, Response, HTTPException
@@ -253,12 +254,36 @@ class VACRoutesFastAPI:
                 methods=route.get("methods", ["GET"])
             )
         
-        # Register startup event for MCP initialization
-        @self.app.on_event("startup")
-        async def startup_event():
-            if self.mcp_servers and self.mcp_client_manager and not self._mcp_initialized:
+        # Set up lifespan for MCP initialization
+        self._setup_lifespan()
+    
+    def _setup_lifespan(self):
+        """Set up lifespan context manager for app initialization."""
+        # Only set lifespan if we have MCP servers to initialize
+        if not (self.mcp_servers and self.mcp_client_manager):
+            return
+        
+        # Store the existing lifespan if any
+        existing_lifespan = getattr(self.app, 'router', self.app).lifespan_context
+        
+        @asynccontextmanager
+        async def lifespan(app: FastAPI):
+            # Startup
+            if not self._mcp_initialized:
                 await self._initialize_mcp_servers()
                 self._mcp_initialized = True
+            
+            # Call existing lifespan startup if any
+            if existing_lifespan:
+                async with existing_lifespan(app) as lifespan_state:
+                    yield lifespan_state
+            else:
+                yield
+            
+            # Shutdown (no cleanup needed for now)
+        
+        # Set the new lifespan
+        self.app.router.lifespan_context = lifespan
     
     async def home(self):
         """Home endpoint."""

{sunholo-0.144.0.dist-info → sunholo-0.144.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sunholo
-Version: 0.144.0
+Version: 0.144.1
 Summary: AI DevOps - a package to help deploy GenAI to the Cloud.
 Author-email: Holosun ApS <multivac@sunholo.com>
 License: Apache License, Version 2.0

{sunholo-0.144.0.dist-info → sunholo-0.144.1.dist-info}/RECORD

@@ -16,7 +16,7 @@ sunholo/agents/swagger.py,sha256=2tzGmpveUMmTREykZvVnDj3j295wyOMu7mUFDnXdY3c,106
 sunholo/agents/fastapi/__init__.py,sha256=f7x7kiEjaNyBiOwJHLJ4vdOiePqkXdI52sIAAHtS-ms,141
 sunholo/agents/fastapi/base.py,sha256=W-cyF8ZDUH40rc-c-Apw3-_8IIi2e4Y9qRtnoVnsc1Q,2521
 sunholo/agents/fastapi/qna_routes.py,sha256=lKHkXPmwltu9EH3RMwmD153-J6pE7kWQ4BhBlV3to-s,3864
-sunholo/agents/fastapi/vac_routes.py,sha256=cU5_t7VCbu1_6zrUBbyekEC3N7LUQCpJzr8nQGKInPM,42143
+sunholo/agents/fastapi/vac_routes.py,sha256=sK2FuJsOBgc8sIKtTOYuyhNql5JX63U_hMpQQ-nLcug,42985
 sunholo/agents/flask/__init__.py,sha256=dEoByI3gDNUOjpX1uVKP7uPjhfFHJubbiaAv3xLopnk,63
 sunholo/agents/flask/base.py,sha256=vnpxFEOnCmt9humqj-jYPLfJcdwzsop9NorgkJ-tSaU,1756
 sunholo/agents/flask/vac_routes.py,sha256=kaPUDyIH5KhCgeCEtag97qErGVZfqpY1ZEiX3y1_r-s,57505
@@ -179,9 +179,9 @@ sunholo/vertex/init.py,sha256=1OQwcPBKZYBTDPdyU7IM4X4OmiXLdsNV30C-fee2scQ,2875
 sunholo/vertex/memory_tools.py,sha256=tBZxqVZ4InTmdBvLlOYwoSEWu4-kGquc-gxDwZCC4FA,7667
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
 sunholo/vertex/type_dict_to_json.py,sha256=uTzL4o9tJRao4u-gJOFcACgWGkBOtqACmb6ihvCErL8,4694
-sunholo-0.144.0.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.144.0.dist-info/METADATA,sha256=SMIB3kckTnKrsx4HfMgnhaMiEl3MOKZzguwC4tFdLec,18700
-sunholo-0.144.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sunholo-0.144.0.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.144.0.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.144.0.dist-info/RECORD,,
+sunholo-0.144.1.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.144.1.dist-info/METADATA,sha256=ia-s14a5R5yxApJZGGqqeT2ebAlxEyc5AJhniD9xtxg,18700
+sunholo-0.144.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sunholo-0.144.1.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.144.1.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.144.1.dist-info/RECORD,,