select the correct region

src/india_api/internal/inputs/indiadb/client.py

@@ -62,7 +62,7 @@ def get_predicted_power_production_for_location(
         """Gets the predicted power production for a location.
 
         Args:
-            location: not used
+            location: the location to get the predicted power production for
             asset_type: The type of asset to get the forecast for
             forecast_horizon: The time horizon to get the data for. Can be latest or day ahead
             forecast_horizon_minutes: The number of minutes to get the forecast for. forecast_horizon must be 'horizon'
@@ -89,8 +89,15 @@ def get_predicted_power_production_for_location(
         with self._get_session() as session:
             sites = get_sites_by_country(session, country="india")
 
-            # just select wind site
-            sites = [s for s in sites if s.asset_type == asset_type]
+            # just select wind site and region
+            sites = [s for s in sites if (s.asset_type == asset_type) and (s.region == location)]
+
+            if len(sites) == 0:
+                raise HTTPException(
+                    status_code=204,
+                    detail=f"Site for {location=} not found and {asset_type=} not found",
+                )
+
             site = sites[0]
 
             # read actual generations

src/india_api/internal/inputs/indiadb/conftest.py

@@ -65,6 +65,7 @@ def sites(db_session):
         ml_id=1,
         asset_type="pv",
         country="india",
+        region='testID'
     )
     db_session.add(site)
     sites.append(site)
@@ -78,6 +79,7 @@ def sites(db_session):
         ml_id=2,
         asset_type="wind",
         country="india",
+        region='testID'
     )
     db_session.add(site)
     sites.append(site)

src/india_api/internal/inputs/indiadb/test_indiadb.py

@@ -32,6 +32,13 @@ def test_get_predicted_wind_power_production_for_location(
         for record in result:
             assert isinstance(record, PredictedPower)
 
+    def test_get_predicted_wind_power_production_for_location_raise_error(
+        self, client, forecast_values
+    ) -> None:
+
+        with pytest.raises(Exception):
+            result = client.get_predicted_wind_power_production_for_location("testID2")
+
     def test_get_predicted_solar_power_production_for_location(
         self, client, forecast_values
     ) -> None: