From ed116367a5704ecab339bf2eeda1491bda8d84cf Mon Sep 17 00:00:00 2001 From: abhishekbhakat Date: Sun, 9 Feb 2025 20:41:22 +0000 Subject: [PATCH] Base Tools skeleton --- .../src/airflow_mcp_server/tools/base_tools.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/airflow-mcp-server/src/airflow_mcp_server/tools/base_tools.py b/airflow-mcp-server/src/airflow_mcp_server/tools/base_tools.py index cff3932..e154d1d 100644 --- a/airflow-mcp-server/src/airflow_mcp_server/tools/base_tools.py +++ b/airflow-mcp-server/src/airflow_mcp_server/tools/base_tools.py @@ -1,10 +1,19 @@ from abc import ABC, abstractmethod +from typing import Any class BaseTools(ABC): - - def __init__(self): + """Abstract base class for tools.""" + + @abstractmethod + def __init__(self) -> None: + """Initialize the tool.""" pass @abstractmethod - def run(self): - pass \ No newline at end of file + def run(self) -> Any: + """Execute the tool's main functionality. + + Returns: + Any: The result of the tool execution + """ + raise NotImplementedError