fix(parser): handle new state legislative district response structure

The API now returns state_legislative_districts as a dict with
house/senate keys containing legislator info, but the parser only
checked for the "stateleg" key and expected a flat list. This caused
all stateleg data to silently land in extras as None.

Add _parse_stateleg helper that handles both the new dict format
and legacy flat list format, and check for both "stateleg" and
"state_legislative_districts" response keys.

Author: MiniCodeMonkey

src/geocodio/client.py

@@ -493,6 +493,28 @@ def _parse_list_response(response_json: dict, response: httpx.Response = None) -
         )
 
 
+    @staticmethod
+    def _parse_stateleg(data) -> list:
+        """Parse state legislative district data.
+
+        Handles both formats:
+        - Dict with house/senate keys: {house: [...], senate: [...]}
+        - Flat list of district dicts (legacy)
+        """
+        if isinstance(data, dict):
+            districts = []
+            for chamber, chamber_districts in data.items():
+                if isinstance(chamber_districts, list):
+                    for district in chamber_districts:
+                        district_data = dict(district)
+                        if "chamber" not in district_data:
+                            district_data["chamber"] = chamber
+                        districts.append(StateLegislativeDistrict.from_api(district_data))
+            return districts
+        elif isinstance(data, list):
+            return [StateLegislativeDistrict.from_api(d) for d in data]
+        return []
+
     def _parse_fields(self, fields_data: dict | None) -> GeocodioFields | None:
         """
         Parse fields data from API response.
@@ -522,17 +544,15 @@ def _parse_fields(self, fields_data: dict | None) -> GeocodioFields | None:
 
         state_legislative_districts = None
         if "stateleg" in fields_data:
-            state_legislative_districts = [
-                StateLegislativeDistrict.from_api(district)
-                for district in fields_data["stateleg"]
-            ]
+            state_legislative_districts = self._parse_stateleg(fields_data["stateleg"])
+        elif "state_legislative_districts" in fields_data:
+            state_legislative_districts = self._parse_stateleg(fields_data["state_legislative_districts"])
 
         state_legislative_districts_next = None
         if "stateleg-next" in fields_data:
-            state_legislative_districts_next = [
-                StateLegislativeDistrict.from_api(district)
-                for district in fields_data["stateleg-next"]
-            ]
+            state_legislative_districts_next = self._parse_stateleg(fields_data["stateleg-next"])
+        elif "state_legislative_districts_next" in fields_data:
+            state_legislative_districts_next = self._parse_stateleg(fields_data["state_legislative_districts_next"])
 
         # School districts - support both nested dict and flat list formats
         school_districts = None
@@ -713,7 +733,7 @@ def parse_census_data(data: dict) -> dict:
         # Collect all known field keys that were parsed
         parsed_keys = {
             "timezone", "cd", "congressional_districts",
-            "stateleg", "stateleg-next",
+            "stateleg", "stateleg-next", "state_legislative_districts", "state_legislative_districts_next",
             "school", "school_districts",  # Both school formats
             "census",  # Nested census structure
             "acs",  # Nested ACS structure

tests/unit/test_geocode.py

@@ -596,4 +596,115 @@ def response_callback(request):
     # This will fail until we fix the parsing logic
     assert result.fields.census2024 is not None
     assert result.fields.census2024.tract == "960100"
-    assert result.fields.census2024.block == "2004"
+    assert result.fields.census2024.block == "2004"
+
+
+def test_geocode_with_stateleg_fields(client, httpx_mock):
+    """Test geocoding with state legislative district fields.
+
+    The API returns state_legislative_districts as a dict with house/senate keys,
+    each containing a list of district objects with legislator info.
+    """
+    def response_callback(request):
+        assert request.url.params["fields"] == "stateleg"
+        return httpx.Response(200, json={
+            "input": {"formatted_address": "600 Santa Ray Ave, Oakland, CA 94610"},
+            "results": [{
+                "address_components": {
+                    "number": "600",
+                    "street": "Santa Ray",
+                    "suffix": "Ave",
+                    "city": "Oakland",
+                    "state": "CA",
+                    "zip": "94610",
+                    "country": "US"
+                },
+                "formatted_address": "600 Santa Ray Ave, Oakland, CA 94610",
+                "location": {"lat": 37.811943, "lng": -122.240213},
+                "accuracy": 1,
+                "accuracy_type": "rooftop",
+                "source": "Alameda",
+                "fields": {
+                    "state_legislative_districts": {
+                        "house": [
+                            {
+                                "name": "Assembly District 18",
+                                "district_number": "18",
+                                "ocd_id": "ocd-division/country:us/state:ca/sldl:18",
+                                "is_upcoming_state_legislative_district": False,
+                                "proportion": 1,
+                                "current_legislators": [
+                                    {
+                                        "type": "representative",
+                                        "bio": {
+                                            "last_name": "Bonta",
+                                            "first_name": "Mia",
+                                            "party": "Democrat"
+                                        }
+                                    }
+                                ]
+                            }
+                        ],
+                        "senate": [
+                            {
+                                "name": "Senate District 7",
+                                "district_number": "7",
+                                "ocd_id": "ocd-division/country:us/state:ca/sldu:7",
+                                "is_upcoming_state_legislative_district": False,
+                                "proportion": 1,
+                                "current_legislators": [
+                                    {
+                                        "type": "senator",
+                                        "bio": {
+                                            "last_name": "Arreguin",
+                                            "first_name": "Jesse",
+                                            "party": "Democrat"
+                                        }
+                                    }
+                                ]
+                            }
+                        ]
+                    }
+                }
+            }]
+        })
+
+    httpx_mock.add_callback(
+        callback=response_callback,
+        url=httpx.URL("https://api.test/v1.9/geocode", params={
+            "q": "600 Santa Ray Ave, Oakland CA 94610",
+            "fields": "stateleg"
+        }),
+        match_headers={"Authorization": "Bearer TEST_KEY"},
+    )
+
+    # Act
+    resp = client.geocode("600 Santa Ray Ave, Oakland CA 94610", fields=["stateleg"])
+
+    # Assert - state legislative districts should be parsed, not None
+    fields = resp.results[0].fields
+    assert fields.state_legislative_districts is not None
+    assert len(fields.state_legislative_districts) == 2
+
+    # Check house district
+    house = [d for d in fields.state_legislative_districts if d.chamber == "house"]
+    assert len(house) == 1
+    assert house[0].name == "Assembly District 18"
+    assert house[0].district_number == "18"
+    assert house[0].ocd_id == "ocd-division/country:us/state:ca/sldl:18"
+    assert house[0].proportion == 1
+
+    # Check senate district
+    senate = [d for d in fields.state_legislative_districts if d.chamber == "senate"]
+    assert len(senate) == 1
+    assert senate[0].name == "Senate District 7"
+    assert senate[0].district_number == "7"
+    assert senate[0].ocd_id == "ocd-division/country:us/state:ca/sldu:7"
+
+    # Check that legislator info is accessible via extras
+    legislators = house[0].get_extra("current_legislators", [])
+    assert len(legislators) == 1
+    assert legislators[0]["bio"]["last_name"] == "Bonta"
+
+    # Ensure state_legislative_districts didn't leak into extras
+    assert "state_legislative_districts" not in fields.extras