From 679523a7c6b81c6abf9258294aa155e35c1914ea Mon Sep 17 00:00:00 2001 From: abhishekbhakat Date: Tue, 25 Feb 2025 11:18:45 +0000 Subject: [PATCH] Using older env variables --- src/airflow_mcp_server/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/airflow_mcp_server/__init__.py b/src/airflow_mcp_server/__init__.py index fe6efa2..4f860f2 100644 --- a/src/airflow_mcp_server/__init__.py +++ b/src/airflow_mcp_server/__init__.py @@ -31,9 +31,9 @@ def main(verbose: int, safe: bool, unsafe: bool, base_url: str = None, spec_path # Read environment variables with proper precedence # Environment variables take precedence over CLI arguments config_base_url = os.environ.get("AIRFLOW_BASE_URL") or base_url - config_spec_path = os.environ.get("AIRFLOW_SPEC_PATH") or spec_path - config_auth_token = os.environ.get("AIRFLOW_AUTH_TOKEN") or auth_token - config_cookie = os.environ.get("AIRFLOW_COOKIE") or cookie + config_spec_path = os.environ.get("OPENAPI_SPEC") or spec_path + config_auth_token = os.environ.get("AUTH_TOKEN") or auth_token + config_cookie = os.environ.get("COOKIE") or cookie # Initialize configuration try: