geoai-py 0.16.0__tar.gz → 0.18.0__tar.gz

{geoai_py-0.16.0 → geoai_py-0.18.0}/.gitignore

@@ -7,6 +7,7 @@ private/
 tests/data/
 CLAUDE.md
 AI_AGENTS.md
+AGENTS.md
 docs/examples/timm_output/
 docs/examples/timm_seg_output/
 docs/examples/timm_buildings/
@@ -17,6 +18,8 @@ docs/examples/timm_buildings/
 **/*.zip
 **/*.las
 *.geojson
+*.gpkg
+*.csv
 docs/examples/*.md
 *.xml
 docs/examples/output/

{geoai_py-0.16.0 → geoai_py-0.18.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geoai-py
-Version: 0.16.0
+Version: 0.18.0
 Summary: A Python package for using Artificial Intelligence (AI) with geospatial data
 Author-email: Qiusheng Wu <giswqs@gmail.com>
 License: MIT License
@@ -39,15 +39,18 @@ Requires-Dist: rioxarray
 Requires-Dist: scikit-image
 Requires-Dist: scikit-learn
 Requires-Dist: timm
+Requires-Dist: tokenizers>=0.22.1
 Requires-Dist: torch
 Requires-Dist: torchgeo
 Requires-Dist: torchinfo
 Requires-Dist: tqdm
-Requires-Dist: transformers
+Requires-Dist: transformers>=4.57.1
 Provides-Extra: extra
 Requires-Dist: overturemaps; extra == "extra"
 Requires-Dist: torchange; extra == "extra"
 Requires-Dist: lightly-train; extra == "extra"
+Requires-Dist: multiclean; extra == "extra"
+Requires-Dist: omnicloudmask; extra == "extra"
 Provides-Extra: agents
 Requires-Dist: strands-agents; extra == "agents"
 Requires-Dist: strands-agents-tools; extra == "agents"

{geoai_py-0.16.0 → geoai_py-0.18.0}/geoai/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = """Qiusheng Wu"""
 __email__ = "giswqs@gmail.com"
-__version__ = "0.16.0"
+__version__ = "0.18.0"
 
 
 import os
@@ -121,3 +121,18 @@ from .timm_segment import (
     timm_semantic_segmentation,
     push_timm_model_to_hub,
 )
+
+# Import tools subpackage
+from . import tools
+
+# Expose commonly used tools at package level for convenience
+try:
+    from .tools import (
+        clean_segmentation_mask,
+        clean_raster,
+        clean_raster_batch,
+        compare_masks,
+    )
+except ImportError:
+    # MultiClean not available (missing dependency)
+    pass

{geoai_py-0.16.0 → geoai_py-0.18.0}/geoai/agents/__init__.py

@@ -1,8 +1,12 @@
+from .catalog_tools import CatalogTools
 from .geo_agents import (
+    CatalogAgent,
     GeoAgent,
+    STACAgent,
     create_ollama_model,
     create_anthropic_model,
     create_openai_model,
     create_bedrock_model,
 )
 from .map_tools import MapTools
+from .stac_tools import STACTools

geoai_py-0.18.0/geoai/agents/catalog_models.py

@@ -0,0 +1,51 @@
+"""Structured output models for catalog search results."""
+
+from typing import Any, Dict, List, Optional
+
+from pydantic import BaseModel, Field
+
+
+class CatalogDatasetInfo(BaseModel):
+    """Information about a catalog dataset."""
+
+    id: str = Field(..., description="Dataset identifier")
+    title: str = Field(..., description="Dataset title")
+    type: Optional[str] = Field(
+        None, description="Dataset type (e.g., image, image_collection, table)"
+    )
+    provider: Optional[str] = Field(None, description="Data provider")
+    description: Optional[str] = Field(None, description="Dataset description")
+    keywords: Optional[str] = Field(None, description="Keywords/tags")
+    snippet: Optional[str] = Field(
+        None, description="Code snippet to access the dataset"
+    )
+    start_date: Optional[str] = Field(None, description="Start date of coverage")
+    end_date: Optional[str] = Field(None, description="End date of coverage")
+    bbox: Optional[str] = Field(None, description="Bounding box")
+    license: Optional[str] = Field(None, description="License information")
+    url: Optional[str] = Field(None, description="Documentation URL")
+    catalog: Optional[str] = Field(None, description="Catalog URL")
+    deprecated: Optional[str] = Field(None, description="Deprecated status")
+
+
+class CatalogSearchResult(BaseModel):
+    """Container for catalog search results."""
+
+    query: str = Field(..., description="Original search query")
+    dataset_count: int = Field(..., description="Number of datasets found")
+    datasets: List[CatalogDatasetInfo] = Field(
+        default_factory=list, description="List of catalog datasets"
+    )
+    filters: Optional[Dict[str, Any]] = Field(
+        None, description="Filters applied to search"
+    )
+
+
+class LocationInfo(BaseModel):
+    """Geographic location information."""
+
+    name: str = Field(..., description="Location name")
+    bbox: List[float] = Field(
+        ..., description="Bounding box [west, south, east, north]"
+    )
+    center: List[float] = Field(..., description="Center coordinates [lon, lat]")