terrakio-core 0.5.2__tar.gz → 0.5.3__tar.gz

{terrakio_core-0.5.2 → terrakio_core-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: terrakio-core
-Version: 0.5.2
+Version: 0.5.3
 Summary: Core package for the terrakio-python-api
 Requires-Python: >=3.11
 Requires-Dist: aiofiles>=24.1.0

{terrakio_core-0.5.2 → terrakio_core-0.5.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "terrakio-core"
-version = "0.5.2"
+version = "0.5.3"
 description = "Core package for the terrakio-python-api"
 readme = "README.md"
 requires-python = ">=3.11"

{terrakio_core-0.5.2 → terrakio_core-0.5.3}/terrakio_core/__init__.py

@@ -8,7 +8,7 @@ Core components for Terrakio API clients.
 # Suppress ONNX Runtime GPU device discovery warnings - MUST BE FIRST!
 import os
 os.environ['ORT_LOGGING_LEVEL'] = '3'
-__version__ = "0.5.2"
+__version__ = "0.5.3"
 
 from .async_client import AsyncClient
 from .sync_client import SyncClient as Client

{terrakio_core-0.5.2 → terrakio_core-0.5.3}/terrakio_core/async_client.py

@@ -9,7 +9,7 @@ from geopandas import GeoDataFrame
 from shapely.geometry.base import BaseGeometry as ShapelyGeometry
 from shapely.geometry import mapping
 from .client import BaseClient
-from .exceptions import APIError, NetworkError
+from .exceptions import APIError, NetworkError, GeoQueryError
 from .endpoints.dataset_management import DatasetManagement
 from .endpoints.user_management import UserManagement
 from .endpoints.mass_stats import MassStats
@@ -184,6 +184,9 @@ class AsyncClient(BaseClient):
         }
         result, status_code = await self._terrakio_request("POST", "geoquery", json=payload)
 
+        if status_code != 200:
+            raise GeoQueryError("Something went wrong with the geoquery request. Please try again later.", status_code=status_code)
+
         return result
 
     async def zonal_stats(self, *args, **kwargs):

{terrakio_core-0.5.2 → terrakio_core-0.5.3}/terrakio_core/exceptions.py

@@ -385,6 +385,12 @@ class CancelCollectionTasksError(Exception):
 
 class CancelAllTasksError(Exception):
    """Exception raised for cancel all tasks errors."""
+   def __init__(self, message: str, status_code: int = None):
+       super().__init__(message)
+       self.status_code = status_code
+
+class GeoQueryError(Exception):
+   """Exception raised for geo query errors."""
    def __init__(self, message: str, status_code: int = None):
        super().__init__(message)
        self.status_code = status_code