From e290b8856885ff4607ee95b6d46065f19aef6c0c Mon Sep 17 00:00:00 2001 From: kingbri <8082010+kingbri1@users.noreply.github.com> Date: Mon, 10 Feb 2025 23:39:46 -0500 Subject: [PATCH] Args: Expose api-servers to subcommands This is required for the export-openapi action. Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com> --- common/args.py | 3 +++ endpoints/server.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) 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