chore: #2 add description to tools
This commit is contained in:
@@ -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
|
||||
"""
|
||||
|
||||
@@ -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,
|
||||
)
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user