"""Tool definitions and JSON Schemas.
Long descriptive strings are intentionally kept on single lines for stability of
user-facing help text. Line length lint (E501) is disabled for this file.
This module returns the list of MCP Tool objects supported by the current
server implementation. It mirrors the previous inline definitions from
``server.py`` (pre-refactor) to preserve backwards compatibility for clients
and tests.
"""
# ruff: noqa: E501
from mcp.types import Tool
def get_tool_definitions() -> list[Tool]:
"""Return tool definitions (schemas + descriptions)."""
return [
Tool(
name="get_root",
description="Get STAC API root document with basic metadata and conformance",
inputSchema={
"type": "object",
"properties": {
"output_format": {
"type": "string",
"description": "Result output format: 'text' (default) or 'json'",
"enum": ["text", "json"],
"default": "text",
},
"catalog_url": {
"type": "string",
"description": "STAC catalog URL (optional, defaults to Microsoft Planetary Computer)",
},
},
},
),
Tool(
name="get_conformance",
description="List STAC API conformance classes; optionally check specific classes",
inputSchema={
"type": "object",
"properties": {
"output_format": {
"type": "string",
"description": "Result output format: 'text' (default) or 'json'",
"enum": ["text", "json"],
"default": "text",
},
"check": {
"oneOf": [
{"type": "string"},
{"type": "array", "items": {"type": "string"}},
],
"description": "Conformance class URI or list of URIs to verify",
},
"catalog_url": {
"type": "string",
"description": "STAC catalog URL (optional, defaults to Microsoft Planetary Computer)",
},
},
},
),
Tool(
name="get_queryables",
description="Get queryable fields (global or per collection) if supported",
inputSchema={
"type": "object",
"properties": {
"output_format": {
"type": "string",
"description": "Result output format: 'text' (default) or 'json'",
"enum": ["text", "json"],
"default": "text",
},
"collection_id": {
"type": "string",
"description": "Collection ID for collection-specific queryables (optional)",
},
"catalog_url": {
"type": "string",
"description": "STAC catalog URL (optional, defaults to Microsoft Planetary Computer)",
},
},
},
),
Tool(
name="get_aggregations",
description="Run a STAC search requesting aggregations (count/stats) if supported",
inputSchema={
"type": "object",
"properties": {
"output_format": {
"type": "string",
"description": "Result output format: 'text' (default) or 'json'",
"enum": ["text", "json"],
"default": "text",
},
"collections": {
"type": "array",
"items": {"type": "string"},
"description": "List of collection IDs to search within",
},
"bbox": {
"type": "array",
"items": {"type": "number"},
"minItems": 4,
"maxItems": 4,
"description": "Bounding box [west, south, east, north] in WGS84",
},
"datetime": {
"type": "string",
"description": "Date/time filter (ISO 8601 format, e.g., '2023-01-01/2023-12-31')",
},
"query": {
"type": "object",
"description": "Additional query parameters for filtering items",
},
"fields": {
"type": "array",
"items": {"type": "string"},
"description": "Fields to aggregate for field-based operations (e.g., stats, histogram)",
},
"operations": {
"type": "array",
"items": {"type": "string"},
"description": "Aggregation operation types (e.g., count, stats, histogram)",
},
"limit": {
"type": "integer",
"description": "Optional item limit for the search (0 means rely on server defaults)",
"default": 0,
"minimum": 0,
"maximum": 1000,
},
"catalog_url": {
"type": "string",
"description": "STAC catalog URL (optional, defaults to Microsoft Planetary Computer)",
},
},
},
),
Tool(
name="search_collections",
description="Search and list available STAC collections",
inputSchema={
"type": "object",
"properties": {
"output_format": {
"type": "string",
"description": "Result output format: 'text' (default) or 'json'",
"enum": ["text", "json"],
"default": "text",
},
"limit": {
"type": "integer",
"description": "Maximum number of collections to return",
"default": 10,
"minimum": 1,
"maximum": 100,
},
"catalog_url": {
"type": "string",
"description": "STAC catalog URL (optional, defaults to Microsoft Planetary Computer)",
},
},
},
),
Tool(
name="get_collection",
description="Get detailed information about a specific STAC collection",
inputSchema={
"type": "object",
"properties": {
"output_format": {
"type": "string",
"description": "Result output format: 'text' (default) or 'json'",
"enum": ["text", "json"],
"default": "text",
},
"collection_id": {
"type": "string",
"description": "ID of the collection to retrieve",
},
"catalog_url": {
"type": "string",
"description": "STAC catalog URL (optional, defaults to Microsoft Planetary Computer)",
},
},
"required": ["collection_id"],
},
),
Tool(
name="search_items",
description="Search for STAC items across collections",
inputSchema={
"type": "object",
"properties": {
"output_format": {
"type": "string",
"description": "Result output format: 'text' (default) or 'json'",
"enum": ["text", "json"],
"default": "text",
},
"collections": {
"type": "array",
"items": {"type": "string"},
"description": "List of collection IDs to search within",
},
"bbox": {
"type": "array",
"items": {"type": "number"},
"minItems": 4,
"maxItems": 4,
"description": "Bounding box [west, south, east, north] in WGS84",
},
"datetime": {
"type": "string",
"description": "Date/time filter (ISO 8601 format, e.g., '2023-01-01/2023-12-31')",
},
"query": {
"type": "object",
"description": "Additional query parameters for filtering items",
},
"limit": {
"type": "integer",
"description": "Maximum number of items to return",
"default": 10,
"minimum": 1,
"maximum": 100,
},
"catalog_url": {
"type": "string",
"description": "STAC catalog URL (optional, defaults to Microsoft Planetary Computer)",
},
},
},
),
Tool(
name="get_item",
description="Get detailed information about a specific STAC item",
inputSchema={
"type": "object",
"properties": {
"output_format": {
"type": "string",
"description": "Result output format: 'text' (default) or 'json'",
"enum": ["text", "json"],
"default": "text",
},
"collection_id": {
"type": "string",
"description": "ID of the collection containing the item",
},
"item_id": {
"type": "string",
"description": "ID of the item to retrieve",
},
"catalog_url": {
"type": "string",
"description": "STAC catalog URL (optional, defaults to Microsoft Planetary Computer)",
},
},
"required": ["collection_id", "item_id"],
},
),
Tool(
name="estimate_data_size",
description="Estimate data size for STAC items using lazy loading with odc.stac",
inputSchema={
"type": "object",
"properties": {
"output_format": {
"type": "string",
"description": "Result output format: 'text' (default) or 'json'",
"enum": ["text", "json"],
"default": "text",
},
"collections": {
"type": "array",
"items": {"type": "string"},
"description": "List of collection IDs to search within",
},
"bbox": {
"type": "array",
"items": {"type": "number"},
"minItems": 4,
"maxItems": 4,
"description": "Bounding box [west, south, east, north] in WGS84",
},
"datetime": {
"type": "string",
"description": "Date/time filter (ISO 8601 format, e.g., '2023-01-01/2023-12-31')",
},
"query": {
"type": "object",
"description": "Additional query parameters for filtering items",
},
"aoi_geojson": {
"type": "object",
"description": "Area of Interest as GeoJSON geometry for clipping (will use smallest bbox between this and bbox parameter)",
},
"limit": {
"type": "integer",
"description": "Maximum number of items to analyze for size estimation",
"default": 100,
"minimum": 1,
"maximum": 500,
},
"catalog_url": {
"type": "string",
"description": "STAC catalog URL (optional, defaults to Microsoft Planetary Computer)",
},
},
},
),
Tool(
name="create_item",
description="Create a new STAC item in a collection",
inputSchema={
"type": "object",
"properties": {
"collection_id": {
"type": "string",
"description": "ID of the collection",
},
"item": {"type": "object", "description": "STAC item JSON"},
},
"required": ["collection_id", "item"],
},
),
Tool(
name="update_item",
description="Update an existing STAC item",
inputSchema={
"type": "object",
"properties": {
"item": {
"type": "object",
"description": "STAC item JSON to update",
},
},
"required": ["item"],
},
),
Tool(
name="delete_item",
description="Delete a STAC item",
inputSchema={
"type": "object",
"properties": {
"collection_id": {
"type": "string",
"description": "ID of the collection",
},
"item_id": {
"type": "string",
"description": "ID of the item to delete",
},
},
"required": ["collection_id", "item_id"],
},
),
Tool(
name="create_collection",
description="Create a new STAC collection",
inputSchema={
"type": "object",
"properties": {
"collection": {
"type": "object",
"description": "STAC collection JSON",
},
},
"required": ["collection"],
},
),
Tool(
name="update_collection",
description="Update an existing STAC collection",
inputSchema={
"type": "object",
"properties": {
"collection": {
"type": "object",
"description": "STAC collection JSON to update",
},
},
"required": ["collection"],
},
),
Tool(
name="delete_collection",
description="Delete a STAC collection",
inputSchema={
"type": "object",
"properties": {
"collection_id": {
"type": "string",
"description": "ID of the collection to delete",
},
},
"required": ["collection_id"],
},
),
]
