diff --git a/python/packages/autogen-studio/autogenstudio/cli.py b/python/packages/autogen-studio/autogenstudio/cli.py index 24e0d5549..ca220b530 100644 --- a/python/packages/autogen-studio/autogenstudio/cli.py +++ b/python/packages/autogen-studio/autogenstudio/cli.py @@ -75,6 +75,8 @@ def serve( os.environ["AUTOGENSTUDIO_API_DOCS"] = str(docs) os.environ["AUTOGENSTUDIO_TEAM_FILE"] = team + os.environ["AUTOGENSTUDIO_HOST"] = host + os.environ["AUTOGENSTUDIO_PORT"] = str(port) # validate the team file if not os.path.exists(team): @@ -86,6 +88,7 @@ def serve( port=port, workers=workers, reload=False, + log_level="info", ) diff --git a/python/packages/autogen-studio/autogenstudio/web/app.py b/python/packages/autogen-studio/autogenstudio/web/app.py index 2e2ad3337..0fda58560 100644 --- a/python/packages/autogen-studio/autogenstudio/web/app.py +++ b/python/packages/autogen-studio/autogenstudio/web/app.py @@ -31,12 +31,10 @@ async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]: Lifecycle manager for the FastAPI application. Handles initialization and cleanup of application resources. """ - # Startup - logger.info("Initializing application...") + try: # Initialize managers (DB, Connection, Team) await init_managers(initializer.database_uri, initializer.config_dir, initializer.app_root) - logger.info("Managers initialized successfully") # Any other initialization code logger.info("Application startup complete")