abhishekbhakat/airflow-mcp-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

accept yaml spec for openapi

Browse Source
This commit is contained in:
abhishekbhakat
2025-02-10 12:34:07 +00:00
parent 447ab7643b
commit 0bb7e0400c
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
airflow-mcp-server/src/airflow_mcp_server/__init__.py
View File

@@ -8,10 +8,10 @@ from airflow_mcp_server.server import serve
@click.command() @click.command()
@click.option("--working-dir", "-w", type=Path, help="Working directory path") @click.option("-y", "--yaml-spec", type=Path, help="YAML spec file")
@click.option("-v", "--verbose", count=True) @click.option("-v", "--verbose", count=True)
def main(working_dir: Path | None, verbose: bool) -> None: def main(yaml_spec: Path | None, verbose: bool) -> None:
"""MCP Code Assist Server - Code operations for MCP""" """MCP server for Airflow"""
import asyncio import asyncio
logging_level = logging.WARN logging_level = logging.WARN
@@ -21,7 +21,7 @@ def main(working_dir: Path | None, verbose: bool) -> None:
logging_level = logging.DEBUG logging_level = logging.DEBUG
logging.basicConfig(level=logging_level, stream=sys.stderr) logging.basicConfig(level=logging_level, stream=sys.stderr)
asyncio.run(serve(working_dir)) asyncio.run(serve(yaml_spec))
if __name__ == "__main__": if __name__ == "__main__":