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