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

refactor for dynamic Models

Browse Source
This commit is contained in:
abhishekbhakat
2025-02-13 10:03:23 +00:00
parent 18962618bc
commit b87fe176bd
6 changed files with 90 additions and 385 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
airflow-mcp-server/src/airflow_mcp_server/tools/airflow_tool.py
View File

@@ -32,14 +32,17 @@ def create_validation_error(field: str, message: str) -> ValidationError:
class AirflowTool(BaseTools):
"""Tool for executing Airflow API operations."""
"""
Tool for executing Airflow API operations.
AirflowTool is supposed to have objects per operation.
"""
def __init__(self, operation_details: OperationDetails, client: AirflowClient) -> None:
"""Initialize tool with operation details and client.
Args:
operation_details: Parsed operation details
client: Configured Airflow API client
operation_details: Operation details
client: AirflowClient instance
"""
super().__init__()
self.operation = operation_details