use geometry model instead of polygon/point

Author: stephenhillier

src/planet_mcp/models.py

@@ -1,11 +1,7 @@
-from typing_extensions import Literal, TypedDict
+from typing import Any
+from typing_extensions import TypedDict
 
 
-class Polygon(TypedDict):
-    type: Literal["Polygon"]
-    coordinates: list[list[list[float]]]
-
-
-class Point(TypedDict):
-    type: Literal["Point"]
-    coordinates: list[float]
+class Geometry(TypedDict):
+    type: str
+    coordinates: Any

src/planet_mcp/servers/sdk.py

@@ -150,8 +150,8 @@ async def wrapper(*args, **kwargs):
             if param_name in ("feature", "quad", "mosaic", "series"):
                 wrapper.__annotations__[param_name] = dict
             elif param_name == "geometry" and "planet.models" in str(param.annotation):
-                # llms should always submit geometry inputs as a Polygon or Point
-                hint = models.Polygon | models.Point
+                # llms should always submit geometry inputs as a geojson geometry
+                hint = models.Geometry
 
                 # add None if originally used (NoneType will be an arg
                 # within a Union type)