From ef3167cb078b50dd6df0d2a18d17968da72e592d Mon Sep 17 00:00:00 2001 From: Stanislav Novosad Date: Tue, 14 Oct 2025 20:05:35 -0600 Subject: [PATCH] Log server start/stop (#3715) --- skyvern/forge/api_app.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/skyvern/forge/api_app.py b/skyvern/forge/api_app.py index 2354300f..68593be6 100644 --- a/skyvern/forge/api_app.py +++ b/skyvern/forge/api_app.py @@ -1,6 +1,7 @@ import uuid +from contextlib import asynccontextmanager from datetime import datetime -from typing import Awaitable, Callable +from typing import Any, AsyncGenerator, Awaitable, Callable import structlog from fastapi import FastAPI, Response, status @@ -50,12 +51,20 @@ def custom_openapi() -> dict: return app.openapi_schema +@asynccontextmanager +async def lifespan(_: FastAPI) -> AsyncGenerator[None, Any]: + """Lifespan context manager for FastAPI app startup and shutdown.""" + LOG.info("Server started") + yield + LOG.info("Server shutting down") + + def get_agent_app() -> FastAPI: """ Start the agent server. """ - app = FastAPI() + app = FastAPI(lifespan=lifespan) # Add CORS middleware app.add_middleware(