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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: terrakio-core
-Version: 0.5.1
+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.1 → terrakio_core-0.5.3}/pyproject.toml

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

{terrakio_core-0.5.1 → 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.1"
+__version__ = "0.5.3"
 
 from .async_client import AsyncClient
 from .sync_client import SyncClient as Client

{terrakio_core-0.5.1 → 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
@@ -54,7 +54,6 @@ class AsyncClient(BaseClient):
         for attempt in range(self.retry + 1):
             try:
                 async with session.request(method, url, **kwargs) as response:
-                    content = await response.text()
                     
                     if not response.ok and self._should_retry(response.status, attempt):
                         self.logger.info(f"Request failed (attempt {attempt+1}/{self.retry+1}): {response.status}. Retrying...")
@@ -183,7 +182,10 @@ class AsyncClient(BaseClient):
             "validated": validated,
             **kwargs
         }
-        result = await self._terrakio_request("POST", "geoquery", json=payload)
+        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
 

{terrakio_core-0.5.1 → 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