PostgreSQL-Performance-Tuner-Mcp

o �?%i��@
s\dZd
dl
mZ
d
dlmZmZ
d
dlmZ
d
dlm Z
d
dl mZmZ
Gdd�de�Z d S) zHAbstract base class for tool handlers following the ToolHandler pattern.�)
�annotations)�ABC�abstractmethod)
�Sequence)
�Any)�TextContent�Toolc@
sneZ
dZUd
ZdZded<dZded<ed dd ��
Zed!d d��
Z d"dd�Z d#dd�Zd$dd�Zd%dd�Z dS)&�ToolHandlera� Abstract base class for MCP tool handlers. This pattern allows for modular organization of tools where each handler is responsible for defining its schema and executing its logic. Subclasses must implement: - name: The unique tool name - description: Human-readable description - get_tool_definition(): Returns the Tool schema - run_tool(): Executes the tool logic Example: class MyToolHandler(ToolHandler): name = "my_tool" description = "Does something useful" def get_tool_definition(self) -> Tool: return Tool( name=self.name, description=self.description, inputSchema={ "type": "object", "properties": { "param": {"type": "string", "description": "A parameter"} }, "required": ["param"] } ) async def run_tool(self, arguments: dict[str, Any]) -> Sequence[TextContent]: result = do_something(arguments["param"]) return [TextContent(type="text", text=result)] ��str�name�description�returnrc


C
sd
S)z� Return the MCP Tool definition including input schema. Returns: Tool: The tool definition with name, description, and inputSchema N�)
�selfrr�R/home/azureuser/mcp_pg_tuning_expert/src/mcp_pg_tuning_expert/tools/toolhandler.py�get_tool_definition3szToolHandler.get_tool_definition� arguments�dict[str, Any]�Sequence[TextContent]c
�
s�
d
S)aw
Execute the tool logic with the provided arguments. Args: arguments: Dictionary of arguments matching the input schema Returns: Sequence[TextContent]: The tool output as text content Raises: ValueError: If required arguments are missing or invalid Exception: If tool execution fails Nr)rrrrr�run_tool=s�zToolHandler.run_tool�required� list[str]�Nonec
s.�f
d
d�|D�
}|rtdd�
|�
���
�
dS)a
Validate that required arguments are present. Args: arguments: The arguments dictionary to validate required: List of required argument names Raises: ValueError: If any required argument is missing c

s$g|]}
|
�v
s�|
dur|
�qS)
Nr)�.0�arg�
rrr� <listcomp>]s$z6ToolHandler.validate_required_args.<locals>.<listcomp>zMissing required arguments: z, N)� ValueError�join)rrr�missingrrr�validate_required_argsNs

�z"ToolHandler.validate_required_args�error� ExceptioncC
std
dt
|
�
��d�g
S)z� Format an error as TextContent for tool response. Args: error: The exception to format Returns: Sequence[TextContent]: Error message as text content �textzError: ��typer$)rr)rr"rrr�format_erroras zToolHandler.format_error�resultcC
std
|
d�g
S)z� Format a string result as TextContent. Args: result: The result string to format Returns: Sequence[TextContent]: Result as text content r$r%)
r)rr(rrr� format_resultms zToolHandler.format_result�datarcC
s"d
dl}t
d|j|
dtd�d�g
S)z� Format data as JSON TextContent. Args: data: The data to serialize as JSON Returns: Sequence[TextContent]: JSON-formatted text content r
Nr$�)�indent�defaultr%)�jsonr�dumpsr)rr*r.rrr�format_json_resultys 
zToolHandler.format_json_resultN)rr)rrrr)rrrrrr)r"r#rr)r(rrr)r*rrr)�__name__� __module__�__qualname__�__doc__r�__annotations__r rrrr!r'r)r0rrrrr s 
#

 
  r N)r4� __future__r�abcrr�collections.abcr�typingr� mcp.typesrrr rrrr�<module>
s