Tool/naip fetcher (#16)

* Add initial NAIP fetcher

* Swap to Element84's EarthSearch API for NAIP STAC search and download

* clip to bounds of aoi

* Swap to Element84's EarthSearch API for NAIP STAC search and download

* rename bands and remove dask chunking

* Add DS_Store to .gitignore

* restrict date range for naip test

* Adjust timerange for tests

* Add xarray to pyproj

* Reduce aoi size

* revert test to use tmp path

* Update return types for tool to ensure state gets updated

* Update unit test for compatibility with Command output from tool

* Save image bytes directly to graph state for summarizer

* add safeguard against large image sizes

* remove print statement

* Fix stac.load to work with MCP API by manually inserting band data into the  extension

* Remove deleted file

* Add comment explaining workaround

---------

Co-authored-by: lillythomas <lillyelizathomas@gmail.com>
Co-authored-by: Daniel Wiesmann <yellowcap@users.noreply.github.com>

tests/tools/test_naip.py

@@ -1,4 +1,4 @@
-from pathlib import Path
+from types import NoneType
 
 import pytest
 from langchain_core.tools.base import ToolCall
@@ -8,8 +8,7 @@ from geo_assistant.tools.naip import fetch_naip_img
 
 
 @pytest.mark.asyncio
-@pytest.mark.xfail
-async def test_fetch_naip(tmp_path):
+async def test_fetch_naip():
     """
     Integration test: hit MPC STAC for NAIP around Union Market (DC),
     load imagery via odc-stac, and save an RGB PNG.
@@ -22,38 +21,67 @@ async def test_fetch_naip(tmp_path):
     # Union Market coordinates from GeoNames: 38.90789, -76.99831
     #   N 38°54'28"  W 76°59'54"
     # We'll use a small neighborhood AOI around that point.
-    # :contentReference[oaicite:0]{index=0}
     lat = 38.90789
     lon = -76.99831
 
-    # ~0.01 degrees (~1.1 km) buffer in each direction
+    # ~0.0001 degrees buffer in each direction
     aoi = box(lon - 0.0001, lat - 0.0001, lon + 0.0001, lat + 0.0001)
     aoi_geojson = mapping(aoi)
 
-    out_png = tmp_path / "naip_test_img.png"
-
     tool_call = ToolCall(
         name="fetch_naip_img",
         args={
             "aoi_geojson": aoi_geojson,
             "start_date": "2021-01-01",
             "end_date": "2021-12-31",
-            "out_png_path": str(out_png),
-            "resolution": 1.0,
         },
         type="tool_call",
         id="test_tool_call_id",
     )
 
     # Call the actual tool - no STAC / odc-stac mocking
-    result = await fetch_naip_img.ainvoke(tool_call["args"])
+    result = await fetch_naip_img.ainvoke(tool_call)
+    assert "naip_img_bytes" in result.update
+    assert result.update["naip_img_bytes"] is not None, "Expected PNG bytes in result"
+    assert isinstance(result.update["naip_img_bytes"], bytes)
+    assert len(result.update["naip_img_bytes"]) > 1, "Expected non-empty PNG bytes"
 
-    # Basic sanity checks on result
-    assert result["stac_item_count"] > 0, "Expected at least one NAIP item"
-    assert "time" in result["dataset_dims"]
-    assert result["dataset_dims"]["time"] >= 1
 
-    # PNG should have been written to disk
-    png_path = Path(result["png_path"])
-    assert png_path == out_png
-    assert png_path.is_file(), f"PNG was not created at {png_path}"
+@pytest.mark.asyncio
+async def test_fetch_naip_too_large():
+    """
+    Integration test: request a larger AOI that should produce an image
+    exceeding the 512x512 pixel limit. The tool should return no image
+    bytes and include a message indicating it skipped output due to size.
+
+    NOTE: This test requires:
+      - Internet access (to reach Planetary Computer STAC + blobs)
+      - Planetary Computer / NAIP service to be up
+    """
+
+    # Union Market coordinates from GeoNames: 38.90789, -76.99831
+    #   N 38°54'28"  W 76°59'54"
+    # We'll use a small neighborhood AOI around that point.
+    lat = 38.90789
+    lon = -76.99831
+
+    # ~0.003 degrees buffer in each direction
+    aoi = box(lon - 0.003, lat - 0.003, lon + 0.003, lat + 0.003)
+    aoi_geojson = mapping(aoi)
+
+    tool_call = ToolCall(
+        name="fetch_naip_img",
+        args={
+            "aoi_geojson": aoi_geojson,
+            "start_date": "2021-01-01",
+            "end_date": "2021-12-31",
+        },
+        type="tool_call",
+        id="test_tool_call_id",
+    )
+
+    # Call the actual tool - no STAC / odc-stac mocking
+    result = await fetch_naip_img.ainvoke(tool_call)
+    assert "naip_img_bytes" in result.update
+    assert result.update["naip_img_bytes"] is None, "Expected no PNG bytes in result"
+    assert isinstance(result.update["naip_img_bytes"], NoneType)