modify search to fix claude issues

Author: angaither

src/planet_mcp/models.py

@@ -6,3 +6,6 @@ class Geometry(TypedDict):
     type: str
     coordinates: Any
     content: str | None
+
+
+# class Filter

src/planet_mcp/servers/__init__.py

@@ -1,7 +1,9 @@
 from . import sdk
 from . import tiles
+from . import search
 
 all = [
     sdk,
     tiles,
+    search,
 ]

src/planet_mcp/servers/sdk.py

@@ -5,11 +5,12 @@
 
 from planet_mcp import models
 from planet_mcp.clients import session
+from planet_mcp.servers import search
 from . import descriptions
 from fastmcp import FastMCP
 import planet
 from typing import Union
-
+from datetime import datetime
 
 # tools we don't want enabled at all.
 # they simply don't work well in an AI context.
@@ -19,6 +20,7 @@
     "data_get_search",
     "data_get_stats",
     "data_list_searches",
+    "data_search",
     "data_update_search",
     "data_wait_asset",
     "destinations_patch_destination",
@@ -35,7 +37,6 @@
 TOOL_SIG_OVERRIDE = {
     "features_add_items",
     "data_get_item_coverage",
-    "data_search",
     "mosaics_download_quad",
     "mosaics_download_quads",
     "mosaics_get_quad",

src/planet_mcp/servers/search.py

@@ -0,0 +1,47 @@
+from datetime import datetime
+
+from fastmcp import FastMCP
+from planet import Planet
+from planet_mcp import models
+from planet_mcp.servers import descriptions
+
+# technically not using the sdk to make the tool but we can add it to that
+
+mcp = FastMCP("search")
+
+
+@mcp.tool(
+    name="data_search",
+    description=descriptions.overrides["data_search"],
+    tags={"data"},
+    enabled=True,
+)
+async def data_search(
+    item_types: list[str],
+    start_date: str,
+    end_date: str,
+    geometry: models.Geometry,
+):
+
+    planet = Planet()
+    start_dt = datetime.fromisoformat(start_date.replace("Z", "+00:00"))
+    end_dt = datetime.fromisoformat(end_date.replace("Z", "+00:00"))
+    datefilter = {
+        "type": "DateRangeFilter",
+        "field_name": "acquired",
+        "config": {"gte": start_dt.isoformat(), "lte": end_dt.isoformat()},
+    }
+    cloud_filter = {
+        "type": "RangeFilter",
+        "field_name": "cloud_cover",
+        "config": {"lte": 0.2},
+    }
+    filter = {
+        "type": "AndFilter",
+        "config": [datefilter, cloud_filter],
+    }
+    return planet.data.search(
+        item_types=item_types,
+        geometry=dict(geometry),
+        search_filter=filter,
+    )