diff --git a/common/args.py b/common/args.py index 56f9659..a0da4f9 100644 --- a/common/args.py +++ b/common/args.py @@ -101,6 +101,9 @@ def add_subcommands(parser: argparse.ArgumentParser): "--export-path", help="Path to export the generated OpenAPI JSON (default: openapi.json)", ) + openapi_export_parser.add_argument( + "--api-servers", nargs="+", help="Sets API servers to run when exporting" + ) # Calls config export action config_export_parser = actions_subparsers.add_parser( diff --git a/endpoints/server.py b/endpoints/server.py index 3555a5b..c17cbb4 100644 --- a/endpoints/server.py +++ b/endpoints/server.py @@ -1,9 +1,9 @@ import asyncio -from typing import Optional import uvicorn from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware from loguru import logger +from typing import Optional from common.logger import UVICORN_LOG_CONFIG from common.networking import get_global_depends