From 420b6fc68f8692774fee61acffeed6bef1a5741a Mon Sep 17 00:00:00 2001 From: abhishekbhakat Date: Tue, 25 Feb 2025 06:10:28 +0000 Subject: [PATCH] safe and unsafe are mutually exclusive --- src/airflow_mcp_server/__init__.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/airflow_mcp_server/__init__.py b/src/airflow_mcp_server/__init__.py index 8decf7f..3b90bcb 100644 --- a/src/airflow_mcp_server/__init__.py +++ b/src/airflow_mcp_server/__init__.py @@ -22,12 +22,15 @@ def main(verbose: int, safe: bool, unsafe: bool) -> None: logging.basicConfig(level=logging_level, stream=sys.stderr) + # Determine server mode with proper precedence if safe and unsafe: + # CLI argument validation raise click.UsageError("Options --safe and --unsafe are mutually exclusive") - - if safe: + elif safe: + # CLI argument for safe mode asyncio.run(serve_safe()) - else: # Default to unsafe mode + else: + # Default to unsafe mode asyncio.run(serve_unsafe())