chore: #2 add description to tools

This commit is contained in:
Bhavani Ravi
2025-03-19 20:41:38 +05:30
parent 707f3747d7
commit 407eb00c1b
4 changed files with 25 additions and 4 deletions

View File

@@ -1,9 +1,10 @@
from abc import ABC, abstractmethod
from typing import Any
class BaseTools(ABC):
"""Abstract base class for tools."""
@abstractmethod
def __init__(self) -> None:
"""Initialize the tool."""
@@ -12,7 +13,7 @@ class BaseTools(ABC):
@abstractmethod
def run(self) -> Any:
"""Execute the tool's main functionality.
Returns:
Any: The result of the tool execution
"""

View File

@@ -105,7 +105,7 @@ async def get_airflow_tools(config: AirflowConfig, mode: str = "unsafe") -> list
tools.append(
Tool(
name=operation_id,
description=tool.operation.operation_id,
description=tool.operation.description,
inputSchema=schema,
)
)