77 lines
3.3 KiB
Python
77 lines
3.3 KiB
Python
from typing import Any, Callable, Dict
|
|
|
|
# Define the Tool class here so it's available to all handlers and the registry
|
|
class Tool:
|
|
def __init__(self, name: str, description: str, schema: Dict[str, Any], handler: Callable):
|
|
self.name = name
|
|
self.description = description
|
|
self.schema = schema
|
|
self.handler = handler
|
|
|
|
def to_mcp_dict(self):
|
|
return {
|
|
"name": self.name,
|
|
"description": self.description,
|
|
"inputSchema": self.schema
|
|
}
|
|
|
|
# Import handlers from separate files
|
|
from .read_image import handle as read_image_handler
|
|
from .read_metadata import handle as read_png_metadata_handler
|
|
from .list_thumbnails import handle as list_thumbnails_handler
|
|
from .list_directory import handle as list_directory_details_handler
|
|
|
|
# Central registry of all available tools
|
|
TOOL_REGISTRY = [
|
|
Tool(
|
|
name="read_image",
|
|
description="Reads an image from the disk and returns it as an image content object.",
|
|
schema={
|
|
"type": "object",
|
|
"properties": {"path": {"type": "string", "description": "Path to the image file"}},
|
|
"required": ["path"],
|
|
},
|
|
handler=read_image_handler
|
|
),
|
|
Tool(
|
|
name="read_png_metadata",
|
|
description="Reads the metadata (tEXt, zTXt, iTXt) from a PNG image to fetch info such as AI prompts.",
|
|
schema={
|
|
"type": "object",
|
|
"properties": {"path": {"type": "string", "description": "Path to the PNG file"}},
|
|
"required": ["path"],
|
|
},
|
|
handler=read_png_metadata_handler
|
|
),
|
|
Tool(
|
|
name="list_thumbnails",
|
|
description="Generates a thumbnail grid of images in a directory. Supports pagination and sorting.",
|
|
schema={
|
|
"type": "object",
|
|
"properties": {
|
|
"path": {"type": "string", "description": "Path to the directory containing images"},
|
|
"page": {"type": "integer", "description": "The page number to retrieve (1-indexed)", "default": 1},
|
|
"page_size": {"type": "integer", "description": "Number of images per page", "default": 64},
|
|
"sort_by": {"type": "string", "description": "Sort order: 'mtime' (newest first, default), 'name' (alphabetical), or 'size' (largest first)", "default": "mtime"},
|
|
},
|
|
"required": ["path"],
|
|
},
|
|
handler=list_thumbnails_handler
|
|
),
|
|
Tool(
|
|
name="list_directory",
|
|
description="Lists the contents of a directory in a file browser format. Directories first, then files. Supports pagination and sorting.",
|
|
schema={
|
|
"type": "object",
|
|
"properties": {
|
|
"path": {"type": "string", "description": "Path to the directory to list"},
|
|
"page": {"type": "integer", "description": "The page number to retrieve (1-indexed)", "default": 1},
|
|
"page_size": {"type": "integer", "description": "Number of items per page", "default": 64},
|
|
"sort_by": {"type": "string", "description": "Sort order: 'mtime' (newest first, default), 'name' (alphabetical), or 'size' (largest first)", "default": "mtime"},
|
|
},
|
|
"required": ["path"],
|
|
},
|
|
handler=list_directory_details_handler
|
|
),
|
|
]
|